'commit'
This commit is contained in:
@@ -246,11 +246,14 @@ def detect_rabbit_ad_close(image_path, debug_dir=None):
|
||||
|
||||
return [best[3], best[4]]
|
||||
|
||||
def setup_logger(name, log_file=None):
|
||||
def setup_logger(name, log_file=None, clear_old_log=False):
|
||||
"""
|
||||
配置日志,支持同时输出到控制台和文件。
|
||||
使用供应商代号作为父级 Logger,所有子 Logger 继承其 Handler,
|
||||
并通过 propagate=False 避免与根 Logger 重复。
|
||||
:param name: Logger 名称
|
||||
:param log_file: 指定日志文件路径,如果不指定则使用默认路径
|
||||
:param clear_old_log: 是否在启动时清空旧日志文件
|
||||
"""
|
||||
# 1. 获取供应商代号 (如 XinDianTu)
|
||||
supplier_code = os.path.basename(os.path.dirname(os.path.abspath(__file__)))
|
||||
@@ -260,15 +263,26 @@ def setup_logger(name, log_file=None):
|
||||
parent_logger.setLevel(logging.INFO)
|
||||
parent_logger.propagate = False # 禁止向上传递给 root logger,防止重复
|
||||
|
||||
if not parent_logger.handlers:
|
||||
if log_file is None:
|
||||
# 获取项目根目录 (aiData)
|
||||
root_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
log_dir = os.path.join(root_dir, "Logs")
|
||||
if not os.path.exists(log_dir):
|
||||
os.makedirs(log_dir)
|
||||
log_file = os.path.join(log_dir, f"{supplier_code}.log")
|
||||
if log_file is None:
|
||||
# 获取项目根目录 (aiData)
|
||||
root_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
log_dir = os.path.join(root_dir, "Logs")
|
||||
if not os.path.exists(log_dir):
|
||||
os.makedirs(log_dir)
|
||||
log_file = os.path.join(log_dir, f"{supplier_code}.log")
|
||||
|
||||
# 如果需要清空旧日志且文件存在
|
||||
if clear_old_log and os.path.exists(log_file):
|
||||
try:
|
||||
# 关闭现有的 handler 以便删除文件
|
||||
for handler in parent_logger.handlers[:]:
|
||||
handler.close()
|
||||
parent_logger.removeHandler(handler)
|
||||
os.remove(log_file)
|
||||
except Exception as e:
|
||||
print(f"无法清空旧日志文件 {log_file}: {e}")
|
||||
|
||||
if not parent_logger.handlers:
|
||||
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||||
|
||||
# 控制台 Handler
|
||||
|
||||
Reference in New Issue
Block a user