add PYTHONPATH for extension's install.py

1 files changed, 6 insertions, 3 deletions

diff --git a/launch.py b/launch.py
index 0d90d553..ff2f74ba 100644
--- a/launch.py
+++ b/launch.py
@@ -17,11 +17,11 @@ def extract_arg(args, name):
     return [x for x in args if x != name], name in args

 

 

-def run(command, desc=None, errdesc=None):

+def run(command, desc=None, errdesc=None, custom_env=None):

     if desc is not None:

         print(desc)

 

-    result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)

+    result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, env=os.environ if custom_env is None else custom_env)

 

     if result.returncode != 0:

 

@@ -115,7 +115,10 @@ def run_extensions_installers():
             continue

 

         try:

-            print(run(f'"{python}" "{path_installer}"', errdesc=f"Error running install.py for extension {dirname_extension}"))

+            env = os.environ.copy()

+            env['PYTHONPATH'] = os.path.abspath(".")

+

+            print(run(f'"{python}" "{path_installer}"', errdesc=f"Error running install.py for extension {dirname_extension}", custom_env=env))

         except Exception as e:

             print(e, file=sys.stderr)