如何实现全局可用函数？toprint装饰器跨模块失效问题求助

Python中每个模块都拥有独立的全局命名空间，你编写的toprint装饰器仅替换了当前模块（main.py）内的print函数，但其他模块（比如test.py）中的函数引用的是该模块自身命名空间里的内置print，因此跨模块使用时装饰器失效。

方案一：替换全局内置print函数 #

直接修改Python的builtins模块中的print，这样所有模块的print调用都会指向自定义实现，装饰器执行完后再恢复原始函数：

import builtins

def toprint(func):
    def wrapper(*args, **kwargs):
        # 保存原始内置print
        original_print = builtins.print
        
        def custom_print(*args, end="\n", sep=" "):
            args = [str(i) for i in args]
            target = sep.join(args)
            with open("output", "a") as file:
                file.write(target + end)
        
        # 替换全局内置print
        builtins.print = custom_print
        try:
            # 执行目标函数
            func(*args, **kwargs)
        finally:
            # 无论是否报错都恢复原始print
            builtins.print = original_print
    return wrapper

方案二：重定向sys.stdout #

print默认输出到sys.stdout，我们可以临时替换sys.stdout为自定义的文件类对象，实现所有控制台输出的重定向：

import sys

class FileRedirector:
    def __init__(self, filename):
        self.filename = filename
        self.original_stdout = sys.stdout
    
    def write(self, text):
        with open(self.filename, "a") as f:
            f.write(text)
    
    def flush(self):
        # 兼容stdout的flush方法，避免报错
        pass

def toprint(func):
    def wrapper(*args, **kwargs):
        redirector = FileRedirector("output")
        sys.stdout = redirector
        try:
            func(*args, **kwargs)
        finally:
            # 恢复原始stdout
            sys.stdout = redirector.original_stdout
    return wrapper

• 方案一：仅替换print函数，不会影响其他直接操作sys.stdout的代码，针对性更强。
• 方案二：覆盖所有输出到sys.stdout的内容，适合需要捕获全部控制台输出的场景。

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