mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-09-06 02:49:35 +02:00
45 lines
1.3 KiB
Python
45 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
|