python - 使用生成器模拟输入

Question

下面的代码不模拟输入，有没有办法用生成器来做到这一点？

    inputlist = [5,1,2,3,4,5,6]

    def mock_input(dummy_prompt=">>> "):
        global inputlist
        for u in inputlist:
                yield str(u)

    __builtins__.input = mock_input

    l=[]
    x= input()
    for u in range(int(x)):
        l.append(int(input()))

    if l==inputlist[1:]:
        print("bravo")

错误是 x 是生成器而不是字符串。

我是否必须使用全局变量创建自己的生成器，或者我错过了语法中的某些内容。

Answer 1

好的，谢谢，我错过了生成器实例化的时刻。

def gg(l):
    for u in l:
            yield str(u)


def mockinput(thelist):
    bob = gg(thelist)
    __builtins__.input = lambda prompt="toto":  str(next(bob))


inputlist = [6,1,2,3,4,5,6]
mockinput(inputlist)

l=[]
for u in range(int(input())):
    l.append(int(input()))
    print(l)

if l==inputlist[1:]:
    print("bravo")

所以我需要分两步进行，因为我不需要任何全局变量