feat/python guess inside ifstat (#240)

* Bumping python version

* Logging stderr in case of python failure

* Adding a failing test

* Adding a query for imports within a top-level "if" statement

* Handling more nested cases

* Reformatting query

* Formatting

internal/backends/python/grab.go

@@ -17,9 +17,25 @@ var importsQuery = `
      name: [(dotted_name) @import
             (aliased_import
               name: (dotted_name) @import)])
-
    (import_from_statement
-     module_name: (dotted_name) @import)]
+     module_name: (dotted_name) @import)
+   (if_statement
+     [(block
+        [(import_statement
+           name:
+            [(dotted_name) @import
+             (aliased_import
+               name: (dotted_name) @import)])
+         (import_from_statement
+           module_name: (dotted_name) @import)])
+      (_ (block
+           [(import_statement
+              name:
+               [(dotted_name) @import
+                (aliased_import
+                  name: (dotted_name) @import)])
+            (import_from_statement
+              module_name: (dotted_name) @import)]))])]
 
   .
 

internal/backends/python/grab_test.go

@@ -1,7 +1,9 @@
 package python
 
 import (
+	"bytes"
 	"context"
+	"fmt"
 	"os"
 	"os/exec"
 	"path"
@@ -18,6 +20,10 @@ func TestParseFile(t *testing.T) {
 		"replit":    true,
 		"replit.ai": true,
 		"bar":       true,
+		"cond1":     true,
+		"cond2":     true,
+		"cond3":     true,
+		"cond4":     true,
 	}
 
 	content := `
@@ -28,6 +34,15 @@ from Flask import Flask
 import replit
 import replit.ai
 import foo #upm package(bar)
+
+if False:
+    import cond1
+elif True:
+    import cond2
+elif True:
+    import cond3
+else:
+    import cond4
 `
 
 	testDir := t.TempDir()
@@ -144,8 +159,13 @@ import flask
 
 	// Sanity test, actually run Python in the environment.
 	cmd := exec.Command("bash", "-c", "poetry lock -n; poetry install; poetry run python -m foo")
+
+	var stderr bytes.Buffer
+	cmd.Stderr = &stderr
 	stdout, err := cmd.Output()
 	if err != nil {
+
+		fmt.Println("stderr:", stderr.String())
 		t.Fatal("failed to execute python", err)
 	}
 	lines := strings.Split(strings.TrimSpace(string(stdout)), "\n")

replit.nix

@@ -14,8 +14,8 @@
     pkgs.nodejs-18_x
     pkgs.yarn
     pkgs.nodePackages.pnpm
-    pkgs.python310Full
-    pkgs.python310Packages.pip
+    pkgs.python311Full
+    pkgs.python311Packages.pip
     pkgs.poetry
     pkgs.R
     pkgs.ruby