Commit 357c9fb0 authored by Brett Cannon's avatar Brett Cannon

Rename importlib.util.ModuleManager to module_to_load so that the name

explains better what the context manager is providing.
parent 335ab5b6
......@@ -789,12 +789,13 @@ an :term:`importer`.
.. versionadded:: 3.3
.. class:: ModuleManager(name)
.. function:: module_to_load(name)
A :term:`context manager` which provides the module to load. The module will
either come from :attr:`sys.modules` in the case of reloading or a fresh
module if loading a new module. Proper cleanup of :attr:`sys.modules` occurs
if the module was new and an exception was raised.
Returns a :term:`context manager` which provides the module to load. The
module will either come from :attr:`sys.modules` in the case of reloading or
a fresh module if loading a new module. Proper cleanup of
:attr:`sys.modules` occurs if the module was new and an exception was
raised.
.. versionadded:: 3.4
......@@ -823,7 +824,7 @@ an :term:`importer`.
in :data:`sys.modules` then it is left alone.
.. note::
:class:`ModuleManager` subsumes the module management aspect of this
:func:`module_to_load` subsumes the module management aspect of this
decorator.
.. versionchanged:: 3.3
......
......@@ -484,7 +484,8 @@ def _verbose_message(message, *args, verbosity=1):
print(message.format(*args), file=sys.stderr)
class ModuleManager:
# Written as a class only because contextlib is not available.
class _ModuleManager:
"""Context manager which returns the module to be loaded.
......@@ -516,6 +517,12 @@ class ModuleManager:
del sys.modules[self._name]
def module_to_load(name):
"""Return a context manager which provides the module object to load."""
# Hiding _ModuleManager behind a function for better naming.
return _ModuleManager(name)
def set_package(fxn):
"""Set __package__ on the returned module."""
def set_package_wrapper(*args, **kwargs):
......@@ -559,7 +566,7 @@ def module_for_loader(fxn):
"""
def module_for_loader_wrapper(self, fullname, *args, **kwargs):
with ModuleManager(fullname) as module:
with module_to_load(fullname) as module:
module.__loader__ = self
try:
is_package = self.is_package(fullname)
......
"""Utility code for constructing importers, etc."""
from ._bootstrap import ModuleManager
from ._bootstrap import module_to_load
from ._bootstrap import module_for_loader
from ._bootstrap import set_loader
from ._bootstrap import set_package
......
......@@ -7,7 +7,7 @@ import types
import unittest
class ModuleManagerTests(unittest.TestCase):
class ModuleToLoadTests(unittest.TestCase):
module_name = 'ModuleManagerTest_module'
......@@ -19,7 +19,7 @@ class ModuleManagerTests(unittest.TestCase):
# Test a new module is created, inserted into sys.modules, has
# __initializing__ set to True after entering the context manager,
# and __initializing__ set to False after exiting.
with util.ModuleManager(self.module_name) as module:
with util.module_to_load(self.module_name) as module:
self.assertIn(self.module_name, sys.modules)
self.assertIs(sys.modules[self.module_name], module)
self.assertTrue(module.__initializing__)
......@@ -28,19 +28,19 @@ class ModuleManagerTests(unittest.TestCase):
def test_new_module_failed(self):
# Test the module is removed from sys.modules.
try:
with util.ModuleManager(self.module_name) as module:
with util.module_to_load(self.module_name) as module:
self.assertIn(self.module_name, sys.modules)
raise exception
except Exception:
self.assertNotIn(self.module_name, sys.modules)
else:
self.fail('importlib.util.ModuleManager swallowed an exception')
self.fail('importlib.util.module_to_load swallowed an exception')
def test_reload(self):
# Test that the same module is in sys.modules.
created_module = imp.new_module(self.module_name)
sys.modules[self.module_name] = created_module
with util.ModuleManager(self.module_name) as module:
with util.module_to_load(self.module_name) as module:
self.assertIs(module, created_module)
def test_reload_failed(self):
......@@ -48,12 +48,12 @@ class ModuleManagerTests(unittest.TestCase):
created_module = imp.new_module(self.module_name)
sys.modules[self.module_name] = created_module
try:
with util.ModuleManager(self.module_name) as module:
with util.module_to_load(self.module_name) as module:
raise Exception
except Exception:
self.assertIn(self.module_name, sys.modules)
else:
self.fail('importlib.util.ModuleManager swallowed an exception')
self.fail('importlib.util.module_to_load swallowed an exception')
class ModuleForLoaderTests(unittest.TestCase):
......
......@@ -103,8 +103,8 @@ Library
- Issue #18070: Have importlib.util.module_for_loader() set attributes
unconditionally in order to properly support reloading.
- Add importlib.util.ModuleManager as a context manager to provide the proper
module object to load.
- Add importlib.util.module_to_load to return a context manager to provide the
proper module object to load.
- Issue #18025: Fixed a segfault in io.BufferedIOBase.readinto() when raw
stream's read() returns more bytes than requested.
......
This diff is collapsed.
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment