1 // namespace object implementation
2
3 #include "Python.h"
4 #include "structmember.h" // PyMemberDef
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_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
136 error:
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
147 static int
namespace_traverse(_PyNamespaceObject * ns,visitproc visit,void * arg)148 namespace_traverse(_PyNamespaceObject *ns, visitproc visit, void *arg)
149 {
150 Py_VISIT(ns->ns_dict);
151 return 0;
152 }
153
154
155 static int
namespace_clear(_PyNamespaceObject * ns)156 namespace_clear(_PyNamespaceObject *ns)
157 {
158 Py_CLEAR(ns->ns_dict);
159 return 0;
160 }
161
162
163 static PyObject *
namespace_richcompare(PyObject * self,PyObject * other,int op)164 namespace_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
174 PyDoc_STRVAR(namespace_reduce__doc__, "Return state information for pickling");
175
176 static PyObject *
namespace_reduce(_PyNamespaceObject * ns,PyObject * Py_UNUSED (ignored))177 namespace_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
190 static PyMethodDef namespace_methods[] = {
191 {"__reduce__", (PyCFunction)namespace_reduce, METH_NOARGS,
192 namespace_reduce__doc__},
193 {NULL, NULL} // sentinel
194 };
195
196
197 PyDoc_STRVAR(namespace_doc,
198 "A simple attribute-based namespace.\n\
199 \n\
200 SimpleNamespace(**kwargs)");
201
202 PyTypeObject _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
246 PyObject *
_PyNamespace_New(PyObject * kwds)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