fix: handle identifier case (import)

Bearer · Jun 5, 2024 · 7b8a1f6 · 7b8a1f6
1 parent e574802
commit 7b8a1f6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/languages/python/analyzer/analyzer.go b/pkg/languages/python/analyzer/analyzer.go
@@ -111,6 +111,10 @@ func (analyzer *analyzer) analyzeCall(node *sitter.Node, visitChildren func() er
 		object := function.ChildByFieldName("object")
 		analyzer.lookupVariable(object)
 
+		if function.Type() == "identifier" {
+			analyzer.builder.Dataflow(node, object)
+		}
+
 		if function.Type() == "attribute" {
 			attribute := function.ChildByFieldName("attribute")
 			if attribute.Type() == "identifier" && slices.Contains(reflexiveMethods, analyzer.builder.ContentFor(attribute)) {