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 void
Noddy_dealloc(Noddy * self)12 Noddy_dealloc(Noddy* self)
13 {
14 Py_XDECREF(self->first);
15 Py_XDECREF(self->last);
16 Py_TYPE(self)->tp_free((PyObject*)self);
17 }
18
19 static PyObject *
Noddy_new(PyTypeObject * type,PyObject * args,PyObject * kwds)20 Noddy_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
21 {
22 Noddy *self;
23
24 self = (Noddy *)type->tp_alloc(type, 0);
25 if (self != NULL) {
26 self->first = PyUnicode_FromString("");
27 if (self->first == NULL) {
28 Py_DECREF(self);
29 return NULL;
30 }
31
32 self->last = PyUnicode_FromString("");
33 if (self->last == NULL) {
34 Py_DECREF(self);
35 return NULL;
36 }
37
38 self->number = 0;
39 }
40
41 return (PyObject *)self;
42 }
43
44 static int
Noddy_init(Noddy * self,PyObject * args,PyObject * kwds)45 Noddy_init(Noddy *self, PyObject *args, PyObject *kwds)
46 {
47 PyObject *first=NULL, *last=NULL, *tmp;
48
49 static char *kwlist[] = {"first", "last", "number", NULL};
50
51 if (! PyArg_ParseTupleAndKeywords(args, kwds, "|SSi", kwlist,
52 &first, &last,
53 &self->number))
54 return -1;
55
56 if (first) {
57 tmp = self->first;
58 Py_INCREF(first);
59 self->first = first;
60 Py_DECREF(tmp);
61 }
62
63 if (last) {
64 tmp = self->last;
65 Py_INCREF(last);
66 self->last = last;
67 Py_DECREF(tmp);
68 }
69
70 return 0;
71 }
72
73 static PyMemberDef Noddy_members[] = {
74 {"number", T_INT, offsetof(Noddy, number), 0,
75 "noddy number"},
76 {NULL} /* Sentinel */
77 };
78
79 static PyObject *
Noddy_getfirst(Noddy * self,void * closure)80 Noddy_getfirst(Noddy *self, void *closure)
81 {
82 Py_INCREF(self->first);
83 return self->first;
84 }
85
86 static int
Noddy_setfirst(Noddy * self,PyObject * value,void * closure)87 Noddy_setfirst(Noddy *self, PyObject *value, void *closure)
88 {
89 if (value == NULL) {
90 PyErr_SetString(PyExc_TypeError, "Cannot delete the first attribute");
91 return -1;
92 }
93
94 if (! PyUnicode_Check(value)) {
95 PyErr_SetString(PyExc_TypeError,
96 "The first attribute value must be a string");
97 return -1;
98 }
99
100 Py_DECREF(self->first);
101 Py_INCREF(value);
102 self->first = value;
103
104 return 0;
105 }
106
107 static PyObject *
Noddy_getlast(Noddy * self,void * closure)108 Noddy_getlast(Noddy *self, void *closure)
109 {
110 Py_INCREF(self->last);
111 return self->last;
112 }
113
114 static int
Noddy_setlast(Noddy * self,PyObject * value,void * closure)115 Noddy_setlast(Noddy *self, PyObject *value, void *closure)
116 {
117 if (value == NULL) {
118 PyErr_SetString(PyExc_TypeError, "Cannot delete the last attribute");
119 return -1;
120 }
121
122 if (! PyUnicode_Check(value)) {
123 PyErr_SetString(PyExc_TypeError,
124 "The last attribute value must be a string");
125 return -1;
126 }
127
128 Py_DECREF(self->last);
129 Py_INCREF(value);
130 self->last = value;
131
132 return 0;
133 }
134
135 static PyGetSetDef Noddy_getseters[] = {
136 {"first",
137 (getter)Noddy_getfirst, (setter)Noddy_setfirst,
138 "first name",
139 NULL},
140 {"last",
141 (getter)Noddy_getlast, (setter)Noddy_setlast,
142 "last name",
143 NULL},
144 {NULL} /* Sentinel */
145 };
146
147 static PyObject *
Noddy_name(Noddy * self)148 Noddy_name(Noddy* self)
149 {
150 return PyUnicode_FromFormat("%S %S", self->first, self->last);
151 }
152
153 static PyMethodDef Noddy_methods[] = {
154 {"name", (PyCFunction)Noddy_name, METH_NOARGS,
155 "Return the name, combining the first and last name"
156 },
157 {NULL} /* Sentinel */
158 };
159
160 static PyTypeObject NoddyType = {
161 PyVarObject_HEAD_INIT(NULL, 0)
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_reserved */
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 |
181 Py_TPFLAGS_BASETYPE, /* tp_flags */
182 "Noddy objects", /* tp_doc */
183 0, /* tp_traverse */
184 0, /* tp_clear */
185 0, /* tp_richcompare */
186 0, /* tp_weaklistoffset */
187 0, /* tp_iter */
188 0, /* tp_iternext */
189 Noddy_methods, /* tp_methods */
190 Noddy_members, /* tp_members */
191 Noddy_getseters, /* tp_getset */
192 0, /* tp_base */
193 0, /* tp_dict */
194 0, /* tp_descr_get */
195 0, /* tp_descr_set */
196 0, /* tp_dictoffset */
197 (initproc)Noddy_init, /* tp_init */
198 0, /* tp_alloc */
199 Noddy_new, /* tp_new */
200 };
201
202 static PyModuleDef noddy3module = {
203 PyModuleDef_HEAD_INIT,
204 "noddy3",
205 "Example module that creates an extension type.",
206 -1,
207 NULL, NULL, NULL, NULL, NULL
208 };
209
210 PyMODINIT_FUNC
PyInit_noddy3(void)211 PyInit_noddy3(void)
212 {
213 PyObject* m;
214
215 if (PyType_Ready(&NoddyType) < 0)
216 return NULL;
217
218 m = PyModule_Create(&noddy3module);
219 if (m == NULL)
220 return NULL;
221
222 Py_INCREF(&NoddyType);
223 PyModule_AddObject(m, "Noddy", (PyObject *)&NoddyType);
224 return m;
225 }
226