]> jfr.im git - yt-dlp.git/blobdiff - devscripts/lazy_load_template.py
[docs] Improve manpage format (#2003)
[yt-dlp.git] / devscripts / lazy_load_template.py
index 2e6e6641b8e385beba9a9b742125eb1a4d1e0cc3..da89e070de493e00924697a123b2920735a8c96f 100644 (file)
@@ -1,19 +1,31 @@
-# encoding: utf-8
-from __future__ import unicode_literals
-
+# coding: utf-8
 import re
 
+from ..utils import bug_reports_message, write_string
+
+
+class LazyLoadMetaClass(type):
+    def __getattr__(cls, name):
+        if '_real_class' not in cls.__dict__:
+            write_string(
+                f'WARNING: Falling back to normal extractor since lazy extractor '
+                f'{cls.__name__} does not have attribute {name}{bug_reports_message()}')
+        return getattr(cls._get_real_class(), name)
+
 
-class LazyLoadExtractor(object):
+class LazyLoadExtractor(metaclass=LazyLoadMetaClass):
     _module = None
+    _WORKING = True
 
     @classmethod
-    def ie_key(cls):
-        return cls.__name__[:-2]
+    def _get_real_class(cls):
+        if '_real_class' not in cls.__dict__:
+            mod = __import__(cls._module, fromlist=(cls.__name__,))
+            cls._real_class = getattr(mod, cls.__name__)
+        return cls._real_class
 
     def __new__(cls, *args, **kwargs):
-        mod = __import__(cls._module, fromlist=(cls.__name__,))
-        real_cls = getattr(mod, cls.__name__)
+        real_cls = cls._get_real_class()
         instance = real_cls.__new__(real_cls)
         instance.__init__(*args, **kwargs)
         return instance