update get data code

PyCode/LogHelper.py

@@ -0,0 +1,105 @@
+# import logging
+# import sys
+
+# class LogHelper:
+#     def __init__(self, 
+#                  level=logging.INFO, 
+#                  format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+#                  handlers=None):
+#         """
+#         初始化日志配置
+        
+#         :param level: 日志级别，默认为 logging.INFO
+#         :param format: 日志格式字符串
+#         :param handlers: 日志处理器列表，默认为空（会自动添加控制台处理器）
+#         """
+#         self.level = level
+#         self.format = format
+#         self.handlers = handlers if handlers is not None else []
+
+#     def add_console_handler(self, stream=sys.stdout, encoding='utf-8'):
+#         """添加控制台处理器"""
+#         console_handler = logging.StreamHandler(stream)
+#         console_handler.setFormatter(logging.Formatter(self.format))
+#         # 设置控制台编码
+#         if encoding:
+#             console_handler.encoding = encoding
+#         self.handlers.append(console_handler)
+
+#     def add_file_handler(self, filename, encoding='utf-8'):
+#         """添加文件处理器（解决中文乱码问题）"""
+#         # 使用支持UTF-8编码的文件处理器
+#         filepath = "log/" + filename
+#         file_handler = logging.FileHandler(filepath, encoding=encoding)
+#         file_handler.setFormatter(logging.Formatter(self.format))
+#         self.handlers.append(file_handler)
+
+#     def setup(self):
+#         """应用日志配置"""
+#         logging.basicConfig(
+#             level=self.level,
+#             format=self.format,
+#             handlers=self.handlers
+#         )
+
+
+import logging
+import sys
+import os
+from datetime import datetime
+
+class LogHelper:
+    def __init__(self, 
+                 level=logging.INFO, 
+                 format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+                 handlers=None,
+                 log_dir="logs"):
+        """
+        初始化日志配置
+        
+        :param level: 日志级别，默认为 logging.INFO
+        :param format: 日志格式字符串
+        :param handlers: 日志处理器列表，默认为空（会自动添加控制台处理器）
+        :param log_dir: 日志存储目录，默认为"logs"
+        """
+        self.level = level
+        self.format = format
+        self.handlers = handlers if handlers is not None else []
+        self.log_dir = log_dir
+        # 确保日志目录存在
+        os.makedirs(self.log_dir, exist_ok=True)
+
+    def add_console_handler(self, stream=sys.stdout, encoding='utf-8'):
+        """添加控制台处理器"""
+        console_handler = logging.StreamHandler(stream)
+        console_handler.setFormatter(logging.Formatter(self.format))
+        if encoding:
+            console_handler.encoding = encoding
+        self.handlers.append(console_handler)
+
+    def _get_daily_log_path(self):
+        """生成基于当前日期的日志文件路径"""
+        today = datetime.now().strftime("%Y-%m-%d")
+        filename = f"{today}.log"
+        return os.path.join(self.log_dir, filename)
+
+    def _add_daily_file_handler(self, encoding='utf-8'):
+        """添加按日期自动创建的文件处理器"""
+        log_path = self._get_daily_log_path()
+        file_handler = logging.FileHandler(log_path, encoding=encoding)
+        file_handler.setFormatter(logging.Formatter(self.format))
+        self.handlers.append(file_handler)
+
+    def setup(self):
+        """应用日志配置（自动添加日期文件处理器）"""
+        # 如果未提供任何处理器，默认添加控制台和日期文件处理器
+        if not self.handlers:
+            self.add_console_handler()
+            self._add_daily_file_handler()
+        
+        # 应用配置
+        logging.basicConfig(
+            level=self.level,
+            format=self.format,
+            handlers=self.handlers
+        )