'commit'

Apps/YeLiTe/Kit.py

@@ -269,3 +269,34 @@ def draw_rectangles(image_path, points, output_path=None):
     except Exception as e:
         logger.error(f"绘制矩形框失败: {e}")
         return image_path
+
+def setup_logger(name, log_file=None):
+    """
+    配置日志，支持同时输出到控制台和文件
+    """
+    if log_file is None:
+        log_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Logs")
+        if not os.path.exists(log_dir):
+            os.makedirs(log_dir)
+        log_file = os.path.join(log_dir, f"{name}_{time.strftime('%Y%m%d')}.log")
+
+    logger = logging.getLogger(name)
+    logger.setLevel(logging.INFO)
+
+    # 如果已经有 handler，不再添加
+    if logger.handlers:
+        return logger
+
+    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+
+    # 控制台 Handler
+    ch = logging.StreamHandler()
+    ch.setFormatter(formatter)
+    logger.addHandler(ch)
+
+    # 文件 Handler
+    fh = logging.FileHandler(log_file, encoding='utf-8')
+    fh.setFormatter(formatter)
+    logger.addHandler(fh)
+
+    return logger