如何在不修改外部Python脚本的情况下捕获其print输出并终止循环？

完全可行，无需修改天文脚本本身，通过捕获标准输出并解析内容即可实现需求。以下是具体实现方法：

核心思路 #

利用Python的contextlib.redirect_stdout上下文管理器，临时将function_1的print输出捕获到内存对象中，再从捕获的文本里提取星系数量，判断是否终止循环。

完整代码示例 #

from astro_script_file import some_astro_class
from contextlib import redirect_stdout
import io

# 待处理的文件列表
file_sequence = ['file_1.txt', 'file_2.txt', 'file_n.txt']

for file_name in file_sequence:
    # 创建内存IO对象用于捕获输出
    output_buffer = io.StringIO()
    # 临时重定向标准输出到buffer
    with redirect_stdout(output_buffer):
        some_astro_class.function_1(file_name)
    
    # 获取捕获的输出内容（去除首尾空白）
    output_content = output_buffer.getvalue().strip()
    # 可选：将输出打印到屏幕（保留原脚本的打印行为）
    print(output_content)
    
    # 解析星系数量
    if 'number of populated galaxies:' in output_content:
        try:
            # 分割字符串提取数字并转换为整数
            galaxy_num = int(output_content.split(':')[-1].strip())
            if galaxy_num == 0:
                print("检测到星系数量为0，终止循环")
                break
        except ValueError:
            # 防止输出格式异常导致报错，继续下一个文件
            print("输出格式异常，无法解析星系数量，继续执行")
            continue

关键细节说明 #

• redirect_stdout是安全的临时重定向方式，退出with代码块后，标准输出会自动恢复原状，不会影响其他代码的打印行为。
• 加入try-except块是为了兼容输出格式异常的情况，避免脚本直接崩溃。
• 如果不需要在屏幕上保留原打印内容，可以去掉print(output_content)这一行。

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