]> jfr.im git - yt-dlp.git/blobdiff - devscripts/lazy_load_template.py
[cleanup] Fix misc bugs (#8968)
[yt-dlp.git] / devscripts / lazy_load_template.py
index 6d9b2774293e2bbbd990995787339742ccbe315b..6f52165c5c551fe58d37d63696eb7c9660d588c2 100644 (file)
@@ -2,16 +2,27 @@
 import random
 import re
 
-from ..utils import bug_reports_message, classproperty, write_string
+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):
-        # "is_suitable" requires "_TESTS". However, they bloat the lazy_extractors
-        if '_real_class' not in cls.__dict__ and name not in ('is_suitable', 'get_testcases'):
-            write_string(
-                'WARNING: Falling back to normal extractor since lazy extractor '
-                f'{cls.__name__} does not have attribute {name}{bug_reports_message()}\n')
+        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)