python 检查调用者函数是否是生成器

Question

我想从函数的装饰器中检索有关调用者函数的信息。也就是说，我想知道调用者函数是否是生成器。正在尝试检查外部堆栈框架，但我找不到方法

def decorator(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        inspect.isgeneratorfunction(func) ### THIS WORKS 
        frame = inspect.currentframe().f_back
        ### HERE THE MAGIC

    return wrapper

这可能吗？或者有一个简单的方法可以做到吗？

Answer 1

generator = bool(frame.f_code.co_flags&32)

从一些小测试来看，至少在我的 python3 版本中，生成器在十进制代码帧中设置了标志 32。您想要检查并看看它的可移植性如何，但至少在 cpython 上可能是相当可移植的。