dont import usbmux or installer on ios

permasigner · Jul 27, 2022 · d936b43 · d936b43
1 parent 932ab0c
commit d936b43
Showing 1 changed file with 15 additions and 12 deletions.
diff --git a/main.py b/main.py
@@ -1,3 +1,6 @@
+from utils.copy import Copy
+from utils.hash import LdidHash
+from utils.downloader import DpkgDeb, Ldid
 import os
 from pathlib import Path
 from shutil import copy, copytree
@@ -13,11 +16,18 @@
 from glob import glob
 from subprocess import PIPE, DEVNULL
 
-from utils.copy import Copy
-from utils.downloader import DpkgDeb, Ldid
-from utils.hash import LdidHash
-from utils.installer import Installer
-from utils.usbmux import USBMux
+
+def is_ios():
+    # Check iOS function is up here so we can conditionally import stuff
+    if not sys.platform == "darwin":
+        return False
+
+    return platform.machine().startswith("i")
+
+
+if not is_ios():
+    from utils.usbmux import USBMux
+    from utils.installer import Installer
 
 """ Functions """
 
@@ -64,13 +74,6 @@ def is_linux():
     return sys.platform == "linux"
 
 
-def is_ios():
-    if not sys.platform == "darwin":
-        return False
-
-    return platform.machine().startswith("i")
-
-
 def is_dpkg_installed(pkg):
     return (os.system("dpkg -s " + pkg + "> /dev/null 2>&1")) == 0