python - 如何测试 Cython 属性是否为生成器？

Question

在 IPython 中，我可以看到 Cython 类的一个属性是一个生成器，只需定义它然后调用:

%%cython
cdef class SomeCls:
    property x:
        def __get__(self):
            yield 1

电话看起来像

SomeCls().x
# prints <generator at 0x102f61ee8>

我无法测试该属性是否为生成器:

import types
print(isinstance(SomeCls().x, types.GeneratorType))
# prints False

import inspect
print(inspect.isgeneratorfunction(SomeCls.x))
# prints False

如何确定 Cython 类的属性是否为生成器？

Answer 1

为什么通常的方法不起作用？

首先，您可能已经知道，inspect.isgeneratorfunction(...) 和 isinstance(..., types.GeneratorType) 之间没有区别 -检查模块只是 calls isinstance(..., types.GeneratorType)。

另一方面，types.GeneratorType 是 defined as

def _g():
    yield 1
GeneratorType = type(_g())

CPython 使用 PyGenObject ( here code , here documentation ) 作为生成器，对于一些 ABC 类的比较没有花哨的逻辑，所以 isinstance 将归结为比较 C -对象类型。

然而，Cython 返回一个 __pyx_CoroutineObject对于生成器(只需检查 cythonized 代码即可查看)

typedef PyObject *(*__pyx_coroutine_body_t)(PyObject *, PyThreadState *, PyObject *);
typedef struct {
    PyObject_HEAD
    __pyx_coroutine_body_t body;
    PyObject *closure;
    ...
    int resume_label;
    char is_running;
} __pyx_CoroutineObject;

就 isinstance 而言，它与 PyGenObject 无关 - 它并不关心 generator 是否在名称中类型的(但对于我们人类来说，这可能真的令人费解，因为 type(obj) 说的是“generator”)。

所以您将不得不推出您自​​己的 isgenerator 版本，它也考虑到了 Cython-“生成器”。方法有很多种，比如

%%cython
def _f():
    yield 1
CyGeneratorType = type(_f())   
def iscygenerator(o):
    return isinstance(o, CyGeneratorType)

现在:

import inspect   
def isgenerator(o):
    return inspect.isgenerator(o)  or iscygenerator(o)

isgenerator(SomeCls().x)          #  True
iscygenerator(SomeCls().x)        #  True
inspect.isgenerator(SomeCls().x)  #  False