summaryrefslogtreecommitdiffstats
path: root/devscripts/lazy_load_template.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-15 16:49:24 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-15 16:49:24 +0000
commit2415e66f889f38503b73e8ebc5f43ca342390e5c (patch)
treeac48ab69d1d96bae3d83756134921e0d90593aa5 /devscripts/lazy_load_template.py
parentInitial commit. (diff)
downloadyt-dlp-2415e66f889f38503b73e8ebc5f43ca342390e5c.tar.xz
yt-dlp-2415e66f889f38503b73e8ebc5f43ca342390e5c.zip
Adding upstream version 2024.03.10.upstream/2024.03.10
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'devscripts/lazy_load_template.py')
-rw-r--r--devscripts/lazy_load_template.py39
1 files changed, 39 insertions, 0 deletions
diff --git a/devscripts/lazy_load_template.py b/devscripts/lazy_load_template.py
new file mode 100644
index 0000000..6f52165
--- /dev/null
+++ b/devscripts/lazy_load_template.py
@@ -0,0 +1,39 @@
+import importlib
+import random
+import re
+
+from ..utils import (
+ age_restricted,
+ bug_reports_message,
+ classproperty,
+ variadic,
+ write_string,
+)
+
+# These bloat the lazy_extractors, so allow them to passthrough silently
+ALLOWED_CLASSMETHODS = {'extract_from_webpage', 'get_testcases', 'get_webpage_testcases'}
+_WARNED = False
+
+
+class LazyLoadMetaClass(type):
+ def __getattr__(cls, name):
+ global _WARNED
+ if ('_real_class' not in cls.__dict__
+ and name not in ALLOWED_CLASSMETHODS and not _WARNED):
+ _WARNED = True
+ write_string('WARNING: Falling back to normal extractor since lazy extractor '
+ f'{cls.__name__} does not have attribute {name}{bug_reports_message()}\n')
+ return getattr(cls.real_class, name)
+
+
+class LazyLoadExtractor(metaclass=LazyLoadMetaClass):
+ @classproperty
+ def real_class(cls):
+ if '_real_class' not in cls.__dict__:
+ cls._real_class = getattr(importlib.import_module(cls._module), cls.__name__)
+ return cls._real_class
+
+ def __new__(cls, *args, **kwargs):
+ instance = cls.real_class.__new__(cls.real_class)
+ instance.__init__(*args, **kwargs)
+ return instance