Commit | Line | Data |
---|---|---|
920dae64 AT |
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 | |
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 | |
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 | |
47 | Noddy_dealloc(Noddy* self) | |
48 | { | |
49 | Noddy_clear(self); | |
50 | self->ob_type->tp_free((PyObject*)self); | |
51 | } | |
52 | ||
53 | static PyObject * | |
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 | |
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 * | |
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 | |
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 | } |