webcam compress update 03

opencv_webcam.py

@@ -70,9 +70,9 @@ def parse_args(known=False):
     parser.add_argument('--is_autoCompressName', '-isacn',
                         action='store_true', help='is auto compress name')
     parser.add_argument('--compressName', '-cn',
-                        default="test.zip", type=str, help='compress save name')
+                        default="test", type=str, help='compress save name')
     parser.add_argument('--compressMode', '-cm',
-                        default="w", type=str, help='compress save mode')
+                        default="w", type=str, help='compress save mode, tar w:gz')
     args = parser.parse_known_args()[0] if known else parser.parse_args()
     return args
 
@@ -100,7 +100,7 @@ def webcam_opencv(device_index="0",
                   is_compress=False,
                   compressStyle="zip",
                   is_autoCompressName=False,
-                  compressName="test.zip",
+                  compressName="test",
                   compressMode="w"):
 
     keyList = [quit_key, frame_capKey, pause_key]  # 快捷键列表
@@ -111,7 +111,7 @@ def webcam_opencv(device_index="0",
     logTime = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())  # 日志时间
     log_management(f'{logTime}\n', logName, logMode)  # 记录日志时间
 
-    is_compressFile(compressName) # 检测压缩文件格式
+    is_compressFile(compressStyle)  # 检测压缩文件格式
 
     time_list = [1, 60, 3600]  # 时间参数列表
 
@@ -201,7 +201,7 @@ def webcam_opencv(device_index="0",
         # webcam_zip(is_autoZipName, f'{ROOT_PATH}/{zipName}', frame_savePath, zipMode)
         # webcam_tar(is_autoZipName, f'{ROOT_PATH}/{zipName}', frame_savePath)
         webcam_compress(compressStyle, is_autoCompressName,
-                        f'{ROOT_PATH}/{compressName}', frame_savePath, compressMode)
+                        f'{ROOT_PATH}/{compressName}.{compressStyle}', frame_savePath, compressMode)
 
 
 def main(args):

utils/compress.py

@@ -27,11 +27,10 @@ def is_tarFile(tarName):
 
 
 # 判断压缩文件名称
-def is_compressFile(compressName):
+def is_compressFile(compressStyle):
     compressList = ['zip', 'tar']
-    compressNameSuffix = compressName.split('.')[-1]
-    if compressNameSuffix not in compressList:
-        print(f'{compressName}：格式不正确！程序退出！')
+    if compressStyle not in compressList:
+        print(f'{compressStyle}：不正确！程序退出！')
         sys.exit()
 
 
@@ -57,7 +56,7 @@ def webcam_zip(is_autoZipName, zipName, preZipFilePath, zipMode):
     print(f'压缩成功！已保存在：{zipName}')
 
 
-# zip压缩
+# tar压缩
 def webcam_tar(is_autoTarName, tarName, preTarFilePath, tarMode="w:gz"):
     if (is_autoTarName):
         # 自动命名
@@ -86,18 +85,12 @@ def webcam_compress(compressStyle, is_autoCompressName, compressName, preCompres
         # 自动命名
         compressNameTmp = str(preCompressFilePath).split('/')[-1]
         compressName = f'{ROOT_PATH}/{compressNameTmp}.{compressStyle}'
-    else:
-        is_compressFile(compressName)
 
     file_list = os.listdir(preCompressFilePath)  # 获取目录下的文件名称
 
     if (compressStyle == "zip"):
-        is_zipFile(compressName)
-        # 实例化zipfile对象
         compress_file = zipfile.ZipFile(compressName, compressMode)
     if (compressStyle == "tar"):
-        is_tarFile(compressName)
-        # 实例化tarfile对象
         compress_file = tarfile.open(compressName, compressMode)
 
     # ----------压缩开始----------