new: pip loader will explicitly import the installed package

dreadnode · Jan 18, 2025 · 4559439 · 4559439
1 parent eed3260
commit 4559439
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/dyana/loaders/npm/main.py b/dyana/loaders/npm/main.py
@@ -20,9 +20,9 @@
         package_name = re.split("[^a-zA-Z0-9_-]", args.package)[0]
         result = subprocess.run(["node", "-e", f"require('{package_name}')"], capture_output=True, text=True)
 
-        profiler.track("require_exit_code", result.returncode)
-        profiler.track("require_stdout", result.stdout)
-        profiler.track("require_stderr", result.stderr)
+        profiler.track("exit_code", result.returncode)
+        profiler.track("stdout", result.stdout)
+        profiler.track("stderr", result.stderr)
 
     except Exception as e:
         profiler.track_error("npm", str(e))

diff --git a/dyana/loaders/pip/main.py b/dyana/loaders/pip/main.py
@@ -1,5 +1,6 @@
 import argparse
 import json
+import re
 import subprocess
 import sys
 
@@ -17,6 +18,14 @@
         )
         profiler.track_memory("after_installation")
         profiler.track_disk("after_installation")
+
+        # explicitly require the package to make sure it's loaded
+        package_name = re.split("[^a-zA-Z0-9_-]", args.package)[0]
+        result = subprocess.run(["python3", "-c", f"import {package_name}"], capture_output=True, text=True)
+
+        profiler.track("exit_code", result.returncode)
+        profiler.track("stdout", result.stdout)
+        profiler.track("stderr", result.stderr)
     except Exception as e:
         profiler.track_error("pip", str(e))