如何通过系统参数向Python的input()提示符传入响应以适配CI与单元测试场景

嘿，我之前做CI自动化的时候刚好遇到过一模一样的问题！不用改那个Python库的代码，这几个简便方法就能搞定input()的自动响应：

1. 用管道直接传递输入（最适合命令行快速调用） #

因为input()本质是从**标准输入(stdin)**读取内容，所以我们可以把API密钥通过管道直接“喂”给脚本。

如果是直接运行配置脚本：

echo "你的API密钥" | python 你的配置脚本.py

如果是用python -c的方式调用库函数：

echo "你的API密钥" | python -c 'import target_library; target_library.run_config_function()'

要是有多个配置项（比如需要依次输入密钥、用户名之类的），只要按input()的提示顺序把每个值用换行分隔就行：

printf "密钥1\n用户名\n" | python 你的配置脚本.py

2. 重定向标准输入到文件（适合多输入场景） #

把需要输入的所有内容按顺序写到一个临时文件里，然后让脚本从文件读取输入：

# 先把配置内容写入临时文件
cat << EOF > config_inputs.txt
你的API密钥
可选的其他配置项
EOF

# 让脚本从文件读取输入
python 你的配置脚本.py < config_inputs.txt

CI环境里可以用临时文件生成工具（比如mktemp）来避免文件残留，用完直接删掉就行。

3. 在单元测试中用subprocess模拟输入 #

如果是写单元测试要自动化这个过程，可以用Python的subprocess模块来控制子进程的输入输出：

import subprocess
import os

def test_auto_configure():
    # 调用库的配置函数
    proc = subprocess.Popen(
        ["python", "-c", "import target_library; target_library.configure()"],
        stdin=subprocess.PIPE,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        text=True  # 用文本模式，不用处理字节流
    )
    
    # 按input()的提示顺序传递输入，每个输入后加换行符
    input_data = "你的API密钥\n"
    stdout, stderr = proc.communicate(input=input_data)
    
    # 验证配置是否成功：检查返回码、生成的配置文件内容等
    assert proc.returncode == 0, f"配置失败，错误信息：{stderr}"
    assert os.path.exists("生成的配置文件路径")

这几个方法都不需要修改目标库的代码，完全适配CI和单元测试的自动化场景，亲测好用！

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