2024-07-07 01:58:23 +08:00
|
|
|
|
import logging
|
2024-07-14 16:52:30 +08:00
|
|
|
|
import os
|
2024-07-07 01:58:23 +08:00
|
|
|
|
from datetime import datetime
|
|
|
|
|
|
|
|
|
|
|
|
|
2024-08-25 20:29:24 +08:00
|
|
|
|
DATETIME = datetime.now().strftime('%Y-%m-%d-%H%M%S')
|
|
|
|
|
|
# DATETIME = "debug" # 为了方便,调试的时候输出到 debug.log 文件
|
2024-09-09 17:07:03 +08:00
|
|
|
|
LOG_FILE = f'saves/log/project-{DATETIME}.log'
|
2024-07-07 01:58:23 +08:00
|
|
|
|
|
2025-03-03 22:06:28 +08:00
|
|
|
|
def setup_logger(name, level=logging.DEBUG, console=True):
|
2024-09-09 17:07:03 +08:00
|
|
|
|
os.makedirs("saves/log", exist_ok=True)
|
2024-07-07 01:58:23 +08:00
|
|
|
|
|
|
|
|
|
|
"""Function to setup logger with the given name and log file."""
|
|
|
|
|
|
logger = logging.getLogger(name)
|
|
|
|
|
|
logger.setLevel(level)
|
|
|
|
|
|
|
2024-08-25 20:29:24 +08:00
|
|
|
|
# 清除已有的 Handler,防止重复添加
|
|
|
|
|
|
if logger.hasHandlers():
|
|
|
|
|
|
logger.handlers.clear()
|
|
|
|
|
|
|
2024-07-07 01:58:23 +08:00
|
|
|
|
# File handler for logging to a file
|
2024-08-25 20:29:24 +08:00
|
|
|
|
file_handler = logging.FileHandler(LOG_FILE)
|
2024-07-07 01:58:23 +08:00
|
|
|
|
file_handler.setLevel(level)
|
|
|
|
|
|
|
|
|
|
|
|
# Formatter for the logs
|
2024-07-16 18:14:27 +08:00
|
|
|
|
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(name)s - %(message)s')
|
2024-07-07 01:58:23 +08:00
|
|
|
|
file_handler.setFormatter(formatter)
|
|
|
|
|
|
logger.addHandler(file_handler)
|
|
|
|
|
|
|
|
|
|
|
|
# Console handler for logging to the console (optional)
|
|
|
|
|
|
if console:
|
|
|
|
|
|
console_handler = logging.StreamHandler()
|
|
|
|
|
|
console_handler.setLevel(level)
|
|
|
|
|
|
console_handler.setFormatter(formatter)
|
|
|
|
|
|
logger.addHandler(console_handler)
|
|
|
|
|
|
|
|
|
|
|
|
return logger
|
|
|
|
|
|
|
2024-08-25 20:29:24 +08:00
|
|
|
|
|
2024-07-07 01:58:23 +08:00
|
|
|
|
# Setup the root logger
|
2024-09-14 02:45:35 +08:00
|
|
|
|
logger = setup_logger('Yuxi')
|
2024-07-07 01:58:23 +08:00
|
|
|
|
|
|
|
|
|
|
# If you want to disable logging from external libraries
|
|
|
|
|
|
# logging.getLogger('some_external_library').setLevel(logging.CRITICAL)
|