webcam compress update

opencv_webcam.py

@@ -15,7 +15,7 @@ from utils.hotkey import hotkey_judge
 from utils.frame_opt import frame_opt
 from utils.log import is_logSuffix, log_management
 from utils.args_yaml import argsYaml
-from utils.compress import webcam_zip, webcam_tar
+from utils.compress import webcam_zip, webcam_tar, webcam_compress
 
 ROOT_PATH = sys.path[0]  # 项目根目录
 

utils/compress.py

@@ -26,6 +26,14 @@ def is_tarFile(tarName):
         sys.exit()
 
 
+# 判断压缩文件名称
+def is_compressFile(compressName):
+    compressNameSuffix = compressName.split('.')[-1]
+    if compressNameSuffix != "zip" or "tar":
+        print(f'{compressName}：格式不正确！程序退出！')
+        sys.exit()
+
+
 # zip压缩
 def webcam_zip(is_autoZipName, zipName, preZipFilePath, zipMode):
     if (is_autoZipName):
@@ -69,3 +77,31 @@ def webcam_tar(is_autoTarName, tarName, preTarFilePath, tarMode="w:gz"):
     # ----------压缩结束----------
     tar_file.close()
     print(f'压缩成功！已保存在：{tarName}')
+
+
+# webcam压缩
+def webcam_compress(compressStyle, is_autoCompressName, compressName, preCompressFilePath, compressMode):
+    if (is_autoCompressName):
+        # 自动命名
+        compressNameTmp = str(preCompressFilePath).split('/')[-1]
+        compressName = f'{ROOT_PATH}/{compressNameTmp}.{compressStyle}'
+
+    file_list = os.listdir(preCompressFilePath)  # 获取目录下的文件名称
+
+    if (compressStyle == "zip"):
+        # 实例化zipfile对象
+        compress_file = zipfile.ZipFile(compressName, compressMode)
+    if (compressStyle == "tar"):
+        # 实例化tarfile对象
+        compress_file = tarfile.open(compressName, compressMode)
+
+    # ----------压缩开始----------
+    for i in range(len(file_list)):
+        if (compressStyle == "zip"):
+            compress_file.write(
+                f'{preCompressFilePath}/{file_list[i]}', compress_type=zipfile.ZIP_DEFLATED)
+        if (compressStyle == "tar"):
+            compress_file.add(f'{preCompressFilePath}/{file_list[i]}')
+    # ----------压缩结束----------
+    compress_file.close()
+    print(f'文件压缩成功！已保存在：{compressName}')