当没有跳过输入的选项时，如何绕过 Python3 中的输入提示？

Question

使用一个库，当我调用某个函数时，我将调用 foo() ，系统会提示我输入这样的内容：

def foo():
   if input("Proceed?") == 'yes':
        take_action()

我需要传递答案才能继续。

但我希望能够使用默认值循环 foo() 来提供提示。

问题是开发人员没有提供为提示提供默认值的选项。

理想情况下，他们会这样编写 foo() ：

def foo(default_response=None):
    if default_response == 'yes':
        take_action()
    elif input("Proceed?") == 'yes':
        take_action()

鉴于他们没有提供默认响应的选项，有没有办法让我循环 foo()并在不更改源代码的情况下自动提供输入？

Answer 1

一种解决方案是monkey patch库以临时替换输入功能。例如，如果我有如下所示的模块 foo.py：

def take_action():
    print("doing something")


def foo():
    if input("Proceed? ") == "yes":
        take_action()

我可以这样编写 bar.py：

import foo
from unittest import mock


def fake_input(default_response=None):
    '''Creates a replacement for the `input` function that will
    return a default response if one was provided.'''

    def _input(prompt):
        return default_response if default_response else input(prompt)

    return _input


with mock.patch("foo.input", new=fake_input("yes")):
    foo.foo()

foo.foo()

如果运行此代码，您将看到第一次调用 foo.foo()
使用默认输入，并在没有提示的情况下继续。第二次通话
to foo.foo() 会正常提示。