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