fix(python): import statements (#1623)

Bearer · Jun 6, 2024 · 6c3035a · 6c3035a
1 parent a23c509
commit 6c3035a
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/pkg/languages/python/analyzer/analyzer.go b/pkg/languages/python/analyzer/analyzer.go
@@ -108,6 +108,9 @@ func (analyzer *analyzer) analyzeAssignment(node *sitter.Node, visitChildren fun
 // foo.bar(a, b)
 func (analyzer *analyzer) analyzeCall(node *sitter.Node, visitChildren func() error) error {
 	if function := node.ChildByFieldName("function"); function != nil {
+		analyzer.lookupVariable(function)
+		analyzer.builder.Dataflow(function)
+
 		object := function.ChildByFieldName("object")
 		analyzer.lookupVariable(object)
 

diff --git a/pkg/languages/python/testdata/import/import.py b/pkg/languages/python/testdata/import/import.py
@@ -35,4 +35,6 @@
 
 import FooClass as SomethingElse
 y = SomethingElse()
-y.qwerty()
+y.qwerty()
+
+foo()
diff --git a/pkg/languages/python/testdata/import_rule.yml b/pkg/languages/python/testdata/import_rule.yml
@@ -1,6 +1,11 @@
 languages:
   - python
 patterns:
+  - pattern: $<IMPORT_FROM>($<...>)
+    filters:
+      - variable: IMPORT_FROM
+        detection: import_test_import_from
+        scope: result
   - pattern: $<IMPORT_FROM>.someMethod($<...>)
     filters:
       - variable: IMPORT_FROM