PageRenderTime 16ms CodeModel.GetById 2ms app.highlight 11ms RepoModel.GetById 1ms app.codeStats 1ms

/Doc/includes/noddy4.c

http://unladen-swallow.googlecode.com/
C | 224 lines | 187 code | 37 blank | 0 comment | 28 complexity | 6278a33e4c6f056496098aad972b97e6 MD5 | raw file
  1#include <Python.h>
  2#include "structmember.h"
  3
  4typedef struct {
  5    PyObject_HEAD
  6    PyObject *first;
  7    PyObject *last;
  8    int number;
  9} Noddy;
 10
 11static int
 12Noddy_traverse(Noddy *self, visitproc visit, void *arg)
 13{
 14    int vret;
 15
 16    if (self->first) {
 17        vret = visit(self->first, arg);
 18        if (vret != 0)
 19            return vret;
 20    }
 21    if (self->last) {
 22        vret = visit(self->last, arg);
 23        if (vret != 0)
 24            return vret;
 25    }
 26
 27    return 0;
 28}
 29
 30static int 
 31Noddy_clear(Noddy *self)
 32{
 33    PyObject *tmp;
 34
 35    tmp = self->first;
 36    self->first = NULL;
 37    Py_XDECREF(tmp);
 38
 39    tmp = self->last;
 40    self->last = NULL;
 41    Py_XDECREF(tmp);
 42
 43    return 0;
 44}
 45
 46static void
 47Noddy_dealloc(Noddy* self)
 48{
 49    Noddy_clear(self);
 50    self->ob_type->tp_free((PyObject*)self);
 51}
 52
 53static PyObject *
 54Noddy_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
 55{
 56    Noddy *self;
 57
 58    self = (Noddy *)type->tp_alloc(type, 0);
 59    if (self != NULL) {
 60        self->first = PyString_FromString("");
 61        if (self->first == NULL)
 62          {
 63            Py_DECREF(self);
 64            return NULL;
 65          }
 66        
 67        self->last = PyString_FromString("");
 68        if (self->last == NULL)
 69          {
 70            Py_DECREF(self);
 71            return NULL;
 72          }
 73
 74        self->number = 0;
 75    }
 76
 77    return (PyObject *)self;
 78}
 79
 80static int
 81Noddy_init(Noddy *self, PyObject *args, PyObject *kwds)
 82{
 83    PyObject *first=NULL, *last=NULL, *tmp;
 84
 85    static char *kwlist[] = {"first", "last", "number", NULL};
 86
 87    if (! PyArg_ParseTupleAndKeywords(args, kwds, "|OOi", kwlist, 
 88                                      &first, &last, 
 89                                      &self->number))
 90        return -1; 
 91
 92    if (first) {
 93        tmp = self->first;
 94        Py_INCREF(first);
 95        self->first = first;
 96        Py_XDECREF(tmp);
 97    }
 98
 99    if (last) {
100        tmp = self->last;
101        Py_INCREF(last);
102        self->last = last;
103        Py_XDECREF(tmp);
104    }
105
106    return 0;
107}
108
109
110static PyMemberDef Noddy_members[] = {
111    {"first", T_OBJECT_EX, offsetof(Noddy, first), 0,
112     "first name"},
113    {"last", T_OBJECT_EX, offsetof(Noddy, last), 0,
114     "last name"},
115    {"number", T_INT, offsetof(Noddy, number), 0,
116     "noddy number"},
117    {NULL}  /* Sentinel */
118};
119
120static PyObject *
121Noddy_name(Noddy* self)
122{
123    static PyObject *format = NULL;
124    PyObject *args, *result;
125
126    if (format == NULL) {
127        format = PyString_FromString("%s %s");
128        if (format == NULL)
129            return NULL;
130    }
131
132    if (self->first == NULL) {
133        PyErr_SetString(PyExc_AttributeError, "first");
134        return NULL;
135    }
136
137    if (self->last == NULL) {
138        PyErr_SetString(PyExc_AttributeError, "last");
139        return NULL;
140    }
141
142    args = Py_BuildValue("OO", self->first, self->last);
143    if (args == NULL)
144        return NULL;
145
146    result = PyString_Format(format, args);
147    Py_DECREF(args);
148    
149    return result;
150}
151
152static PyMethodDef Noddy_methods[] = {
153    {"name", (PyCFunction)Noddy_name, METH_NOARGS,
154     "Return the name, combining the first and last name"
155    },
156    {NULL}  /* Sentinel */
157};
158
159static PyTypeObject NoddyType = {
160    PyObject_HEAD_INIT(NULL)
161    0,                         /*ob_size*/
162    "noddy.Noddy",             /*tp_name*/
163    sizeof(Noddy),             /*tp_basicsize*/
164    0,                         /*tp_itemsize*/
165    (destructor)Noddy_dealloc, /*tp_dealloc*/
166    0,                         /*tp_print*/
167    0,                         /*tp_getattr*/
168    0,                         /*tp_setattr*/
169    0,                         /*tp_compare*/
170    0,                         /*tp_repr*/
171    0,                         /*tp_as_number*/
172    0,                         /*tp_as_sequence*/
173    0,                         /*tp_as_mapping*/
174    0,                         /*tp_hash */
175    0,                         /*tp_call*/
176    0,                         /*tp_str*/
177    0,                         /*tp_getattro*/
178    0,                         /*tp_setattro*/
179    0,                         /*tp_as_buffer*/
180    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC, /*tp_flags*/
181    "Noddy objects",           /* tp_doc */
182    (traverseproc)Noddy_traverse,   /* tp_traverse */
183    (inquiry)Noddy_clear,           /* tp_clear */
184    0,		               /* tp_richcompare */
185    0,		               /* tp_weaklistoffset */
186    0,		               /* tp_iter */
187    0,		               /* tp_iternext */
188    Noddy_methods,             /* tp_methods */
189    Noddy_members,             /* tp_members */
190    0,                         /* tp_getset */
191    0,                         /* tp_base */
192    0,                         /* tp_dict */
193    0,                         /* tp_descr_get */
194    0,                         /* tp_descr_set */
195    0,                         /* tp_dictoffset */
196    (initproc)Noddy_init,      /* tp_init */
197    0,                         /* tp_alloc */
198    Noddy_new,                 /* tp_new */
199};
200
201static PyMethodDef module_methods[] = {
202    {NULL}  /* Sentinel */
203};
204
205#ifndef PyMODINIT_FUNC	/* declarations for DLL import/export */
206#define PyMODINIT_FUNC void
207#endif
208PyMODINIT_FUNC
209initnoddy4(void) 
210{
211    PyObject* m;
212
213    if (PyType_Ready(&NoddyType) < 0)
214        return;
215
216    m = Py_InitModule3("noddy4", module_methods,
217                       "Example module that creates an extension type.");
218
219    if (m == NULL)
220      return;
221
222    Py_INCREF(&NoddyType);
223    PyModule_AddObject(m, "Noddy", (PyObject *)&NoddyType);
224}