This commit is contained in:
HuangHai
2026-01-14 07:28:25 +08:00
parent af60249fce
commit 66714aab8b

View File

@@ -196,12 +196,17 @@ def clear_temp_dir(save_dir=None):
def setup_logger(name, log_file=None):
"""
配置日志,支持同时输出到控制台和文件
统一存放在根目录的 Logs 文件夹下
"""
if log_file is None:
log_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Logs")
# 获取项目根目录 (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"{name}_{time.strftime('%Y%m%d')}.log")
# 统一使用供应商代号作为文件名
supplier_code = os.path.basename(os.path.dirname(os.path.abspath(__file__)))
log_file = os.path.join(log_dir, f"{supplier_code}.log")
logger = logging.getLogger(name)
logger.setLevel(logging.INFO)