1 // namespace object implementation
2
3 #include "Python.h"
4 #include "structmember.h"
5
6
7 typedef struct {
8 PyObject_HEAD
9 PyObject *ns_dict;
10 } _PyNamespaceObject;
11
12
13 static PyMemberDef namespace_members[] = {
14 {"__dict__", T_OBJECT, offsetof(_PyNamespaceObject, ns_dict), READONLY},
15 {NULL}
16 };
17
18
19 // Methods
20
21 static PyObject *
namespace_new(PyTypeObject * type,PyObject * args,PyObject * kwds)22 namespace_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
23 {
24 PyObject *self;
25
26 assert(type != NULL && type->tp_alloc != NULL);
27 self = type->tp_alloc(type, 0);
28 if (self != NULL) {
29 _PyNamespaceObject *ns = (_PyNamespaceObject *)self;
30 ns->ns_dict = PyDict_New();
31 if (ns->ns_dict == NULL) {
32 Py_DECREF(ns);
33 return NULL;
34 }
35 }
36 return self;
37 }
38
39
40 static int
namespace_init(_PyNamespaceObject * ns,PyObject * args,PyObject * kwds)41 namespace_init(_PyNamespaceObject *ns, PyObject *args, PyObject *kwds)
42 {
43 if (PyTuple_GET_SIZE(args) != 0) {
44 PyErr_Format(PyExc_TypeError, "no positional arguments expected");
45 return -1;
46 }
47 if (kwds == NULL) {
48 return 0;
49 }
50 if (!PyArg_ValidateKeywordArguments(kwds)) {
51 return -1;
52 }
53 return PyDict_Update(ns->ns_dict, kwds);
54 }
55
56
57 static void
namespace_dealloc(_PyNamespaceObject * ns)58 namespace_dealloc(_PyNamespaceObject *ns)
59 {
60 PyObject_GC_UnTrack(ns);
61 Py_CLEAR(ns->ns_dict);
62 Py_TYPE(ns)->tp_free((PyObject *)ns);
63 }
64
65
66 static PyObject *
namespace_repr(PyObject * ns)67 namespace_repr(PyObject *ns)
68 {
69 int i, loop_error = 0;
70 PyObject *pairs = NULL, *d = NULL, *keys = NULL, *keys_iter = NULL;
71 PyObject *key;
72 PyObject *separator, *pairsrepr, *repr = NULL;
73 const char * name;
74
75 name = (Py_TYPE(ns) == &_PyNamespace_Type) ? "namespace"
76 : ns->ob_type->tp_name;
77
78 i = Py_ReprEnter(ns);
79 if (i != 0) {
80 return i > 0 ? PyUnicode_FromFormat("%s(...)", name) : NULL;
81 }
82
83 pairs = PyList_New(0);
84 if (pairs == NULL)
85 goto error;
86
87 d = ((_PyNamespaceObject *)ns)->ns_dict;
88 assert(d != NULL);
89 Py_INCREF(d);
90
91 keys = PyDict_Keys(d);
92 if (keys == NULL)
93 goto error;
94 if (PyList_Sort(keys) != 0)
95 goto error;
96
97 keys_iter = PyObject_GetIter(keys);
98 if (keys_iter == NULL)
99 goto error;
100
101 while ((key = PyIter_Next(keys_iter)) != NULL) {
102 if (PyUnicode_Check(key) && PyUnicode_GET_LENGTH(key) > 0) {
103 PyObject *value, *item;
104
105 value = PyDict_GetItemWithError(d, key);
106 if (value != NULL) {
107 item = PyUnicode_FromFormat("%U=%R", key, value);
108 if (item == NULL) {
109 loop_error = 1;
110 }
111 else {
112 loop_error = PyList_Append(pairs, item);
113 Py_DECREF(item);
114 }
115 }
116 else if (PyErr_Occurred()) {
117 loop_error = 1;
118 }
119 }
120
121 Py_DECREF(key);
122 if (loop_error)
123 goto error;
124 }
125
126 separator = PyUnicode_FromString(", ");
127 if (separator == NULL)
128 goto error;
129
130 pairsrepr = PyUnicode_Join(separator, pairs);
131 Py_DECREF(separator);
132 if (pairsrepr == NULL)
133 goto error;
134
135 repr = PyUnicode_FromFormat("%s(%S)", name, pairsrepr);
136 Py_DECREF(pairsrepr);
137
138 error:
139 Py_XDECREF(pairs);
140 Py_XDECREF(d);
141 Py_XDECREF(keys);
142 Py_XDECREF(keys_iter);
143 Py_ReprLeave(ns);
144
145 return repr;
146 }
147
148
149 static int
namespace_traverse(_PyNamespaceObject * ns,visitproc visit,void * arg)150 namespace_traverse(_PyNamespaceObject *ns, visitproc visit, void *arg)
151 {
152 Py_VISIT(ns->ns_dict);
153 return 0;
154 }
155
156
157 static int
namespace_clear(_PyNamespaceObject * ns)158 namespace_clear(_PyNamespaceObject *ns)
159 {
160 Py_CLEAR(ns->ns_dict);
161 return 0;
162 }
163
164
165 static PyObject *
namespace_richcompare(PyObject * self,PyObject * other,int op)166 namespace_richcompare(PyObject *self, PyObject *other, int op)
167 {
168 if (PyObject_TypeCheck(self, &_PyNamespace_Type) &&
169 PyObject_TypeCheck(other, &_PyNamespace_Type))
170 return PyObject_RichCompare(((_PyNamespaceObject *)self)->ns_dict,
171 ((_PyNamespaceObject *)other)->ns_dict, op);
172 Py_RETURN_NOTIMPLEMENTED;
173 }
174
175
176 PyDoc_STRVAR(namespace_reduce__doc__, "Return state information for pickling");
177
178 static PyObject *
namespace_reduce(_PyNamespaceObject * ns,PyObject * Py_UNUSED (ignored))179 namespace_reduce(_PyNamespaceObject *ns, PyObject *Py_UNUSED(ignored))
180 {
181 PyObject *result, *args = PyTuple_New(0);
182
183 if (!args)
184 return NULL;
185
186 result = PyTuple_Pack(3, (PyObject *)Py_TYPE(ns), args, ns->ns_dict);
187 Py_DECREF(args);
188 return result;
189 }
190
191
192 static PyMethodDef namespace_methods[] = {
193 {"__reduce__", (PyCFunction)namespace_reduce, METH_NOARGS,
194 namespace_reduce__doc__},
195 {NULL, NULL} // sentinel
196 };
197
198
199 PyDoc_STRVAR(namespace_doc,
200 "A simple attribute-based namespace.\n\
201 \n\
202 SimpleNamespace(**kwargs)");
203
204 PyTypeObject _PyNamespace_Type = {
205 PyVarObject_HEAD_INIT(&PyType_Type, 0)
206 "types.SimpleNamespace", /* tp_name */
207 sizeof(_PyNamespaceObject), /* tp_basicsize */
208 0, /* tp_itemsize */
209 (destructor)namespace_dealloc, /* tp_dealloc */
210 0, /* tp_vectorcall_offset */
211 0, /* tp_getattr */
212 0, /* tp_setattr */
213 0, /* tp_as_async */
214 (reprfunc)namespace_repr, /* tp_repr */
215 0, /* tp_as_number */
216 0, /* tp_as_sequence */
217 0, /* tp_as_mapping */
218 0, /* tp_hash */
219 0, /* tp_call */
220 0, /* tp_str */
221 PyObject_GenericGetAttr, /* tp_getattro */
222 PyObject_GenericSetAttr, /* tp_setattro */
223 0, /* tp_as_buffer */
224 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
225 Py_TPFLAGS_BASETYPE, /* tp_flags */
226 namespace_doc, /* tp_doc */
227 (traverseproc)namespace_traverse, /* tp_traverse */
228 (inquiry)namespace_clear, /* tp_clear */
229 namespace_richcompare, /* tp_richcompare */
230 0, /* tp_weaklistoffset */
231 0, /* tp_iter */
232 0, /* tp_iternext */
233 namespace_methods, /* tp_methods */
234 namespace_members, /* tp_members */
235 0, /* tp_getset */
236 0, /* tp_base */
237 0, /* tp_dict */
238 0, /* tp_descr_get */
239 0, /* tp_descr_set */
240 offsetof(_PyNamespaceObject, ns_dict), /* tp_dictoffset */
241 (initproc)namespace_init, /* tp_init */
242 PyType_GenericAlloc, /* tp_alloc */
243 (newfunc)namespace_new, /* tp_new */
244 PyObject_GC_Del, /* tp_free */
245 };
246
247
248 PyObject *
_PyNamespace_New(PyObject * kwds)249 _PyNamespace_New(PyObject *kwds)
250 {
251 PyObject *ns = namespace_new(&_PyNamespace_Type, NULL, NULL);
252 if (ns == NULL)
253 return NULL;
254
255 if (kwds == NULL)
256 return ns;
257 if (PyDict_Update(((_PyNamespaceObject *)ns)->ns_dict, kwds) != 0) {
258 Py_DECREF(ns);
259 return NULL;
260 }
261
262 return (PyObject *)ns;
263 }
264