yt_dlp/compat/compat_utils.py

   1 import collections
   2 import contextlib
   3 import importlib
   4 import sys
   5 import types
   6
   7
   8 _NO_ATTRIBUTE = object()
   9
  10 _Package = collections.namedtuple('Package', ('name', 'version'))
  11
  12
  13 def get_package_info(module):
  14     parent = module.__name__.split('.')[0]
  15     parent_module = None
  16     with contextlib.suppress(ImportError):
  17         parent_module = importlib.import_module(parent)
  18
  19     for attr in ('__version__', 'version_string', 'version'):
  20         version = getattr(parent_module, attr, None)
  21         if version is not None:
  22             break
  23     return _Package(getattr(module, '_yt_dlp__identifier', parent), str(version))
  24
  25
  26 def _is_package(module):
  27     try:
  28         module.__getattribute__('__path__')
  29     except AttributeError:
  30         return False
  31     return True
  32
  33
  34 def passthrough_module(parent, child, *, callback=lambda _: None):
  35     parent_module = importlib.import_module(parent)
  36     child_module = importlib.import_module(child, parent)
  37
  38     class PassthroughModule(types.ModuleType):
  39         def __getattr__(self, attr):
  40             if _is_package(parent_module):
  41                 with contextlib.suppress(ImportError):
  42                     return importlib.import_module(f'.{attr}', parent)
  43
  44             ret = _NO_ATTRIBUTE
  45             with contextlib.suppress(AttributeError):
  46                 ret = getattr(child_module, attr)
  47
  48             if _is_package(child_module):
  49                 with contextlib.suppress(ImportError):
  50                     ret = importlib.import_module(f'.{attr}', child)
  51
  52             if ret is _NO_ATTRIBUTE:
  53                 raise AttributeError(f'module {parent} has no attribute {attr}')
  54
  55             callback(attr)
  56             return ret
  57
  58     # Python 3.6 does not have module level __getattr__
  59     # https://peps.python.org/pep-0562/
  60     sys.modules[parent].__class__ = PassthroughModule