如何在不修改原模块的情况下自动回复Python模块的y/n输入？

不需要修改原模块代码的前提下，有两种可靠的实现方式，优先推荐第一种：

方法一：使用unittest.mock.patch临时替换input函数 #

这是最简洁且安全的方案，通过临时替换全局的input函数，让它始终返回'y'，作用域结束后自动恢复原函数，不会影响其他代码逻辑。

代码示例：

from unittest.mock import patch
import target_module  # 替换为你实际要导入的模块名称

# 在with块内，所有input调用都会返回'y'
with patch('builtins.input', return_value='y'):
    target_module.target_function()  # 调用模块中需要确认的目标函数

方法二：修改标准输入流sys.stdin #

通过替换标准输入流为预先写入内容的内存流，让input从该流中读取预设的'y'。这种方式适合需要批量模拟多个输入的场景，但用完后记得恢复原输入流。

代码示例：

import sys
from io import StringIO
import target_module

# 将标准输入替换为包含'y'的内存流，注意末尾要加换行符'\n'
sys.stdin = StringIO('y\n')
target_module.target_function()

# 恢复原标准输入，避免影响后续代码的输入操作
sys.stdin = sys.__stdin__

方案对比 #

• mock.patch方式：作用域明确，自动恢复，不会污染全局环境，适合大多数场景，优先推荐。
• 修改sys.stdin方式：更底层，适合复杂输入模拟，但需要手动恢复，容易遗漏导致后续输入异常。

内容的提问来源于stack exchange，提问作者Emilio M Bumachar