pythonpath try outs

Author: bitterpanda63

aikido_zen/cli/__init__.py

@@ -1,28 +1,32 @@
+import os
+import shutil
 import sys
 import subprocess
 from typing import List
 
 import aikido_zen
 
 
-def run_command(args: List[str]) -> int:
+def add_preload_directory(preload_directory: str):
     """
-    Run the provided command as a subprocess.
+    Idea here is to load in the code stored in the preload_directory first, Inspired by ddtrace-run:
+    https://github.com/DataDog/dd-trace-py/blob/3ad335edd4a032ea53680fefbcc10f8fca0690a0/ddtrace/commands/ddtrace_run.py#L41
     """
-    if not args:
-        print("Error: No command provided.", file=sys.stderr)
-        return 1
+    python_path = os.environ.get("PYTHONPATH", "")
 
-    try:
-        # Run the command as a subprocess
-        result = subprocess.run(args, check=False)
-        return result.returncode
-    except KeyboardInterrupt:
-        print("\nCommand interrupted by user.", file=sys.stderr)
-        return 130  # SIGINT exit code
-    except Exception as e:
-        print(f"Error: {e}", file=sys.stderr)
-        return 1
+    if python_path:
+        new_path = "%s%s%s" % (preload_directory, os.path.pathsep, os.environ["PYTHONPATH"])
+        os.environ["PYTHONPATH"] = new_path
+    else:
+        os.environ["PYTHONPATH"] = preload_directory
+
+    print(os.environ["PYTHONPATH"])
+
+
+def get_executable(command):
+    if os.path.isfile(command):
+        return command
+    return shutil.which(command)
 
 def main() -> int:
     """
@@ -32,14 +36,21 @@ def main() -> int:
         print("Usage: aikido_zen <command> [args...]", file=sys.stderr)
         return 1
 
-    # Start background process
-    aikido_zen.protect(mode="daemon_only")
+    root_dir = os.path.dirname(aikido_zen.__file__)
+    preload_directory = os.path.join(root_dir, "cli/preload")
+    add_preload_directory(preload_directory)
 
     # The command to run is everything after `aikido_zen`
-    command = sys.argv[1:]
-    status = run_command(command)
+    executable = get_executable(sys.argv[1])
+    try:
+        os.execl(executable, executable, *sys.argv[2:])
+    except PermissionError:
+        print("aikido_zen: permission error while launching '%s'" % executable)
+        return 1
+    except Exception:
+        print("ddtrace-run: error launching '%s'" % executable)
+        raise
 
-    return status
 
 if __name__ == "__main__":
     sys.exit(main())

aikido_zen/cli/preload/__init__.py

@@ -0,0 +1,2 @@
+print("debug")
+raise Exception("debug")