aboutsummaryrefslogtreecommitdiff
path: root/yt_dlp/compat/compat_utils.py
diff options
context:
space:
mode:
authorpukkandan <pukkandan.ytdlp@gmail.com>2023-02-09 01:34:39 +0530
committerpukkandan <pukkandan.ytdlp@gmail.com>2023-02-09 01:47:13 +0530
commit768a00178109508893488e53a0e720b117fbccf6 (patch)
tree102cb1d0816e91e0a0de0bc14b0c22e6025d0cfa /yt_dlp/compat/compat_utils.py
parentacb1042a9ffa8769fe691beac1011d6da1fcf321 (diff)
[compat_utils] Simplify `EnhancedModule`
Diffstat (limited to 'yt_dlp/compat/compat_utils.py')
-rw-r--r--yt_dlp/compat/compat_utils.py18
1 files changed, 2 insertions, 16 deletions
diff --git a/yt_dlp/compat/compat_utils.py b/yt_dlp/compat/compat_utils.py
index f8679c98e..8956b3bf1 100644
--- a/yt_dlp/compat/compat_utils.py
+++ b/yt_dlp/compat/compat_utils.py
@@ -28,20 +28,6 @@ def _is_dunder(name):
class EnhancedModule(types.ModuleType):
- def __new__(cls, name, *args, **kwargs):
- if name not in sys.modules:
- return super().__new__(cls, name, *args, **kwargs)
-
- assert not args and not kwargs, 'Cannot pass additional arguments to an existing module'
- module = sys.modules[name]
- module.__class__ = cls
- return module
-
- def __init__(self, name, *args, **kwargs):
- # Prevent __new__ from trigerring __init__ again
- if name not in sys.modules:
- super().__init__(name, *args, **kwargs)
-
def __bool__(self):
return vars(self).get('__bool__', lambda: True)()
@@ -60,8 +46,6 @@ class EnhancedModule(types.ModuleType):
def passthrough_module(parent, child, allowed_attributes=(..., ), *, callback=lambda _: None):
"""Passthrough parent module into a child module, creating the parent if necessary"""
- parent = EnhancedModule(parent)
-
def __getattr__(attr):
if _is_package(parent):
with contextlib.suppress(ImportError):
@@ -93,5 +77,7 @@ def passthrough_module(parent, child, allowed_attributes=(..., ), *, callback=la
return _NO_ATTRIBUTE
+ parent = sys.modules.get(parent, types.ModuleType(parent))
+ parent.__class__ = EnhancedModule
parent.__getattr__ = __getattr__
return parent