如何通过-analyse参数启动analyse.py并捕获python program.py的输出？

我来给你梳理两个可行的方案，根据你是否能修改目标脚本（program.py）来选择：

方案一：可以修改目标脚本（最直接的方式） #

如果有权限修改program.py，我们可以在脚本入口处检测-analyse参数，一旦发现就捕获自身的输出，再传给analyse.py处理。

步骤1：修改program.py #

在program.py的开头添加参数检测和输出捕获逻辑：

import sys
from io import StringIO
import subprocess

def main():
    # 这里是你原本的program.py业务逻辑，比如生成输出
    print("用户登录成功")
    print("执行操作: 导出数据")
    print("操作完成，耗时0.5s")

if __name__ == "__main__":
    # 检查是否带有-analyse参数
    if "-analyse" in sys.argv:
        # 移除参数，避免影响原程序逻辑
        sys.argv.remove("-analyse")
        
        # 临时重定向标准输出，捕获程序输出
        old_stdout = sys.stdout
        output_buffer = StringIO()
        sys.stdout = output_buffer
        
        try:
            # 执行原程序逻辑
            main()
        finally:
            # 恢复标准输出
            sys.stdout = old_stdout
        
        # 获取捕获到的输出内容
        program_output = output_buffer.getvalue()
        
        # 调用analyse.py，将输出传入
        subprocess.run(
            ["python", "analyse.py"],
            input=program_output.encode("utf-8"),
            check=True
        )
    else:
        # 没有-analyse参数，正常执行原程序
        main()

步骤2：编写analyse.py #

这个脚本负责读取输入内容、执行分析并写入文件：

import sys

def analyse_content(output):
    # 这里写你的自定义分析逻辑，比如统计关键词、行数等
    line_count = len(output.splitlines())
    keyword_count = output.count("操作")
    return f"=== 分析结果 ===\n总输出行数: {line_count}\n'操作'关键词出现次数: {keyword_count}\n\n原始输出:\n{output}"

if __name__ == "__main__":
    # 从标准输入读取program.py的输出
    input_output = sys.stdin.read()
    
    # 执行分析
    analysis_result = analyse_content(input_output)
    
    # 将结果写入文件
    with open("analysis_report.txt", "w", encoding="utf-8") as f:
        f.write(analysis_result)
    
    print("✅ 分析完成，结果已保存到analysis_report.txt")

现在你运行python program.py -analyse，就能自动触发分析流程啦。

方案二：无法修改目标脚本（用Shell包装实现） #

如果不能改动program.py，我们可以通过Shell的别名/函数来实现“添加-analyse参数触发分析”的效果。

步骤1：配置Shell别名 #

在你的Bash/Zsh配置文件（比如~/.bashrc或~/.zshrc）中添加以下函数和别名：

analyse_wrapper() {
    local target_cmd=()
    local enable_analyse=0
    
    # 遍历所有参数，分离目标命令和-analyse标记
    for arg in "$@"; do
        if [ "$arg" = "-analyse" ]; then
            enable_analyse=1
        else
            target_cmd+=("$arg")
        fi
    done
    
    if [ $enable_analyse -eq 1 ]; then
        # 执行目标命令，并将输出管道到analyse.py
        "${target_cmd[@]}" 2>&1 | python /path/to/your/analyse.py
    else
        # 没有-analyse参数，直接执行原命令
        "${target_cmd[@]}"
    fi
}

# 将python命令替换为我们的包装函数
alias python=analyse_wrapper

配置完成后，运行source ~/.bashrc（或对应的配置文件）使生效。

步骤2：编写analyse.py #

和方案一中的analyse.py代码一致，它会从标准输入读取目标命令的输出（包括错误输出），分析后写入文件。

现在你运行python program.py -analyse，Shell会自动把它转换成python program.py 2>&1 | python analyse.py，完美实现你的需求。

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