Python 日志配置

代码实现

import logging
import sys

def setup_logging(log_file="./output.log", console_level=logging.INFO, file_level=logging.DEBUG):
    # 创建一个日志记录器
    logger = logging.getLogger()
    logger.setLevel(logging.DEBUG)

    # 创建一个文件处理器
    try:
        file_handler = logging.FileHandler(log_file, encoding="utf-8")
        file_handler.setLevel(file_level)
        file_formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(filename)s:%(lineno)d - %(message)s")
        file_handler.setFormatter(file_formatter)
        logger.addHandler(file_handler)
    except IOError:
        print(f"无法创建日志文件: {log_file}")

    # 创建控制台处理器
    console_handler = logging.StreamHandler(sys.stdout)
    console_handler.setLevel(console_level)
    console_formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")
    console_handler.setFormatter(console_formatter)
    logger.addHandler(console_handler)

    # 重定向 sys.stdout 和 sys.stderr
    sys.stdout = LoggerWriter(logger, logging.INFO)
    sys.stderr = LoggerWriter(logger, logging.ERROR)

class LoggerWriter:
    def __init__(self, logger, level):
        self.logger = logger
        self.level = level
        self.buffer = []

    def write(self, message):
        self.buffer.append(message)
        if message.endswith('\n'):
            self.flush()

    def flush(self):
        message = ''.join(self.buffer).strip()
        if message:
            self.logger.log(self.level, message)
        self.buffer = []

def main():
    setup_logging("output.log")
    
    print("这是一条通过 print 输出的消息")
    logging.info("这是一条 info 级别的日志消息")
    logging.error("这是一条 error 级别的日志消息")

if __name__ == "__main__":
    main()

使用

# 假设您的主程序文件名为 main.py

import logging
from log_config import setup_logging

def some_function():
    logging.info("这是一个来自 some_function 的信息")
    print("这是通过 print 输出的内容")

def main():
    # 在程序开始时设置日志
    setup_logging("my_app.log")

    # 使用日志记录信息
    logging.debug("这是一条调试信息")
    logging.info("程序开始运行")
    logging.warning("这是一条警告")

    # 调用使用日志的函数
    some_function()

    # 模拟一个错误
    try:
        1 / 0
    except ZeroDivisionError:
        logging.error("发生了除以零的错误", exc_info=True)

if __name__ == "__main__":
    main()