From 7b8a1f66f448f37a427554dad5ee2435978a4d1e Mon Sep 17 00:00:00 2001
From: elsapet <elizabeth@bearer.sh>
Date: Wed, 5 Jun 2024 12:41:42 +0200
Subject: [PATCH] fix: handle identifier case (import)

---
 pkg/languages/python/analyzer/analyzer.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pkg/languages/python/analyzer/analyzer.go b/pkg/languages/python/analyzer/analyzer.go
index 49eca350d..5fd75a2a6 100644
--- 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)) {