diff --git a/Lib/test/test_types.py b/Lib/test/test_types.py index 3f5ac98a0e3..3ee4c6bc5fd 100644 --- a/Lib/test/test_types.py +++ b/Lib/test/test_types.py @@ -1135,6 +1135,15 @@ class SimpleNamespaceTests(unittest.TestCase): with self.assertRaises(TypeError): ns['spam'] + def test_subclass(self): + class Spam(types.SimpleNamespace): + pass + + spam = Spam(ham=8, eggs=9) + + self.assertIs(type(spam), Spam) + self.assertEqual(vars(spam), {'ham': 8, 'eggs': 9}) + def test_main(): run_unittest(TypesTests, MappingProxyTests, ClassCreationTests, diff --git a/Misc/NEWS b/Misc/NEWS index aaf07aae4c0..8efe3860fbb 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -16,6 +16,8 @@ Core and Builtins - Issue #14783: Improve int() docstring and switch docstrings for str(), range(), and slice() to use multi-line signatures. +- Issue #16160: Subclass support now works for types.SimpleNamespace. + - Upgrade Unicode data (UCD) to version 6.2. - Issue #15379: Fix passing of non-BMP characters as integers for the charmap diff --git a/Objects/namespaceobject.c b/Objects/namespaceobject.c index 753874c3cbe..ff278d33474 100644 --- a/Objects/namespaceobject.c +++ b/Objects/namespaceobject.c @@ -21,19 +21,19 @@ static PyMemberDef namespace_members[] = { static PyObject * namespace_new(PyTypeObject *type, PyObject *args, PyObject *kwds) { - _PyNamespaceObject *ns; - ns = PyObject_GC_New(_PyNamespaceObject, &_PyNamespace_Type); - if (ns == NULL) - return NULL; + PyObject *self; - ns->ns_dict = PyDict_New(); - if (ns->ns_dict == NULL) { - Py_DECREF(ns); - return NULL; + assert(type != NULL && type->tp_alloc != NULL); + self = type->tp_alloc(type, 0); + if (self != NULL) { + _PyNamespaceObject *ns = (_PyNamespaceObject *)self; + ns->ns_dict = PyDict_New(); + if (ns->ns_dict == NULL) { + Py_DECREF(ns); + return NULL; + } } - - PyObject_GC_Track(ns); - return (PyObject *)ns; + return self; }