Python C-API Заставить len (...) работать с классом расширения - PullRequest
4 голосов
/ 14 января 2010

При создании класса в Python я могу просто сделать метод def __len__(self):, чтобы заставить len(InstanceOfMyClass) работать, однако я не могу узнать, как это сделать с помощью класса расширения через C-API.

Я попытался добавить метод __len__, но, похоже, он не работает

{"__len__",(PyCFunction)&TestClass_GetLen,METH_NOARGS,""},

Тестовый код Python:

def test(my_instance):
    x = len(my_instance)#exception
    return x

TypeError: object of type 'test_module.test_class' has no len()

Код для TestClass

struct TestClass;
static int TestClass_Init(TestClass *self, PyObject *args, PyObject* kwds);
static void TestClass_Dealloc(TestClass *self);
static PyObject* TestClass_GetLen(TestClass *self);

struct TestClass
{
    PyObject_HEAD;
};
static PyMethodDef TestClass_methods[] =
{
    {"__len__",(PyCFunction)&TestClass_GetLen,METH_O,""},
    {NULL}
};
static PyTypeObject TestClass_type = {PyObject_HEAD_INIT(NULL)};
bool InitTestClass(PyObject *module)
{
    TestClass_type.tp_basicsize = sizeof(TestClass);
    TestClass_type.tp_name      = PY_MODULE_NAME".TestClass";
    TestClass_type.tp_doc       = "";
    TestClass_type.tp_flags     = Py_TPFLAGS_DEFAULT;
    TestClass_type.tp_methods   = TestClass_methods;
    TestClass_type.tp_new       = PyType_GenericNew;
    TestClass_type.tp_init      = (initproc)TestClass_Init;
    TestClass_type.tp_dealloc   = (destructor)TestClass_Dealloc;
    if(PyType_Ready(TestClass_type) < 0) return false;

    Py_INCREF(TestClass_type);
    PyModule_AddObject(module, "TestClass", (PyObject*)&TestClass_type);
    return true;
};
void TestClass_Dealloc(TestClass *self)
{
    Py_TYPE(self)->tp_free((PyObject*)self);
}
int TestClass_Init(TestClass *self, PyObject *args, PyObject* kwds)
{
    return 0;
}
PyObject* TestClass_GetLen(TestClass *self)
{
    return PyLong_FromLong(55);
}

Ответы [ 2 ]

3 голосов
/ 16 января 2010

Элемент tp_as_sequence структуры typeobject должен быть заполнен чем-то, в котором заполнен элемент sq_length.

3 голосов
/ 14 января 2010

Как говорит Игнасио, правильный способ - заполнить элемент tp_as_sequence вашего типаобъекта. Вот минимальный пример:

#include <Python.h>

/**
 * C structure and methods definitions
 */

typedef struct {
    PyObject_HEAD;
} TestObject;

static Py_ssize_t
TestClass_len(TestObject* self) 
{   
    return 55;
}

/**
 * Python definitions
 */

static PySequenceMethods TestClass_sequence_methods = {
    TestClass_len,                  /* sq_length */
};

static PyTypeObject TestClass = {
    PyObject_HEAD_INIT(NULL)    
    0,                              /*ob_size*/
    "testmodule.TestClass",         /*tp_name*/
    sizeof(TestObject),             /*tp_basicsize*/
};

/**
 * Module entry point
 */
#ifndef PyMODINIT_FUNC  /* declarations for DLL import/export */
#define PyMODINIT_FUNC void
#endif
PyMODINIT_FUNC
inittestmodule(void) 
{
    PyObject* m;

    TestClass.tp_new = PyType_GenericNew;
    TestClass.tp_as_sequence = &TestClass_sequence_methods;
    TestClass.tp_flags = Py_TPFLAGS_DEFAULT;

    if (PyType_Ready(&TestClass) < 0)
        return;

    m = Py_InitModule3("testmodule", NULL, "");

    Py_INCREF(&TestClass);
    PyModule_AddObject(m, "TestClass", (PyObject *)&TestClass);
}

Другой (более сложный и медленный) способ - это динамическое добавление метода __len__ в ваш класс в функции init модуля с PyCFunction_New, PyMethod_New и PyObject_SetAttr. Это будет больше похоже на код, который вы вставили, за исключением того, что вы определили __len__ в таблице методов C, что не работает.

...