keep track of visited

vyperlang · Sep 8, 2024 · 0716f0a · 0716f0a
1 parent c412402
commit 0716f0a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/vyper/semantics/analysis/imports.py b/vyper/semantics/analysis/imports.py
@@ -78,6 +78,8 @@ def __init__(self, input_bundle: InputBundle, graph: _ImportGraph):
         self.graph = graph
         self._ast_of: dict[int, vy_ast.Module] = {}
 
+        self.seen = set()
+
         self.integrity_sum = None
 
     def resolve_imports(self, module_ast: vy_ast.Module):
@@ -98,12 +100,15 @@ def _calculate_integrity_sum_r(self, module_ast: vy_ast.Module):
         return sha256sum("".join(acc))
 
     def _resolve_imports_r(self, module_ast: vy_ast.Module):
+        if id(module_ast) in self.seen:
+            return
         with self.graph.enter_path(module_ast):
             for node in module_ast.body:
                 if isinstance(node, vy_ast.Import):
                     self._handle_Import(node)
                 elif isinstance(node, vy_ast.ImportFrom):
                     self._handle_ImportFrom(node)
+        self.seen.add(id(module_ast))
 
     def _handle_Import(self, node: vy_ast.Import):
         # import x.y[name] as y[alias]