Include codeflows when filtering applicable findings

drdavella · drdavella · commit 6b45f18db1d7 · 2024-12-03T15:35:39.000-05:00
diff --git a/src/codemodder/file_context.py b/src/codemodder/file_context.py
@@ -58,6 +58,11 @@ def get_findings_for_location(self, line_number: int):
                 location.start.line <= line_number <= location.end.line
                 for location in result.locations
             )
+            or any(
+                location.start.line <= line_number <= location.end.line
+                for codeflow in result.codeflows
+                for location in codeflow
+            )
             and result.finding is not None
         ]
 

Original file line number	Diff line number	Diff line change
`@@ -58,6 +58,11 @@ def get_findings_for_location(self, line_number: int):`
`58`	`58`	`location.start.line <= line_number <= location.end.line`
`59`	`59`	`for location in result.locations`
`60`	`60`	`)`
	`61`	`+ or any(`
	`62`	`+ location.start.line <= line_number <= location.end.line`
	`63`	`+ for codeflow in result.codeflows`
	`64`	`+ for location in codeflow`
	`65`	`+ )`
`61`	`66`	`and result.finding is not None`
`62`	`67`	`]`
`63`	`68`