从 python 生成器获取函数

Question

我试图在 Python 中了解有关生成器函数的更多信息。据我所知，生成器函数在没有更多的yield调用之前不会完全返回，因此 function 返回的生成器中存在一个堆栈帧。

堆栈帧应该引用可调用的 function，所以我的问题是：如何从生成器中获取可调用的 function？

运行下面的代码时，我有一个生成器 function, test() 。

def test():
    for i in range(10):
        yield i

generator = test()

在这个例子中，有什么方法可以从generator中获取可调用的 function test()吗？

在查看这个答案之后，CPython 似乎会跟踪其中的一些，例如generator.frame和generator.code ，但是，我似乎无法将这些对象转换为函数。

我需要可调用的 function。像这样：

func = generator.something_special
new_generator = func()

Answer 1

您可以使用__name__属性获取原始 function 的名称，然后在 locals 中访问它，前提是它仍在 scope 中。

def test():
    for i in range(10):
        yield i


generator = test()
print(list(generator)) 
# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]


func_name = generator.__name__ 
new_generator = locals()[func_name]()
print(func_name, list(new_generator)) 
# test [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]