224 lines · 5.4 KB · C
RawDownload
#include <Python.h>#include "structmember.h" typedef struct {    PyObject_HEAD    PyObject *first;    PyObject *last;    int number;} Noddy; static intNoddy_traverse(Noddy *self, visitproc visit, void *arg){    int vret;     if (self->first) {        vret = visit(self->first, arg);        if (vret != 0)            return vret;    }    if (self->last) {        vret = visit(self->last, arg);        if (vret != 0)            return vret;    }     return 0;} static int Noddy_clear(Noddy *self){    PyObject *tmp;     tmp = self->first;    self->first = NULL;    Py_XDECREF(tmp);     tmp = self->last;    self->last = NULL;    Py_XDECREF(tmp);     return 0;} static voidNoddy_dealloc(Noddy* self){    Noddy_clear(self);    self->ob_type->tp_free((PyObject*)self);} static PyObject *Noddy_new(PyTypeObject *type, PyObject *args, PyObject *kwds){    Noddy *self;     self = (Noddy *)type->tp_alloc(type, 0);    if (self != NULL) {        self->first = PyString_FromString("");        if (self->first == NULL)          {            Py_DECREF(self);            return NULL;          }                self->last = PyString_FromString("");        if (self->last == NULL)          {            Py_DECREF(self);            return NULL;          }         self->number = 0;    }     return (PyObject *)self;} static intNoddy_init(Noddy *self, PyObject *args, PyObject *kwds){    PyObject *first=NULL, *last=NULL, *tmp;     static char *kwlist[] = {"first", "last", "number", NULL};     if (! PyArg_ParseTupleAndKeywords(args, kwds, "|OOi", kwlist,                                       &first, &last,                                       &self->number))        return -1;      if (first) {        tmp = self->first;        Py_INCREF(first);        self->first = first;        Py_XDECREF(tmp);    }     if (last) {        tmp = self->last;        Py_INCREF(last);        self->last = last;        Py_XDECREF(tmp);    }     return 0;}  static PyMemberDef Noddy_members[] = {    {"first", T_OBJECT_EX, offsetof(Noddy, first), 0,     "first name"},    {"last", T_OBJECT_EX, offsetof(Noddy, last), 0,     "last name"},    {"number", T_INT, offsetof(Noddy, number), 0,     "noddy number"},    {NULL}  /* Sentinel */}; static PyObject *Noddy_name(Noddy* self){    static PyObject *format = NULL;    PyObject *args, *result;     if (format == NULL) {        format = PyString_FromString("%s %s");        if (format == NULL)            return NULL;    }     if (self->first == NULL) {        PyErr_SetString(PyExc_AttributeError, "first");        return NULL;    }     if (self->last == NULL) {        PyErr_SetString(PyExc_AttributeError, "last");        return NULL;    }     args = Py_BuildValue("OO", self->first, self->last);    if (args == NULL)        return NULL;     result = PyString_Format(format, args);    Py_DECREF(args);        return result;} static PyMethodDef Noddy_methods[] = {    {"name", (PyCFunction)Noddy_name, METH_NOARGS,     "Return the name, combining the first and last name"    },    {NULL}  /* Sentinel */}; static PyTypeObject NoddyType = {    PyObject_HEAD_INIT(NULL)    0,                         /*ob_size*/    "noddy.Noddy",             /*tp_name*/    sizeof(Noddy),             /*tp_basicsize*/    0,                         /*tp_itemsize*/    (destructor)Noddy_dealloc, /*tp_dealloc*/    0,                         /*tp_print*/    0,                         /*tp_getattr*/    0,                         /*tp_setattr*/    0,                         /*tp_compare*/    0,                         /*tp_repr*/    0,                         /*tp_as_number*/    0,                         /*tp_as_sequence*/    0,                         /*tp_as_mapping*/    0,                         /*tp_hash */    0,                         /*tp_call*/    0,                         /*tp_str*/    0,                         /*tp_getattro*/    0,                         /*tp_setattro*/    0,                         /*tp_as_buffer*/    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC, /*tp_flags*/    "Noddy objects",           /* tp_doc */    (traverseproc)Noddy_traverse,   /* tp_traverse */    (inquiry)Noddy_clear,           /* tp_clear */    0,		               /* tp_richcompare */    0,		               /* tp_weaklistoffset */    0,		               /* tp_iter */    0,		               /* tp_iternext */    Noddy_methods,             /* tp_methods */    Noddy_members,             /* tp_members */    0,                         /* tp_getset */    0,                         /* tp_base */    0,                         /* tp_dict */    0,                         /* tp_descr_get */    0,                         /* tp_descr_set */    0,                         /* tp_dictoffset */    (initproc)Noddy_init,      /* tp_init */    0,                         /* tp_alloc */    Noddy_new,                 /* tp_new */}; static PyMethodDef module_methods[] = {    {NULL}  /* Sentinel */}; #ifndef PyMODINIT_FUNC	/* declarations for DLL import/export */#define PyMODINIT_FUNC void#endifPyMODINIT_FUNCinitnoddy4(void) {    PyObject* m;     if (PyType_Ready(&NoddyType) < 0)        return;     m = Py_InitModule3("noddy4", module_methods,                       "Example module that creates an extension type.");     if (m == NULL)      return;     Py_INCREF(&NoddyType);    PyModule_AddObject(m, "Noddy", (PyObject *)&NoddyType);}