Update main.py

kdo2064 · Jun 29, 2024 · b458d86 · b458d86
1 parent 6cd5fd0
commit b458d86
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -1,17 +1,20 @@
 import os
 import platform
+
 try:
     import pytz
     import httpx
-except:
-    _ = os.system('pip install illusionanime illusioncolor httpx datetime pytz' if os.name == 'nt' else 'pip3 install illusionanime illusioncolor httpx datetime pytz')
+except ImportError:
+    os.system('pip install illusionanime illusioncolor httpx datetime pytz' if os.name == 'nt' else 'pip3 install illusionanime illusioncolor httpx datetime pytz')
+
 def check_os():
     os_name = platform.system()
 
     if os_name == 'Windows':
         os.system('python .\\windows\\run.py')
     elif os_name == 'Linux':
-        if 'TERMUX_PREFIX' in os.environ:
+        whoami = os.popen('whoami').read().strip()
+        if 'TERMUX_PREFIX' in os.environ or whoami == 'u0':
             os.system('cd linux && python3 run.py')
         else:
             os.system('cd linux && python3 run.py')