44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import contextlib
|
|
import importlib
|
|
import sys
|
|
import types
|
|
|
|
|
|
def _is_package(module):
|
|
try:
|
|
module.__getattribute__('__path__')
|
|
except AttributeError:
|
|
return False
|
|
return True
|
|
|
|
|
|
_NO_ATTRIBUTE = object()
|
|
|
|
|
|
def passthrough_module(parent, child, *, callback=lambda _: None):
|
|
parent_module = importlib.import_module(parent)
|
|
child_module = importlib.import_module(child, parent)
|
|
|
|
class PassthroughModule(types.ModuleType):
|
|
def __getattr__(self, attr):
|
|
if _is_package(parent_module):
|
|
with contextlib.suppress(ImportError):
|
|
return importlib.import_module(f'.{attr}', parent)
|
|
|
|
ret = _NO_ATTRIBUTE
|
|
with contextlib.suppress(AttributeError):
|
|
ret = getattr(child_module, attr)
|
|
|
|
if _is_package(child_module):
|
|
with contextlib.suppress(ImportError):
|
|
ret = importlib.import_module(f'.{attr}', child)
|
|
|
|
if ret is _NO_ATTRIBUTE:
|
|
raise AttributeError(f'module {parent} has no attribute {attr}')
|
|
|
|
callback(attr)
|
|
return ret
|
|
|
|
# Python 3.6 does not have module level __getattr__
|
|
# https://peps.python.org/pep-0562/
|
|
sys.modules[parent].__class__ = PassthroughModule
|