From 5702bbd0a1d219a6a4113e3617286fccd4054d7e Mon Sep 17 00:00:00 2001
From: 13339479676 <zyfiy1314@163.com>
Date: Wed, 19 Jan 2022 10:10:55 +0800
Subject: [PATCH] v04 zip mode

---
 README.md         |  2 ++
 opencv_webcam.py  | 11 ++++++++---
 utils/compress.py |  4 ++--
 v04.md            |  1 +
 4 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 50e1661..cf82d7a 100644
--- a/README.md
+++ b/README.md
@@ -268,6 +268,8 @@ python opencv_webcam.py -isasf -ic
 python opencv_webcam.py -isasf -ic -zn test02.zip
 # 自动命名压缩文件
 python opencv_webcam.py -isasf -ic -isazn
+# 自定义压缩模式，默认为写覆盖
+python opencv_webcam.py -isasf -ic -zn test03.zip -zm a # 追加模式
 ```
 
 
diff --git a/opencv_webcam.py b/opencv_webcam.py
index 2757138..e1e8623 100644
--- a/opencv_webcam.py
+++ b/opencv_webcam.py
@@ -69,6 +69,8 @@ def parse_args(known=False):
                         action='store_true', help='is auto zip name')
     parser.add_argument('--zipName', '-zn',
                         default="test.zip", type=str, help='zip save name')
+    parser.add_argument('--zipMode', '-zm',
+                        default="w", type=str, help='zip save mode')
     args = parser.parse_known_args()[0] if known else parser.parse_args()
     return args
 
@@ -95,7 +97,8 @@ def webcam_opencv(device_index="0",
                   logMode="a",
                   is_compress=False,
                   is_autoZipName=False,
-                  zipName="test.zip"):
+                  zipName="test.zip",
+                  zipMode="w"):
 
     keyList = [quit_key, frame_capKey, pause_key]  # 快捷键列表
     hotkey_judge(keyList)  # 快捷键冲突判断
@@ -190,7 +193,7 @@ def webcam_opencv(device_index="0",
     # ------------------压缩文件------------------
     if (is_compress):
         # 自定义压缩文件名称
-        webcam_zip(is_autoZipName, f'{ROOT_PATH}/{zipName}', frame_savePath)
+        webcam_zip(is_autoZipName, f'{ROOT_PATH}/{zipName}', frame_savePath, zipMode)
 
 
 def main(args):
@@ -220,6 +223,7 @@ def main(args):
     is_compress = args.is_compress
     is_autoZipName = args.is_autoZipName
     zipName = args.zipName
+    zipMode = args.zipMode
 
     argsYaml(args)  # 脚本参数
 
@@ -245,7 +249,8 @@ def main(args):
                   logMode,
                   is_compress,
                   is_autoZipName,
-                  zipName)
+                  zipName,
+                  zipMode)
 
 
 if __name__ == '__main__':
diff --git a/utils/compress.py b/utils/compress.py
index f9e35b7..f1754ac 100644
--- a/utils/compress.py
+++ b/utils/compress.py
@@ -18,13 +18,13 @@ def is_zipFile(zipName):
 
 
 # zip压缩
-def webcam_zip(is_autoZipName, zipName, preZipFilePath):
+def webcam_zip(is_autoZipName, zipName, preZipFilePath, zipMode):
     if (is_autoZipName):
         zipNameTmp = str(preZipFilePath).split('/')[-1]
         zipName = f'{ROOT_PATH}/{zipNameTmp}.zip'
     else:
         is_zipFile(zipName)
-    zip_file = zipfile.ZipFile(zipName, 'a')
+    zip_file = zipfile.ZipFile(zipName, zipMode)
 
     # ----------压缩开始----------
     file_list = os.listdir(preZipFilePath)
diff --git a/v04.md b/v04.md
index cd8474d..de2204a 100644
--- a/v04.md
+++ b/v04.md
@@ -11,6 +11,7 @@
 - 加入视频帧压缩功能
 - 加入自定义压缩文件名称
 - 加入自动命名压缩文件名称
+- 加入自定义压缩模式
 
 
 ### BUG修复