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 * |
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 |
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 |
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 * |
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 |
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 |
156 | namespace_clear(_PyNamespaceObject *ns) |
157 | { |
158 | Py_CLEAR(ns->ns_dict); |
159 | return 0; |
160 | } |
161 | |
162 | |
163 | static PyObject * |
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 * |
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 * |
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 | |