修改日志文件记录方式

watermark_generate/__init__.py

@@ -0,0 +1,27 @@
+import logging
+import sys
+from logging.handlers import RotatingFileHandler
+
+VERSION = '1.0.0'
+
+log_format = '%(asctime)s - %(levelname)s - [%(filename)s] - [%(funcName)s] - line:[%(lineno)d] - %(message)s'
+log_size = 20 * 1024 * 1024  # 默认为 1MB
+log_level = logging.INFO
+log_name = 'logs/watermark_generate.log'
+
+for arg in sys.argv:  # 跳过脚本名称（sys.argv[0]）
+    if arg.startswith('-debug'):
+        log_level = logging.DEBUG
+
+# 配置日志
+logging.basicConfig(level=log_level, format=log_format)
+
+# 获取默认的 logger
+logger = logging.getLogger()
+
+# 添加 RotatingFileHandler，设置日志文件大小限制
+handler = RotatingFileHandler(log_name, maxBytes=log_size, backupCount=2)
+handler.setFormatter(logging.Formatter(log_format))
+logger.addHandler(handler)
+for handler in logger.handlers:
+    handler.setLevel(logging.DEBUG)

watermark_generate/controller/watermark_generate_controller.py

@@ -5,10 +5,9 @@
 from flask import Blueprint, request, send_file, jsonify
 
 from watermark_generate.exceptions import BusinessException
-from watermark_generate.tools import logger_tool
+from watermark_generate import logger
 
 generator = Blueprint('generator', __name__)
-logger = logger_tool.logger
 
 # 允许的扩展名
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}

watermark_generate/logs/watermark_generate.log

@@ -0,0 +1,10 @@
+2024-07-31 16:17:08,596 - INFO - [_internal.py] - [_log] - line:[97] - WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.
+ * Running on all addresses (0.0.0.0)
+ * Running on http://127.0.0.1:5000
+ * Running on http://172.31.132.181:5000
+2024-07-31 16:17:08,598 - INFO - [_internal.py] - [_log] - line:[97] - Press CTRL+C to quit
+2024-07-31 16:17:43,685 - INFO - [_internal.py] - [_log] - line:[97] - WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.
+ * Running on all addresses (0.0.0.0)
+ * Running on http://127.0.0.1:5000
+ * Running on http://172.31.132.181:5000
+2024-07-31 16:17:43,686 - INFO - [_internal.py] - [_log] - line:[97] - Press CTRL+C to quit

watermark_generate/run.py

@@ -1,11 +1,5 @@
-import os
-import sys
-
 from watermark_generate.app import create_app
 
-rootpath = str(os.path.abspath(os.path.join(os.path.dirname(__file__), '../')))
-sys.path.append(rootpath)
-
 # 运行
 if __name__ == "__main__":
     app = create_app()

watermark_generate/tools/logger_tool.py

@@ -1,19 +0,0 @@
-# 设置初始的日志格式和大小
-import logging
-from logging.handlers import RotatingFileHandler
-
-log_format = '%(asctime)s - %(levelname)s - [%(filename)s] - [%(funcName)s] - line:[%(lineno)d] - %(message)s'
-log_size = 1024 * 1024  # 默认为 1MB
-log_level = logging.INFO
-# 配置日志
-logging.basicConfig(level=logging.DEBUG, format=log_format)
-
-# 获取默认的 logger
-logger = logging.getLogger(__name__)
-
-# 添加 RotatingFileHandler，设置日志文件大小限制
-handler = RotatingFileHandler('app.log', maxBytes=log_size, backupCount=1)
-handler.setFormatter(logging.Formatter(log_format))
-logger.addHandler(handler)
-for handler in logger.handlers:
-    handler.setLevel(log_level)