Python之代码片段-日志功能

import logging
import os
from logging.handlers import TimedRotatingFileHandler


def setup_logger(log_path: str, file_name: str = "server.log"):
    # 创建 Logger 实例
    log = logging.getLogger('server-logger')
    log.setLevel(logging.DEBUG)  # 设置全局最低日志级别
    # 避免重复添加 Handler（防止重复日志）
    if log.handlers:
        return log
    # 定义日志格式
    formatter = logging.Formatter(
        '%(asctime)s %(levelname)s thread[%(thread)d] %(module)s.%(funcName)s:%(lineno)d -> %(message)s'
    )
    # 控制台 Handler（输出到终端）
    console_handler = logging.StreamHandler()
    console_handler.setLevel(logging.INFO)  # 控制台只输出 INFO 及以上级别
    console_handler.setFormatter(formatter)
    log.addHandler(console_handler)
    # 按日期切割的 File Handler（输出到文件）
    filename = f"{log_path}/{file_name}"
    file_handler = TimedRotatingFileHandler(
        filename=filename,
        when='midnight',  # 每天午夜切割
        interval=1,  # 间隔1天
        backupCount=7,  # 保留7天日志
        encoding='utf-8'
    )
    file_handler.setLevel(logging.INFO)  # 文件记录所有 DEBUG 及以上级别
    file_handler.setFormatter(formatter)
    log.addHandler(file_handler)
    return log


# 获取当前目录的上一级目录下，的logs文件夹路径
config_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "logs")
# 查询日志目录是否存在，不存在创建
os.makedirs(config_path, exist_ok=True)
logger = setup_logger(config_path, "server.log")


if __name__ == '__main__':
    logger.debug('This is a debug message')
    logger.info('This is an info message')
    logger.warning('This is a warning message')
    logger.error('This is an error message')

from manager.LoggerManager import logger

if __name__ == '__main__':
    logger.debug('This is a debug message')
    logger.info('This is an info message')
    logger.warning('This is a warning message')
    logger.error('This is an error message')