Add method to retrieve locations

Author: drdavella

src/codemodder/result.py

@@ -103,6 +103,13 @@ def match_location(self, pos: CodeRange, node: cst.CSTNode) -> bool:
     def __hash__(self):
         return hash(self.rule_id)
 
+    def get_locations(self) -> Sequence[Sequence[Location]]:
+        return self.codeflows or (
+            [self.locations]
+            if self.locations
+            else [[loc.location for loc in self.related_locations]]
+        )
+
 
 @dataclass(frozen=True, kw_only=True)
 class SASTResult(Result):