From ef873765058c3441ee62991089f51f7e9fb33c47 Mon Sep 17 00:00:00 2001
From: 13339479676 <zyfiy1314@163.com>
Date: Sat, 15 Jan 2022 15:37:36 +0800
Subject: [PATCH] log update

---
 opencv_webcam.py | 3 ++-
 utils/log.py     | 6 ++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/opencv_webcam.py b/opencv_webcam.py
index 7287796..6c97e1c 100644
--- a/opencv_webcam.py
+++ b/opencv_webcam.py
@@ -12,7 +12,7 @@ import sys
 from utils.ows_path import increment_path
 from utils.hotkeyjudge import hotkey_judge
 from utils.frame_opt import frame_opt
-from utils.log import log_management
+from utils.log import is_logSuffix, log_management
 
 
 def parse_args(known=False):
@@ -70,6 +70,7 @@ def webcam_opencv(device_index=0, quit_key="q", pause_key="p", is_autoSaveFrame=
     hotkey_judge(keyList)  # 快捷键冲突判断
 
     # 日志文件
+    is_logSuffix(logName) # 检测日志格式
     logTime = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())  # 日志时间
     log_management(f'{logTime}\n', logName, logMode)  # 记录日志时间
     # ------------------程序开始------------------
diff --git a/utils/log.py b/utils/log.py
index e0bdbdf..3e4d7a7 100644
--- a/utils/log.py
+++ b/utils/log.py
@@ -5,8 +5,7 @@
 import sys
 
 
-# 日志管理
-def log_management(logContent, logName, logSaveMode):
+def is_logSuffix(logName):
     suffixList = ['log', 'txt', 'data']  # 日志后缀列表
     logSuffixFlag = 0  # 日志后缀标志
     for i in range(len(suffixList)):
@@ -16,6 +15,9 @@ def log_management(logContent, logName, logSaveMode):
     if logSuffixFlag == 0:
         print(f'日志格式不正确！程序结束！')
         sys.exit()  # 结束程序
+
+# 日志管理
+def log_management(logContent, logName, logSaveMode):
     ROOT_PATH = sys.path[0]  # 项目根目录
     logFile = open(f'{ROOT_PATH}/{logName}', logSaveMode)  # 日志文件
     logFile.write(logContent)  # 日志写入