如何使用Py_LIMITED_API定义C++扩展中定义的Python类型的常量值?

xn1cxnb4  于 11个月前  发布在  Python
关注(0)|答案(1)|浏览(101)

我想使用Py_LIMITED_API。因此,我没有静态地定义Python类型,而是动态地定义它,使用如下的specs和slot。

struct MyType { PyObject_HEAD };

static void MyType_dealloc( MyType * self ) { PyObject_DEL( self ); }

static PyObject * MyType_new( PyTypeObject * type, PyObject * args, PyObject * kwds ) {
    return (PyObject *)PyObject_NEW( MyType, type );
}

static int MyType_init( MyType * self, PyObject * args, PyObject * kwds ) { return 0; }

static PyType_Slot MyType_slots[] = {
    { Py_tp_doc, (void *)PyDoc_STR("MyType objects") },
    { Py_tp_dealloc, (void *)&MyType_dealloc },
    { Py_tp_init, (void *)&MyType_init },
    { Py_tp_new, (void *)&MyType_new },
    { 0, NULL }
};

static PyType_Spec MyType_spec = {
    "MyType",
    sizeof(MyType),
    0,
    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
    MyType_slots
};

static PyTypeObject * MyTypePtr = NULL;

static PyModuleDef MyModule = {
    PyModuleDef_HEAD_INIT,
    "MyModule",
    "My C++ Module.",
    -1,
    NULL, NULL, NULL, NULL, NULL
};

PyMODINIT_FUNC
PyInit_MyModule(void)
{
    PyObject* m = PyModule_Create( &MyModule );
    if (m == NULL) return NULL;
    MyTypePtr = (PyTypeObject*)PyType_FromSpec( &MyType_spec );
    if ( MyTypePtr == NULL ) return NULL;
    if ( PyType_Ready( MyTypePtr ) < 0 ) return NULL;
    PyDict_SetItemString( MyTypePtr->tp_dict, "Normal", PyLong_FromLong( 0 ) );
    PyDict_SetItemString( MyTypePtr->tp_dict, "Custom", PyLong_FromLong( 15 ) );
    Py_INCREF( MyTypePtr );
    PyModule_AddObject( m, "MyType", (PyObject *)MyTypePtr );
    return m;
}

字符串
有了这个模块的定义,我可以写下面的代码。

from MyModule import *
print( MyType.Normal )
print( MyType.Custom )


我的问题是,对于Py_LIMITED_API,我不能使用tp_dict来定义常量值NormalCustom
我试过使用PyType_GetSlot,但是Py_tp_dict并不存在(我想我明白为什么:因为它允许修改现有类型)。
现在我的问题是:在上面的spec和slot代码中,如何在不使用tp_dict的情况下定义MyType类型的常量NormalCustom

wd2eg0qa

wd2eg0qa1#

有一个非常简单的答案:在类型对象上调用PyObject_SetAttrString

PyMODINIT_FUNC
PyInit_MyModule(void)
{
    PyObject* m = PyModule_Create( &MyModule );
    if (m == NULL) return NULL;
    MyTypePtr = (PyTypeObject*)PyType_FromSpec( &MyType_spec );
    if ( MyTypePtr == NULL ) return NULL;
    if ( PyType_Ready( MyTypePtr ) < 0 ) return NULL;
    PyObject_SetAttrString( (PyObject*)MyTypePtr, "Normal", PyLong_FromLong( 0 ) );
    PyObject_SetAttrString( (PyObject*)MyTypePtr, "Custom", PyLong_FromLong( 15 ) );
    Py_INCREF( MyTypePtr );
    PyModule_AddObject( m, "MyType", (PyObject *)MyTypePtr );
    return m;
}

字符串
我猜它也适用于PyObject_SetAttr
不再使用tp_dict
可以使用Py_LIMITED_API

相关问题