Fix for SF bug #1029475 : reload() doesn't work with PEP 302 loaders.
This commit is contained in:
parent
f4aca755bc
commit
7ec642a4d2
|
@ -12,7 +12,13 @@ def get_file():
|
|||
return __file__
|
||||
"""
|
||||
|
||||
reload_src = test_src+"""\
|
||||
reloaded = True
|
||||
"""
|
||||
|
||||
test_co = compile(test_src, "<???>", "exec")
|
||||
reload_co = compile(reload_src, "<???>", "exec")
|
||||
|
||||
test_path = "!!!_test_!!!"
|
||||
|
||||
|
||||
|
@ -32,6 +38,7 @@ class TestImporter:
|
|||
"hooktestpackage": (True, test_co),
|
||||
"hooktestpackage.sub": (True, test_co),
|
||||
"hooktestpackage.sub.subber": (False, test_co),
|
||||
"reloadmodule": (False, test_co),
|
||||
}
|
||||
|
||||
def __init__(self, path=test_path):
|
||||
|
@ -52,8 +59,7 @@ class TestImporter:
|
|||
|
||||
def load_module(self, fullname):
|
||||
ispkg, code = self.modules[fullname]
|
||||
mod = imp.new_module(fullname)
|
||||
sys.modules[fullname] = mod
|
||||
mod = sys.modules.setdefault(fullname,imp.new_module(fullname))
|
||||
mod.__file__ = "<%s>" % self.__class__.__name__
|
||||
mod.__loader__ = self
|
||||
if ispkg:
|
||||
|
@ -163,6 +169,14 @@ class ImportHooksTestCase(ImportHooksBaseTestCase):
|
|||
self.assertEqual(hooktestpackage.sub.__loader__, importer)
|
||||
self.assertEqual(hooktestpackage.sub.subber.__loader__, importer)
|
||||
|
||||
TestImporter.modules['reloadmodule'] = (False, test_co)
|
||||
import reloadmodule
|
||||
self.failIf(hasattr(reloadmodule,'reloaded'))
|
||||
|
||||
TestImporter.modules['reloadmodule'] = (False, reload_co)
|
||||
reload(reloadmodule)
|
||||
self.failUnless(hasattr(reloadmodule,'reloaded'))
|
||||
|
||||
def testMetaPath(self):
|
||||
i = MetaImporter()
|
||||
sys.meta_path.append(i)
|
||||
|
|
|
@ -2252,7 +2252,7 @@ PyObject *
|
|||
PyImport_ReloadModule(PyObject *m)
|
||||
{
|
||||
PyObject *modules = PyImport_GetModuleDict();
|
||||
PyObject *path = NULL;
|
||||
PyObject *path = NULL, *loader = NULL;
|
||||
char *name, *subname;
|
||||
char buf[MAXPATHLEN+1];
|
||||
struct filedescr *fdp;
|
||||
|
@ -2295,11 +2295,17 @@ PyImport_ReloadModule(PyObject *m)
|
|||
PyErr_Clear();
|
||||
}
|
||||
buf[0] = '\0';
|
||||
fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, NULL);
|
||||
fdp = find_module(name, subname, path, buf, MAXPATHLEN+1, &fp, &loader);
|
||||
Py_XDECREF(path);
|
||||
if (fdp == NULL)
|
||||
|
||||
if (fdp == NULL) {
|
||||
Py_XDECREF(loader);
|
||||
return NULL;
|
||||
newm = load_module(name, fp, buf, fdp->type, NULL);
|
||||
}
|
||||
|
||||
newm = load_module(name, fp, buf, fdp->type, loader);
|
||||
Py_XDECREF(loader);
|
||||
|
||||
if (fp)
|
||||
fclose(fp);
|
||||
if (newm == NULL) {
|
||||
|
|
Loading…
Reference in New Issue