webcam zip 02

utils/compress.py

@@ -6,6 +6,8 @@ import zipfile
 import sys
 import os
 
+ROOT_PATH = sys.path[0]  # 项目根目录
+
 
 def is_zipFile(zipName):
     zipNameSuffix = zipName.split('.')[-1]
@@ -14,18 +16,16 @@ def is_zipFile(zipName):
         sys.exit()
 
 
-def webcam_zip():
-    zip_name = "./aaa.zip"
-    is_zipFile(zip_name)
+def webcam_zip(zipName, zipFilePath):
+    is_zipFile(zipName)
 
-    zip_file = zipfile.ZipFile(zip_name, 'a')
+    zip_file = zipfile.ZipFile(zipName, 'a')
 
-    # ----------压缩----------
-    file_list = os.listdir("./abc")
+    # ----------压缩开始----------
+    file_list = os.listdir(zipFilePath)
     for i in range(len(file_list)):
-        zip_file.write(f'./abc/{file_list[i]}',
-                    compress_type=zipfile.ZIP_DEFLATED)  # 压缩单个文件
-
-
+        zip_file.write(f'{zipFilePath}/{file_list[i]}',
+                       compress_type=zipfile.ZIP_DEFLATED)  # 压缩单个文件
+    # ----------压缩结束----------
     zip_file.close()
-    print(f'压缩完毕！')
+    print(f'压缩成功！')