91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
import logging
|
||
import os
|
||
from datetime import datetime
|
||
from .config import get_instance as get_config
|
||
|
||
# 定义颜色代码
|
||
class Colors:
|
||
RED = '\033[31m'
|
||
GREEN = '\033[32m'
|
||
YELLOW = '\033[33m'
|
||
BLUE = '\033[34m'
|
||
MAGENTA = '\033[35m'
|
||
CYAN = '\033[36m'
|
||
WHITE = '\033[37m'
|
||
RESET = '\033[0m' # 重置颜色
|
||
|
||
# 自定义控制台处理器，为不同日志级别添加颜色
|
||
class ColoredConsoleHandler(logging.StreamHandler):
|
||
def emit(self, record):
|
||
# 为不同日志级别设置颜色
|
||
colors = {
|
||
logging.DEBUG: Colors.CYAN,
|
||
logging.INFO: Colors.GREEN,
|
||
logging.WARNING: Colors.YELLOW,
|
||
logging.ERROR: Colors.RED,
|
||
logging.CRITICAL: Colors.MAGENTA
|
||
}
|
||
|
||
# 获取对应级别的颜色，默认为白色
|
||
color = colors.get(record.levelno, Colors.WHITE)
|
||
|
||
# 获取原始消息
|
||
message = self.format(record)
|
||
|
||
# 添加颜色并输出
|
||
self.stream.write(f"{color}{message}{Colors.RESET}\n")
|
||
self.flush()
|
||
|
||
def init():
|
||
config = get_config()
|
||
log_dir = config.get("log", "log_dir", fallback="logs")
|
||
log_file = config.get("log", "log_file", fallback="if.u.service")
|
||
log_level = config.get("log", "log_level", fallback=logging.INFO)
|
||
console_log_level = config.get("log", "console_log_level", fallback=logging.DEBUG)
|
||
|
||
# 创建logs目录（如果不存在）
|
||
if not os.path.exists(log_dir):
|
||
os.makedirs(log_dir)
|
||
|
||
# 设置日志格式
|
||
log_format = "[%(asctime)s.%(msecs)03d][%(filename)s:%(lineno)d][%(levelname)s] %(message)s"
|
||
date_format = "%Y-%m-%d %H:%M:%S"
|
||
|
||
# 创建格式化器
|
||
formatter = logging.Formatter(log_format, datefmt=date_format)
|
||
|
||
# 获取根日志记录器
|
||
root_logger = logging.getLogger()
|
||
root_logger.setLevel(logging.NOTSET)
|
||
|
||
# 清除现有的处理器
|
||
root_logger.handlers.clear()
|
||
|
||
# 创建控制台处理器并设置颜色
|
||
console_handler = ColoredConsoleHandler()
|
||
console_handler.setFormatter(formatter)
|
||
console_handler.setLevel(console_log_level)
|
||
root_logger.addHandler(console_handler)
|
||
|
||
# 创建文件处理器
|
||
log_filename = os.path.join(log_dir, f"{log_file}_{datetime.now().strftime('%Y%m%d')}.log")
|
||
file_handler = logging.FileHandler(log_filename, encoding='utf-8')
|
||
file_handler.setFormatter(formatter)
|
||
file_handler.setLevel(log_level)
|
||
root_logger.addHandler(file_handler)
|
||
|
||
# 确保日志消息被正确处理
|
||
logging.addLevelName(logging.DEBUG, "D")
|
||
logging.addLevelName(logging.INFO, "I")
|
||
logging.addLevelName(logging.WARNING, "W")
|
||
logging.addLevelName(logging.ERROR, "E")
|
||
logging.addLevelName(logging.CRITICAL, "C")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
init(log_dir="logs", log_file="test", log_level=logging.INFO, console_log_level=logging.DEBUG)
|
||
logging.debug("debug log")
|
||
logging.info("info log")
|
||
logging.warning("warning log")
|
||
logging.error("error log")
|
||
logging.critical("critical log") |