1 #include <Python.h>
2 #include "structmember.h"
3
4 typedef struct {
5 PyObject_HEAD
6 PyObject *first;
7 PyObject *last;
8 int number;
9 } Noddy;
10
11 static int
Noddy_traverse(Noddy * self,visitproc visit,void * arg)12 Noddy_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
30 static int
Noddy_clear(Noddy * self)31 Noddy_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
46 static void
Noddy_dealloc(Noddy * self)47 Noddy_dealloc(Noddy* self)
48 {
49 Noddy_clear(self);
50 self->ob_type->tp_free((PyObject*)self);
51 }
52
53 static PyObject *
Noddy_new(PyTypeObject * type,PyObject * args,PyObject * kwds)54 Noddy_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
80 static int
Noddy_init(Noddy * self,PyObject * args,PyObject * kwds)81 Noddy_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
110 static 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
120 static PyObject *
Noddy_name(Noddy * self)121 Noddy_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
152 static 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
159 static 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
201 static PyMethodDef module_methods[] = {
202 {NULL} /* Sentinel */
203 };
204
205 #ifndef PyMODINIT_FUNC /* declarations for DLL import/export */
206 #define PyMODINIT_FUNC void
207 #endif
208 PyMODINIT_FUNC
initnoddy4(void)209 initnoddy4(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 }
225