sync ms

utils.py

@@ -1,7 +1,10 @@
 import os
 import shutil
+import rarfile
 import zipfile
 
+from py7zr import SevenZipFile
+
 EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
 TMP_DIR = "./__pycache__"
 
@@ -18,12 +21,22 @@ def clean_dir(dir_path: str):
     os.makedirs(dir_path)
 
 
-def unzip(zip_path: str, extract_to: str):
+def unzip(archive: str, extract_to: str):
     mk_dir(extract_to)
-    # 打开ZIP文件
-    with zipfile.ZipFile(zip_path, "r") as zip_ref:
-        # 解压文件
-        zip_ref.extractall(extract_to)
+    if archive.endswith(".zip"):
+        with zipfile.ZipFile(archive, "r") as f:
+            f.extractall(extract_to)
+
+    elif archive.endswith(".7z"):
+        with SevenZipFile(archive, "r") as f:
+            f.extractall(extract_to)
+
+    elif archive.endswith(".rar"):
+        with rarfile.RarFile(archive, "r") as f:
+            f.extractall(extract_to)
+
+    else:
+        raise ValueError("Unsupported file type!")
 
 
 def compress(folder_path: str, zip_file: str):
@@ -40,6 +53,5 @@ def compress(folder_path: str, zip_file: str):
                 relative_path = os.path.relpath(file_path, folder_path)
                 zipf.write(
                     file_path,
-                    arcname=os.path.join(os.path.basename(
-                        folder_path), relative_path),
+                    arcname=os.path.join(os.path.basename(folder_path), relative_path),
                 )