summaryrefslogtreecommitdiffstats
path: root/Lib/importlib
diff options
context:
space:
mode:
authorMeador Inge <meadori@gmail.com>2011-12-15 04:23:46 (GMT)
committerMeador Inge <meadori@gmail.com>2011-12-15 04:23:46 (GMT)
commit416f12ddb3b7d780bb75563414b88b32c0cf7e67 (patch)
treecbe4c2d385714638e0bda1c2d60289dc3182f5a4 /Lib/importlib
parent061c0289af979213984046c0f7570192063ab319 (diff)
downloadcpython-416f12ddb3b7d780bb75563414b88b32c0cf7e67.zip
cpython-416f12ddb3b7d780bb75563414b88b32c0cf7e67.tar.gz
cpython-416f12ddb3b7d780bb75563414b88b32c0cf7e67.tar.bz2
Issue #13591: import_module potentially imports a module twice.
Diffstat (limited to 'Lib/importlib')
-rw-r--r--Lib/importlib/_bootstrap.py4
-rw-r--r--Lib/importlib/test/test_api.py17
-rw-r--r--Lib/importlib/test/util.py7
3 files changed, 26 insertions, 2 deletions
diff --git a/Lib/importlib/_bootstrap.py b/Lib/importlib/_bootstrap.py
index 425b8bf..90eb1a7 100644
--- a/Lib/importlib/_bootstrap.py
+++ b/Lib/importlib/_bootstrap.py
@@ -816,7 +816,9 @@ def _gcd_import(name, package=None, level=0):
for finder in meta_path:
loader = finder.find_module(name, path)
if loader is not None:
- loader.load_module(name)
+ # The parent import may have already imported this module.
+ if name not in sys.modules:
+ loader.load_module(name)
break
else:
raise ImportError(_ERR_MSG.format(name))
diff --git a/Lib/importlib/test/test_api.py b/Lib/importlib/test/test_api.py
index 0ffa3c4..a151626 100644
--- a/Lib/importlib/test/test_api.py
+++ b/Lib/importlib/test/test_api.py
@@ -67,6 +67,23 @@ class ImportModuleTests(unittest.TestCase):
importlib.import_module('.support')
+ def test_loaded_once(self):
+ # Issue #13591: Modules should only be loaded once when
+ # initializing the parent package attempts to import the
+ # module currently being imported.
+ b_load_count = 0
+ def load_a():
+ importlib.import_module('a.b')
+ def load_b():
+ nonlocal b_load_count
+ b_load_count += 1
+ code = {'a': load_a, 'a.b': load_b}
+ modules = ['a.__init__', 'a.b']
+ with util.mock_modules(*modules, module_code=code) as mock:
+ with util.import_state(meta_path=[mock]):
+ importlib.import_module('a.b')
+ self.assertEqual(b_load_count, 1)
+
def test_main():
from test.support import run_unittest
run_unittest(ImportModuleTests)
diff --git a/Lib/importlib/test/util.py b/Lib/importlib/test/util.py
index 0c0c84c..93b7cd2 100644
--- a/Lib/importlib/test/util.py
+++ b/Lib/importlib/test/util.py
@@ -84,8 +84,9 @@ class mock_modules:
"""A mock importer/loader."""
- def __init__(self, *names):
+ def __init__(self, *names, module_code={}):
self.modules = {}
+ self.module_code = {}
for name in names:
if not name.endswith('.__init__'):
import_name = name
@@ -105,6 +106,8 @@ class mock_modules:
if import_name != name:
module.__path__ = ['<mock __path__>']
self.modules[import_name] = module
+ if import_name in module_code:
+ self.module_code[import_name] = module_code[import_name]
def __getitem__(self, name):
return self.modules[name]
@@ -120,6 +123,8 @@ class mock_modules:
raise ImportError
else:
sys.modules[fullname] = self.modules[fullname]
+ if fullname in self.module_code:
+ self.module_code[fullname]()
return self.modules[fullname]
def __enter__(self):