Skip to content

Commit 2ec02e5

Browse files
chore: linting
1 parent 6ba4bc3 commit 2ec02e5

File tree

1 file changed

+19
-12
lines changed

1 file changed

+19
-12
lines changed

sbom4python/scanner.py

Lines changed: 19 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def __init__(
4343
lifecycle="build",
4444
include_service=False,
4545
use_pip=False,
46-
python_path:str=None
46+
python_path: str = None,
4747
):
4848
self.record = []
4949
self.debug = debug
@@ -69,15 +69,15 @@ def __init__(
6969
def set_parent(self, module):
7070
self.parent = f"Python-{module}"
7171

72-
def run_pip_cmd(self, params:Iterable[str]):
72+
def run_pip_cmd(self, params: Iterable[str]):
7373
cmd = ["pip"]
7474
if self.python_path.exists():
7575
cmd.extend(("--python", str(self.python_path)))
7676

7777
cmd.extend(params)
7878
return self.run_program(cmd)
7979

80-
def run_program(self, params:Iterable[str]):
80+
def run_program(self, params: Iterable[str]):
8181
res = subprocess.run(list(params), capture_output=True, text=True)
8282
return res.stdout.splitlines()
8383

@@ -567,14 +567,16 @@ def process_python_module(self, module_name):
567567
def _get_installed_modules(self):
568568
modules = []
569569
if self.use_pip:
570-
out = self.run_pip_cmd(("list", ))
570+
out = self.run_pip_cmd(("list",))
571571
if len(out) > 0:
572572
# Ignore headers in output stream
573573
for m in out[2:]:
574574
modules.append(m.split(" ")[0])
575575
else:
576576
installed_packages_info = importlib_metadata.distributions()
577-
modules = sorted([p.metadata["Name"].lower() for p in installed_packages_info])
577+
modules = sorted(
578+
[p.metadata["Name"].lower() for p in installed_packages_info]
579+
)
578580
if self.debug:
579581
print(modules)
580582
return modules
@@ -715,18 +717,21 @@ def process_pylock(self, filename):
715717
pylock_data = toml.load(file)
716718
if "lock-version" in pylock_data:
717719
if self.debug:
718-
print (pylock_data)
720+
print(pylock_data)
719721
if "packages" in pylock_data:
720722
self.set_lifecycle("pre-build")
721723
self.set_parent(filename)
722724
for package in pylock_data["packages"]:
723725
if "version" in package:
724726
self._process_requirement_dependency(
725-
f"{package['name']}=={package['version']}", filename
727+
f"{package['name']}=={package['version']}",
728+
filename,
726729
)
727730
if "dependencies" in package:
728-
for dependency in package['dependencies']:
729-
self._create_relationship(dependency['name'], package['name'])
731+
for dependency in package["dependencies"]:
732+
self._create_relationship(
733+
dependency["name"], package["name"]
734+
)
730735

731736
def process_uvlock_file(self, filename):
732737
# Process uv.lock file
@@ -738,7 +743,7 @@ def process_uvlock_file(self, filename):
738743
with open(filename) as file:
739744
uvlock_data = toml.load(file)
740745
if self.debug:
741-
print (uvlock_data)
746+
print(uvlock_data)
742747
if "package" in uvlock_data:
743748
self.set_lifecycle("build")
744749
self.set_parent(filename)
@@ -748,5 +753,7 @@ def process_uvlock_file(self, filename):
748753
f"{package['name']}=={package['version']}", filename
749754
)
750755
if "dependencies" in package:
751-
for dependency in package['dependencies']:
752-
self._create_relationship(dependency['name'], package['name'])
756+
for dependency in package["dependencies"]:
757+
self._create_relationship(
758+
dependency["name"], package["name"]
759+
)

0 commit comments

Comments
 (0)