如何为Python中每个线程的print输出添加专属前缀？

你之前的代码失效核心原因是：sys.stdout是全局对象，多线程修改会互相覆盖；同时自定义流未处理print的换行机制、缓冲区刷新，以及线程局部状态隔离问题。以下是两种无需替换为logger的可行方案：

方案一：线程局部存储+全局stdout重写（适配所有stdout输出） #

通过threading.local隔离每个线程的前缀，自定义stdout流自动为当前线程的输出添加前缀：

import sys
import threading

# 线程局部存储，保存每个线程的专属前缀
thread_local = threading.local()

class ThreadPrefixStdout:
    def __init__(self, original_stdout):
        self.original_stdout = original_stdout

    def write(self, text):
        # 处理空文本（如print()单独输出的换行）
        if not text.strip():
            self.original_stdout.write(text)
            self.original_stdout.flush()
            return
        
        # 获取当前线程的前缀，默认[Unknown]
        prefix = getattr(thread_local, 'prefix', '[Unknown]')
        
        # 处理多行输出：换行后新行仍带前缀
        if '\n' in text:
            lines = text.split('\n')
            for line in lines[:-1]:
                self.original_stdout.write(f"{prefix} {line}\n")
            if lines[-1]:
                self.original_stdout.write(f"{prefix} {lines[-1]}")
        else:
            self.original_stdout.write(f"{prefix} {text}")
        
        self.original_stdout.flush()

    def flush(self):
        self.original_stdout.flush()

# 替换全局stdout
sys.stdout = ThreadPrefixStdout(sys.stdout)

def func1():
    # 设置当前线程的前缀
    thread_local.prefix = '[Foo]'
    print('Working foo')
    print('多行测试\n第二行内容')

def func2():
    thread_local.prefix = '[Bar]'
    print("Working bar")

if __name__ == '__main__':
    threading.Thread(target=func1).start()
    threading.Thread(target=func2).start()

方案二：全局包装print函数（仅针对print调用） #

直接替换全局print函数，无需修改原有代码中的print调用：

import builtins
import threading

thread_local = threading.local()
# 保存原始print函数
original_print = builtins.print

def threaded_print(*args, **kwargs):
    # 获取当前线程前缀
    prefix = getattr(thread_local, 'prefix', '[Unknown]')
    # 在原有输出前添加前缀
    original_print(f"{prefix}", *args, **kwargs)

# 替换全局print
builtins.print = threaded_print

def func1():
    thread_local.prefix = '[Foo]'
    print('Working foo')

def func2():
    thread_local.prefix = '[Bar]'
    print("Working bar")

if __name__ == '__main__':
    threading.Thread(target=func1).start()
    threading.Thread(target=func2).start()

原代码问题说明 #

1. 全局变量冲突：sys.stdout是全局对象，线程修改后会互相覆盖，导致最终输出使用最后修改的前缀。
2. 未处理print机制：print会分两次调用write（内容+换行符），原代码会给换行符也添加前缀，导致输出混乱。
3. 缺少flush逻辑：自定义流未实现flush方法，输出可能滞留在缓冲区无法及时显示。

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