Handle case where CodeQL location doesn't have region (#532)

pixee · Apr 30, 2024 · a678975 · a678975
1 parent 235e800
commit a678975
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/codemodder/codeql.py b/src/codemodder/codeql.py
@@ -45,7 +45,12 @@ def from_sarif(
 
         locations: list[Location] = []
         for location in sarif_result["locations"]:
-            codeql_location = CodeQLLocation.from_sarif(location)
+            try:
+                codeql_location = CodeQLLocation.from_sarif(location)
+            except KeyError:
+                # TODO: handle this case more gracefully
+                continue
+
             locations.append(codeql_location)
         return cls(rule_id=rule_data["id"], locations=locations)