From 42d1c3750b6e9dc38646c4e0ff56383fc5b178c6 Mon Sep 17 00:00:00 2001
From: SummerGift <SummerGift@qq.com>
Date: Tue, 27 Aug 2019 16:33:39 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E5=AE=8C=E5=96=84=E3=80=91=E8=87=AA?=
 =?UTF-8?q?=E5=8A=A8=E5=8D=87=E7=BA=A7=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 init_env.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/init_env.py b/init_env.py
index d27f52e..bd7ed32 100644
--- a/init_env.py
+++ b/init_env.py
@@ -25,17 +25,22 @@
 
 from multiprocessing import Process
 import os
-import time
+import sys
+
+def run_proc(name, env_root):
+    exec_file = os.path.join(env_root, "tools\scripts\env.py")
 
-def run_proc(name):
-    exec_file = os.path.join(os.getcwd(), "tools\scripts\env.py")
     try:
         os.system("python %s package --upgrade 1>std_null 2>err_null"%exec_file)
     except Exception, e:
         print("Auto upgrade failed, please check your network.")
         pass
 
-if __name__=='__main__':
-    p = Process(target=run_proc, args=('upgrade',) )
+def main():
+    env_root = sys.argv[1]
+    p = Process(target=run_proc, args=('upgrade', env_root) )
     p.start()
     p.join()
+
+if __name__=='__main__':
+    main()