如何从Python生成器对象中获取对应的可调用生成器函数?
从生成器对象中获取原可调用函数
在CPython中,生成器对象并没有标准的公开属性可以直接返回原可调用函数,但可以通过一些CPython专属的隐藏属性或命名空间查找的方式实现需求:
方法一:通过模块命名空间查找
如果生成器对应的函数处于模块的全局作用域,可以借助函数名和模块信息获取原函数:
import inspect def test(): for i in range(10): yield i generator = test() # 获取生成器对应函数的限定名称 func_qualname = generator.__qualname__ # 获取生成器所属的模块 module = inspect.getmodule(generator) # 从模块全局命名空间中取出函数对象 func = getattr(module, func_qualname) # 生成新的生成器实例 new_generator = func()
这个方法的局限性是:函数必须在模块全局作用域中,且不能有同名函数覆盖。
方法二:利用CPython专属栈帧属性
生成器对象的gi_frame属性(CPython私有属性)指向当前的栈帧,通过栈帧的全局命名空间可以找到原函数:
def test(): for i in range(10): yield i generator = test() # 访问CPython专属的栈帧属性 frame = generator.gi_frame # 从栈帧的全局命名空间中根据函数名获取函数 func = frame.f_globals[generator.__qualname__] new_generator = func()
注意:gi_frame是CPython特有的非标准属性,PyPy等其他Python实现不支持,且未来CPython版本可能移除该属性,生产环境需谨慎使用。
关于代码对象的说明
你提到的生成器关联的代码对象(generator.gi_code)仅包含函数的字节码和元信息,不包含函数的上下文绑定(比如闭包环境、命名空间关联),因此无法直接从代码对象还原出可调用的函数对象,必须结合命名空间信息才能获取完整的函数引用。
内容的提问来源于stack exchange,提问作者Michael M.
相关产品推荐
相关产品推荐

