如何在Python IDLE中将程序最后输出保存至文本文件？

需求说明 #

• 在Python IDLE中运行指定的龙穴游戏程序
• 自动将程序最后一次运行的输出保存到文本文件，禁止手动操作
• 运行过程中需保持控制台正常输出，方便玩家交互

现有方法无效的原因 #

你尝试的两种方法存在以下问题：

1. io.StringIO重定向方法：仅捕获了函数内的输出，未适配游戏的循环交互逻辑，无法区分最后一轮运行的输出，且会导致控制台暂时看不到输出
2. 直接重定向sys.stdout到文件：会让控制台完全看不到输出，破坏游戏的交互体验，同样无法精准捕获最后一轮的内容

适配的解决方案 #

以下是修改后的完整游戏代码，整合了输出捕获与最后一次运行结果保存的逻辑，同时保留控制台输出：

import random
import time
import sys
import io

class DualOutput(io.StringIO):
    """自定义输出流，同时输出到控制台和内存缓冲区"""
    def write(self, data):
        # 先输出到控制台
        sys.__stdout__.write(data)
        # 再写入内存缓冲区
        super().write(data)

def play_round():
    """单轮游戏逻辑，返回本轮的输出内容"""
    output_stream = DualOutput()
    # 重定向stdout到自定义流
    original_stdout = sys.stdout
    sys.stdout = output_stream
    
    try:
        # 游戏介绍
        print("""                                            You are in the Kingdom of Dragons. In front of you, you see two caves. In one
                                                cave, the dragon is friendly and will share his treasure with you. The other
                                                               dragon is hungry and will eat you on sight!
                                                               
""")

        # 让玩家选择洞穴
        while True:
            select = input("cave 1,2,3,4 or 5 : ")
            if select in {"1", "2", "3", "4", "5"}:
                break

        # 游戏过程输出
        print("\nYou approach the cave...", select)
        time.sleep(2)
        print("A large dragon jumps out in front of you!")
        time.sleep(1.5)
        print("He opens his jaws and ...")

        time.sleep(3)

        # 随机结果
        choices = ["Gobbles you down!", "Greets you before share his treasure"]
        print("\n", random.choice(choices))

        time.sleep(2)
        # 返回本轮输出内容
        return output_stream.getvalue()
    finally:
        # 恢复原始stdout
        sys.stdout = original_stdout

# 主循环
last_round_output = ""
while True:
    last_round_output = play_round()
    again = input("\nDo you wanna play again?(y/n)")
    if again.lower() != "y":
        # 保存最后一轮输出到文件
        with open("last_game_output.txt", "w", encoding="utf-8") as f:
            f.write(last_round_output)
        break

代码说明 #

1. DualOutput类：实现同时向控制台和内存缓冲区输出内容，既保证玩家能正常看到游戏过程，又能捕获每一轮的输出
2. play_round函数：封装单轮游戏逻辑，捕获本轮所有输出并返回
3. 主循环：每次运行单轮游戏后记录输出，当玩家选择退出时，将最后一轮的输出写入last_game_output.txt文件

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