LeeKurnrun->优化install

installer/install.py

@@ -8,6 +8,7 @@
 import os
 import shutil
 import json
+import platform
 
 ############ Config ############
 KEYWORD = "winpe"
@@ -15,39 +16,51 @@ PYTHON = "python"
 DISK = "{disk}"
 ################################
 
-# Linux please run: mount /dev/sdX /mnt/disk --mkdir
+# if you using Linux based OS, please run this command: mount /dev/sdX /mnt/disk --mkdir
 # DISK = "/mnt/disk"
 
 
-def find_disk():  # This code from dashedgeless
+def find_disk():
-    disk_list = "CDEFGHIJKABLMNOPQRSTUVWYZ"
+    if platform.system() == 'Linux':
-    for i in disk_list:
+        if "{disk}" in DISK:
+            raise OSError(
+                "Cannot auto find the Ventoy disk in Linux! Please change \"DISK\"")
+    elif platform.system() == 'Windows':
+        ventoy_drives = []
+        for i in "CDEFGHIJKABLMNOPQRSTUVWYZ":
             this_disk = i+":"
             if os.path.exists(this_disk + "/ventoy/ventoy.json"):
-            return this_disk
+                ventoy_drives.append(this_disk)
+        if len(ventoy_drives) != 1:
+            for (index, value) in enumerate(ventoy_drives):
+                print(f"{i}:{ventoy_drives[i]}", end=" ")
+            return ventoy_drives[int(input("Please choose your Ventoy device:"))]
+    else:
+        raise OSError(f"Unsupported operating system:{platform.system()}")
+
     raise OSError(
         "Cannot find the Ventoy disk!(try to run ventoyPlugson once)")
 
 
 DISK = DISK.replace("{disk}", find_disk())
 
-print("use disk: "+DISK)
+print("Use disk: "+DISK)
 with open(DISK+"/ventoy/ventoy.json", "r") as file:
-    vtoycfg = json.load(file)
+    ventoy_cfg = json.load(file)
 searchroot = ""
-searchlevel = 3
+search_level = 3
-if "control" in vtoycfg:
+if "control" in ventoy_cfg:
-    for i in vtoycfg["control"]:
+    for i in ventoy_cfg["control"]:
         if "VTOY_DEFAULT_SEARCH_ROOT" in i:
             searchroot = i["VTOY_DEFAULT_SEARCH_ROOT"]
         if "VTOY_MAX_SEARCH_LEVEL" in i:
             if i["VTOY_MAX_SEARCH_LEVEL"] != "max":
-                searchlevel = int(i["VTOY_MAX_SEARCH_LEVEL"])
+                search_level = int(i["VTOY_MAX_SEARCH_LEVEL"])
 filelist = []
 
 
-def dfs(deep, dir):
+def dfs(deep, dir, search_level):
-    if deep > searchlevel:
+    if deep > search_level:
         return
     lists = os.listdir(DISK+dir)
     for i in lists:
@@ -55,30 +68,33 @@ def dfs(deep, dir):
             if ".iso" in i.lower() and KEYWORD in i.lower():
                 filelist.append(dir+"/"+i)
         else:
-            dfs(deep+1, dir+"/"+i)
+            dfs(deep+1, dir+"/"+i, search_level)
 
 
-dfs(0, searchroot)
+dfs(0, searchroot, search_level)
 
-if "injection" in vtoycfg:
+if "injection" in ventoy_cfg:
-    for i in vtoycfg["injection"]:
+    for i in ventoy_cfg["injection"]:
         if "image" in i:
             if i["image"] in filelist:
                 filelist.remove(i["image"])
 else:
-    vtoycfg["injection"] = []
+    ventoy_cfg["injection"] = []
-print("find iso:", filelist)
+
+print("Found iso:", filelist)
 for i in filelist:
-    vtoycfg["injection"].append({
+    ventoy_cfg["injection"].append({
         "image": i,
         "archive": "/PEinjector/PEinjector.7z"
     })
-print("write config")
+
+
+print("Writing config")
 with open(DISK+"/ventoy/ventoy.json", "w") as file:
-    file.write(json.dumps(vtoycfg))
+    file.write(json.dumps(ventoy_cfg))
 
 
-print("copy file")
+print("Copying files")
 if not os.path.exists(DISK+"/PEinjector"):
     os.mkdir(DISK+"/PEinjector")
 if os.path.exists(DISK+"/PEinjector/PEinjector.7z"):
@@ -88,14 +104,17 @@ if not os.path.exists(DISK+"/PEinjector/disable.txt"):
     with open(DISK+"/PEinjector/disable.txt", "w") as file:
         pass
 
-print("write version info")
+
+print("Writing version info")
 shutil.copyfile("version", DISK+"/PEinjector/VERSION")
 
-print("make dir")
+
+print("Making dir")
 if not os.path.exists(DISK+"/PEinjector/install"):
     os.mkdir(DISK+"/PEinjector/install")
 
-print("install basic package")
+
+print("Installing basic package")
 if not os.path.exists(DISK+"/PEinjector/package"):
     os.mkdir(DISK+"/PEinjector/package")
 for i in os.listdir("package"):
@@ -103,4 +122,5 @@ for i in os.listdir("package"):
         shutil.rmtree(DISK+"/PEinjector/package/"+i)
     shutil.copytree("package/"+i, DISK+"/PEinjector/package/"+i)
 
-print("done")
+
+print("All done.")