如何使用save_output函数保存所有函数的输出至文本文件？

我来给你捋清楚怎么实现这个需求哈，核心是要同时捕获指定函数的输出——既要在控制台正常打印所有函数的内容，又要把abc、help、please这三个函数的输出单独保存到文本文件里。下面是具体的实现步骤和可直接运行的代码示例：

1. 先实现save_output核心函数 #

这个函数要搞定两件事：一是捕获指定函数的输出，二是让这些输出同时显示在控制台和写入文件。我们可以通过自定义一个“双写”的输出流来实现：

import sys
from io import StringIO

def save_output(target_funcs, save_filename):
    # 先保存原始的控制台输出流，后面要恢复回去
    original_stdout = sys.stdout
    # 创建一个内存缓冲区用来临时存储要保存的输出
    captured_content = StringIO()

    try:
        # 定义一个双写类，既输出到控制台，又写入内存缓冲区
        class DualOutput:
            def write(self, data):
                original_stdout.write(data)  # 打印到控制台
                captured_content.write(data) # 存到内存里
        
        # 把系统输出替换成我们的双写对象
        sys.stdout = DualOutput()

        # 依次执行所有需要捕获输出的目标函数
        for func in target_funcs:
            func()
    
    finally:
        # 不管有没有异常，都要把系统输出恢复成原始的
        sys.stdout = original_stdout
        # 把内存里捕获的内容写入目标文件
        with open(save_filename, 'w', encoding='utf-8') as file:
            file.write(captured_content.getvalue())
        # 关闭内存缓冲区
        captured_content.close()

2. 定义你的业务函数 #

比如你提到的abc、help、please，还有其他需要打印但不需要保存的函数：

def abc():
    print("这是abc函数的输出内容")

def help():
    print("这是help函数的输出内容")

def please():
    print("这是please函数的输出内容")

# 示例：其他不需要保存的函数，只会在控制台打印
def other_function():
    print("这个函数的输出仅显示在控制台，不会保存到文件")

3. 编写main函数，统筹执行逻辑 #

在main里，我们可以自由控制哪些函数直接打印，哪些函数的输出需要被保存：

def main():
    print("=== 开始执行所有函数 ===")
    
    # 先执行不需要保存的函数，直接打印到控制台
    other_function()
    
    # 调用save_output，把abc、help、please的输出保存到文件，同时打印到控制台
    save_output([abc, help, please], "function_outputs.txt")
    
    print("=== 所有函数执行完成 ===")

if __name__ == "__main__":
    main()

额外说明 #

• 如果你的目标函数需要传参数，可以用lambda来包装，比如要给abc传参数的话，写成lambda: abc("自定义参数")放进target_funcs列表里就行。
• 运行后，控制台会显示所有函数的输出，同时当前目录下会生成function_outputs.txt文件，里面只包含abc、help、please三个函数的输出内容。

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