Skip to content

Commit 1c08d6f

Browse files
committed
Fix linting errors
Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
1 parent 971e688 commit 1c08d6f

File tree

4 files changed

+34
-36
lines changed

4 files changed

+34
-36
lines changed

scanpipe/models.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2016,8 +2016,7 @@ def win_exes(self):
20162016
Keep sync with the content type implementation at ``typecode.contenttype``.
20172017
"""
20182018
return self.files().filter(
2019-
Q(file_type__icontains="for ms windows")
2020-
| Q(file_type__istartswith="pe32")
2019+
Q(file_type__icontains="for ms windows") | Q(file_type__istartswith="pe32")
20212020
)
20222021

20232022
def mach_os(self):

scanpipe/pipelines/deploy_to_develop.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -191,16 +191,12 @@ def map_javascript(self):
191191

192192
@group("Elf")
193193
def map_elf(self):
194-
"""
195-
Map DWARF paths using similarities of path suffixes.
196-
"""
194+
"""Map DWARF paths using similarities of path suffixes."""
197195
d2d.map_elfs(project=self.project, logger=self.log)
198196

199197
@group("Go")
200198
def map_go(self):
201-
"""
202-
Map DWARF paths using similarities of path suffixes.
203-
"""
199+
"""Map GO file paths using similarities of path suffixes."""
204200
d2d.map_go_paths(project=self.project, logger=self.log)
205201

206202
def match_directories_to_purldb(self):

scanpipe/pipes/d2d.py

Lines changed: 20 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1686,7 +1686,7 @@ def _map_paths_resource(
16861686
continue
16871687

16881688
matched_path_length = match.matched_path_length
1689-
if matched_path_length == 1 and len(match.resource_ids) != 1:
1689+
if check_match(match, matched_path_length):
16901690
not_mapped_paths.append(path)
16911691
continue
16921692

@@ -1717,26 +1717,28 @@ def _map_paths_resource(
17171717

17181718
if relations:
17191719
rels = CodebaseRelation.objects.bulk_create(relations.values())
1720-
if logger:
1721-
logger(
1722-
f"Created {len(rels)} mappings using {', '.join(map_types).upper()} for: {to_resource.path!r}"
1723-
)
1720+
logger(
1721+
f"""Created {len(rels)} mappings using
1722+
{', '.join(map_types).upper()} for: {to_resource.path!r}"""
1723+
)
17241724
else:
1725-
if logger:
1726-
logger(
1727-
f"No mappings using {', '.join(map_types).upper()} for: {to_resource.path!r}"
1728-
)
1725+
logger(
1726+
f"""No mappings using {', '.join(map_types).upper()} for:
1727+
{to_resource.path!r}"""
1728+
)
17291729

17301730
for map_type in map_types:
1731-
if to_resource.extra_data[f"{map_type}_not_mapped"]:
1731+
if to_resource.extra_data.get(f"{map_type}_not_mapped"):
17321732
to_resource.save()
1733-
if logger:
1734-
logger(
1735-
f"WARNING: {map_type.upper()} paths NOT mapped for: {to_resource.path!r}: "
1736-
+ ", ".join(
1737-
map(repr, to_resource.extra_data[f"{map_type}_not_mapped"])
1738-
)
1739-
)
1733+
logger(
1734+
f"""WARNING: {map_type.upper()} paths NOT mapped for:
1735+
{to_resource.path!r}: """
1736+
+ ", ".join(map(repr, to_resource.extra_data[f"{map_type}_not_mapped"]))
1737+
)
1738+
1739+
1740+
def check_match(match, matched_path_length):
1741+
return matched_path_length == 1 and len(match.resource_ids) != 1
17401742

17411743

17421744
def map_paths(project, file_type, collect_paths_func, map_types, logger=None):
@@ -1760,7 +1762,7 @@ def map_paths(project, file_type, collect_paths_func, map_types, logger=None):
17601762
)
17611763

17621764
if logger:
1763-
logger(f"Done building from/ resources index.")
1765+
logger("Done building from/ resources index.")
17641766

17651767
resource_iterator = to_resources.iterator(chunk_size=2000)
17661768
progress = LoopProgress(resource_count, logger)

scanpipe/tests/pipes/test_d2d.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1485,14 +1485,15 @@ def test_scanpipe_pipes_d2d_map_elfs(self):
14851485
)
14861486
pipes.collect_and_create_codebase_resources(self.project1)
14871487
print(self.project1.codebaseresources.files().from_codebase())
1488-
d2d.map_elfs(project=self.project1, logger=None)
1489-
assert (
1488+
buffer = io.StringIO()
1489+
d2d.map_elfs(project=self.project1, logger=buffer.write)
1490+
self.assertEqual(
1491+
1,
14901492
CodebaseRelation.objects.filter(
14911493
project=self.project1, map_type="dwarf_included_paths"
1492-
).count()
1493-
== 1
1494+
).count(),
14941495
)
1495-
1496+
14961497
def test_scanpipe_pipes_d2d_map_go_paths(self):
14971498
input_dir = self.project1.input_path
14981499
input_resources = [
@@ -1514,11 +1515,11 @@ def test_scanpipe_pipes_d2d_map_go_paths(self):
15141515
recurse=True,
15151516
)
15161517
pipes.collect_and_create_codebase_resources(self.project1)
1517-
1518-
d2d.map_go_paths(project=self.project1, logger=None)
1519-
assert (
1518+
buffer = io.StringIO()
1519+
d2d.map_go_paths(project=self.project1, logger=buffer.write)
1520+
self.assertEqual(
1521+
1,
15201522
CodebaseRelation.objects.filter(
15211523
project=self.project1, map_type="go_file_paths"
1522-
).count()
1523-
== 1
1524+
).count(),
15241525
)

0 commit comments

Comments
 (0)