给定生成器对象,是否可以识别用于创建它的函数?
def gen_func():
yield
gen = gen_func()
// From `gen`, how do I get a reference to `gen_func` ?
我正在实现一个装饰器,用于装饰具有附加属性的生成器函数。我需要稍后从生成器本身访问该属性。
答案 0 :(得分:1)
因为生成器对象无法访问其生成器函数,并且因为无法在生成器上创建新属性,因此无法以干净的方式实现您所要求的内容。
但是你想要实现的目标可能。这是一个示例装饰器,它将生成器包装在迭代器对象中,并将自定义属性添加到迭代器对象中:
def add_stuff(f):
def replacement(*args, **kwargs):
gen = f(*args, **kwargs)
class FakeGenerator(object):
def __iter__(self):
return self
def next(self):
return gen.next()
fakegen = FakeGenerator()
fakegen.foo = "whatever"
fakegen.generator_function = f
return fakegen
return replacement
@add_stuff
def gen_func():
yield
gen = gen_func()
print gen
print gen.foo