mirror of https://github.com/python/cpython.git
Surprising fix for SF bug 563060: module can be used as base class.
Change the module constructor (module_init) to have the signature __init__(name:str, doc=None); this prevents the call from type_new() to succeed. While we're at it, prevent repeated calling of module_init for the same module from leaking the dict, changing the semantics so that __dict__ is only initialized if NULL. Also adding a unittest, test_module.py. This is an incompatibility with 2.2, if anybody was instantiating the module class before, their argument list was probably empty; so this can't be backported to 2.2.x.
This commit is contained in:
parent
88f72ff955
commit
c3a787e090
|
@ -0,0 +1,48 @@
|
||||||
|
# Test the module type
|
||||||
|
|
||||||
|
from test_support import verify, vereq, verbose, TestFailed
|
||||||
|
|
||||||
|
import sys
|
||||||
|
module = type(sys)
|
||||||
|
|
||||||
|
# An uninitialized module has no __dict__ or __name__, and __doc__ is None
|
||||||
|
foo = module.__new__(module)
|
||||||
|
verify(foo.__dict__ is None)
|
||||||
|
try:
|
||||||
|
s = foo.__name__
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
raise TestFailed, "__name__ = %s" % repr(s)
|
||||||
|
vereq(foo.__doc__, None)
|
||||||
|
|
||||||
|
# Regularly initialized module, no docstring
|
||||||
|
foo = module("foo")
|
||||||
|
vereq(foo.__name__, "foo")
|
||||||
|
vereq(foo.__doc__, None)
|
||||||
|
vereq(foo.__dict__, {"__name__": "foo", "__doc__": None})
|
||||||
|
|
||||||
|
# ASCII docstring
|
||||||
|
foo = module("foo", "foodoc")
|
||||||
|
vereq(foo.__name__, "foo")
|
||||||
|
vereq(foo.__doc__, "foodoc")
|
||||||
|
vereq(foo.__dict__, {"__name__": "foo", "__doc__": "foodoc"})
|
||||||
|
|
||||||
|
# Unicode docstring
|
||||||
|
foo = module("foo", u"foodoc\u1234")
|
||||||
|
vereq(foo.__name__, "foo")
|
||||||
|
vereq(foo.__doc__, u"foodoc\u1234")
|
||||||
|
vereq(foo.__dict__, {"__name__": "foo", "__doc__": u"foodoc\u1234"})
|
||||||
|
|
||||||
|
# Reinitialization should not replace the __dict__
|
||||||
|
foo.bar = 42
|
||||||
|
d = foo.__dict__
|
||||||
|
foo.__init__("foo", "foodoc")
|
||||||
|
vereq(foo.__name__, "foo")
|
||||||
|
vereq(foo.__doc__, "foodoc")
|
||||||
|
vereq(foo.bar, 42)
|
||||||
|
vereq(foo.__dict__, {"__name__": "foo", "__doc__": "foodoc", "bar": 42})
|
||||||
|
verify(foo.__dict__ is d)
|
||||||
|
|
||||||
|
if verbose:
|
||||||
|
print "All OK"
|
|
@ -147,10 +147,23 @@ _PyModule_Clear(PyObject *m)
|
||||||
/* Methods */
|
/* Methods */
|
||||||
|
|
||||||
static int
|
static int
|
||||||
module_init(PyModuleObject *m, PyObject *args, PyObject *kw)
|
module_init(PyModuleObject *m, PyObject *args, PyObject *kwds)
|
||||||
{
|
{
|
||||||
m->md_dict = PyDict_New();
|
static char *kwlist[] = {"name", "doc", NULL};
|
||||||
if (m->md_dict == NULL)
|
PyObject *dict, *name = Py_None, *doc = Py_None;
|
||||||
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "S|O", kwlist,
|
||||||
|
&name, &doc))
|
||||||
|
return -1;
|
||||||
|
dict = m->md_dict;
|
||||||
|
if (dict == NULL) {
|
||||||
|
dict = PyDict_New();
|
||||||
|
if (dict == NULL)
|
||||||
|
return -1;
|
||||||
|
m->md_dict = dict;
|
||||||
|
}
|
||||||
|
if (PyDict_SetItemString(dict, "__name__", name) < 0)
|
||||||
|
return -1;
|
||||||
|
if (PyDict_SetItemString(dict, "__doc__", doc) < 0)
|
||||||
return -1;
|
return -1;
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue