Python 生成器中有什么类似于“self”的东西吗？

Question

有没有办法在生成器的定义中获取对返回的生成器对象的引用？这类似于传递给迭代器的__next__ 方法内部的方法的self 参数。在浏览了 Python 的文档后，我没有发现任何类似的东西。

当我在探索以下论文中的多少想法时，我提出了这个问题，我可以使用生成器作为协程在 Python 中实现多少。论文：http://citeseerx.ist.psu.edu/viewdoc/summary?doi=10.1.1.19.79

我能做的最接近的方法是使用装饰器，它基于 David Beazley 的 coroutine 装饰器，但感觉有点像 hack。

from functools import wraps


def coroutine(func):
    @wraps(func)
    def decorated(*args, **kwargs):
        f = func(*args, **kwargs)
        next(f)
        f.send(f)
        return f
    return decorated


@coroutine
def A():
    self = yield
    # do stuff...

---

编辑：基于下面的答案，以下类可用作装饰器，以使生成器接收对 self 的引用作为其第一个参数。它还有一个额外的好处是，任何用它装饰的生成器都将具有 coroutine 类型。

class coroutine(object):
    """Decorator class for coroutines with a self parameter."""
    def __new__(cls, func):
        @wraps(func)
        def decorated(*args, **kwargs):
            o = object.__new__(cls)
            o.__init__(func, args, kwargs)
            return o
        return decorated

    def __init__(self, generator, args, kw):
        self.generator = generator(self, *args, **kw)
        next(self.generator)

    def __iter__(self):
        return self

    def __next__(self):
        return next(self.generator)

    next = __next__

    def send(self, value):
        return self.generator.send(value)


# Usage:

@coroutine
def A(self):
    while True:
        message = yield
        print self is message


a = A()
b = A()
a.send(a)  # outputs True
a.send(b)  # outputs False

Answer 1

这是使用代理的建议。

def selfDecorator(func):
    def wrap(*args, **kw):
        return SelfGenerator(func, args, kw)
    return wrap

class SelfGenerator(object):
    """This class implements the generator interface"""
    def __init__(self, generator, args, kw):
        self.generator = generator(self, *args, **kw)
    def __iter__(self):
        return self
    def __next__(self):
        return next(self.generator)
    next = __next__
    def send(self, value):
        return self.generator.send(value)

@selfDecorator
def gen(self, x): # your generator function with self
    for i in range(x):
        yield self


for x in gen(5):
    print x # prints <__main__.SelfGenerator object at 0x02BB16D0>

由于SelfGenerator 是原始生成器的代理，因此它具有相同的接口，可以完全用作 Python 自己的生成器。

第一个答案

你不能自己调用​​生成器：

>>> def generator():
    for value in g:
        yield value


>>> g = generator()
>>> next(g)

Traceback (most recent call last):
  File "<pyshell#13>", line 1, in <module>
    next(g)
  File "<pyshell#11>", line 2, in generator
    for value in g:
ValueError: generator already executing

猜测：论文中的 self 可能不是生成器本身，而是一个对象，一个可能在一些生成器之间共享的状态持有者。

更准确地说，约束的有向图被认为是无向图时，要求它是无环的。

这让我觉得生成器没有引用它的“相同”执行。为什么它要通过迭代来获得自己的价值？它可以使用局部变量。

协程源自 Simula。也许要了解自我是什么，您可以看一下语言。