访问 Python 生成器内的局部变量

Question

如何从生成器外部访问 Python 生成器内部定义的局部变量？

我有一个例子，我的生成器操纵本地状态，对于单元测试，我想检查这个状态以确保它包含正确的值。

我无法将状态存储到实例变量（例如 self.state = blah），因为我可能从同一个类实例创建多个生成器，这意味着生成器可能会覆盖彼此的状态。我也无法返回yield 表达式中的状态，因为状态名称可能会因各个生成器实例而更改或变化。

例如我想做这样的事情（尽管这个代码不起作用）

from random import random

class MyIter(object):
    def __iter__(self):
        context = {}
        for i in xrange(10):
            context[random()] = random()
            yield i

obj = MyIter()
i1 = iter(obj)
i2 = iter(obj)
while 1:
    try:
        i1.next()
        i2.next()
        print i1.context
        print i2.context
    except StopIteration:
        break

有没有办法通过检查Python的执行堆栈来访问局部变量？

Answer 1

很抱歉回答我自己的问题，但是在深入研究生成器接口后，我找到了访问生成器局部变量所需的确切路径：

from random import random

class MyIter(object):
    def __iter__(self):
        context = {}
        for i in xrange(10):
            context[random()] = random()
            yield i

obj = MyIter()
i1 = iter(obj)
i2 = iter(obj)
while 1:
    try:
        i1.next()
        i2.next()
        print i1.gi_frame.f_locals['context']
        print i2.gi_frame.f_locals['context']
    except StopIteration:
        break

Answer 2

您应该将生成器视为黑匣子。单元测试不应该关心其内部状态，因为那只是一个实现细节；他们应该只关心指定的行为。

Answer 3

如果你真的想这样做，请将迭代器类与容器类分开：

from random import random

class MyContainer(object):
    def __iter__(self):
        return MyIter(self)

class MyIter(object):
    def __init__(self, container):
        self.container = container
        self.context = {}
        self.it = iter(xrange(10))
    def next(self):
        self.context[random()] = random()
        return next(self.it)
    def __iter__(self):
        return self

obj = MyContainer()
# ...

但我认为这不是很有用......