如何将不同类的日志输出到不同文件？

不用修改现有代码里的logging.info调用，通过线程本地存储+自定义日志过滤器的方式，就能让PS1和PS2的日志分别输出到对应文件。

核心思路 #

1. 用线程本地存储（threading.local）记录当前执行代码所属的类标识
2. 自定义日志过滤器，根据线程本地的类标识，决定日志是否要被对应文件处理器（Handler）处理
3. 给根Logger添加两个分别对应PS1、PS2的Handler，每个Handler绑定专属过滤器
4. 在类的方法执行前后切换线程本地的类标识，确保日志被正确路由

具体实现代码 #

1. 导入依赖并定义线程本地存储 #

import logging
import threading

# 线程本地存储，用来存储当前执行代码的类标识
local_class = threading.local()

2. 自定义日志过滤器 #

过滤器会检查当前线程的类标识是否与自身匹配，匹配则允许日志通过：

class ClassSpecificFilter(logging.Filter):
    def __init__(self, target_class: str):
        super().__init__()
        self.target_class = target_class

    def filter(self, record: logging.LogRecord) -> bool:
        # 获取当前线程的类标识，匹配则放行日志
        return getattr(local_class, 'current', None) == self.target_class

3. 配置日志处理器与根Logger #

分别创建对应PS1.log、PS2.log的Handler，并绑定过滤器：

# 配置PS1日志处理器
ps1_handler = logging.FileHandler('PS1.log', encoding='utf-8')
ps1_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
ps1_handler.addFilter(ClassSpecificFilter('ps1'))

# 配置PS2日志处理器
ps2_handler = logging.FileHandler('PS2.log', encoding='utf-8')
ps2_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
ps2_handler.addFilter(ClassSpecificFilter('ps2'))

# 配置根Logger，添加两个处理器
root_logger = logging.getLogger()
root_logger.setLevel(logging.INFO)
root_logger.addHandler(ps1_handler)
root_logger.addHandler(ps2_handler)

4. 修改类的方法，添加标识切换逻辑 #

用上下文管理器封装标识切换，确保方法执行前后标识正确恢复：

class ClassLogContext:
    def __init__(self, class_id: str):
        self.class_id = class_id
        self.previous_id = None

    def __enter__(self):
        # 保存之前的类标识，避免干扰其他代码
        self.previous_id = getattr(local_class, 'current', None)
        local_class.current = self.class_id

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 恢复之前的类标识
        local_class.current = self.previous_id

修改PS1和PS2的do_sth方法：

class PS1:
    def __init__(self):
        pass

    def do_sth(self):
        with ClassLogContext('ps1'):
            logging.info('PS1 do sth')

class PS2:
    def __init__(self):
        pass

    def do_sth(self):
        with ClassLogContext('ps2'):
            logging.info('PS2 do sth')

测试验证 #

执行以下代码：

ps1 = PS1()
ps2 = PS2()

ps1.do_sth()
ps2.do_sth()

你会看到PS1.log中出现PS1 do sth，PS2.log中出现PS2 do sth，完全不用修改原来的logging.info调用。

注意事项 #

• 多线程环境下安全：线程本地存储的标识是每个线程独立的，不会出现日志串流问题
• 嵌套调用兼容：如果类方法中有嵌套的其他日志调用，上下文管理器会自动处理标识的切换与恢复
• 无侵入性：仅需修改类的方法逻辑，现有数百处logging.info完全不需要改动

内容的提问来源于stack exchange，提问作者OceanBlue