Python 生成器发送：发送后不要产生新值

Question

这是一个

奇怪的问题，所以我会解释一下：

我有一个这样的生成器，它充当 IRC 服务器的生成器前端：

def irc_iter(): # not the real code, simplified
    msgs = get_msgs()
    for msg in msgs:
        if is_ping(msg):
            pong()
        else:
            to_send = yield msg
            for s in to_send:
                send(s)

从理论上讲，这应该允许我做一些很酷的事情，比如：

server = connect()
for line in server:
       if should_respond(line):
           server.send('WOW SUCH MESSAGE')

但是，有一个障碍：generator.send也会产生下一个值。这意味着server.send也在给我下一条消息......我更愿意像所有其他消息一样处理，生成为 line .

我知道我可以用一种丑陋的方式解决这个问题，只需在收到发送后产生一个垃圾值，但我试图保持我的代码优雅，事实恰恰相反。有没有办法告诉生成器我还不想要新值？

谢谢。

Answer 1

我也刚刚遇到了这个问题，也没有找到比虚拟yield更好的东西......

所以在你的生成器代码中：

        # Input
        input_value = yield
        if input_value is None:
            # Handle situation if someone used your generator in a wrong way
            sys.exit(-1)
        yield "DUMMY YIELD JUST STOP HERE!!!"

在客户端代码中：

while True:
    values = next(my_generator)
    ...
    if values == 'INPUT_NEEDED':
        # At this point you realize it's input time
        next(my_generator) # Rewind to the input point
        my_generator.send(12345) # Returns "DUMMY YIELD JUST STOP HERE!!!"
        continue
    # Continue the main loop

Answer 2

看起来问题来自每次迭代两次调用生成器，一次使用 .send(None)（在 for 循环中），一次使用 .send(response) .

如果 for 循环可以迭代 .send（） 而不是 .next（），那将是一个简单的修复，但我不熟悉任何让它工作的方法（pep342 中的可选扩展继续语句？）而不将其包装在另一个生成器中（可能使用队列将值推送到 .next（） 调用上的 .send（） 中）。 但是，最简单的解决方案可能是：

server = connect()
response = None 
try:
    while True:
        line = server.send(response)
        response = None
        if should_respond(line):
            response ='WOW SUCH MESSAGE'
except StopIteration:
    pass