]> jfr.im git - yt-dlp.git/blobdiff - devscripts/lazy_load_template.py
[extractor] Framework for embed detection (#4307)
[yt-dlp.git] / devscripts / lazy_load_template.py
index cdafaf1ef6799dedc2bac12f97582411da530d87..a6e26b6f63c33a1c044eb90bedb1cab872c5f1b1 100644 (file)
@@ -9,11 +9,13 @@
     write_string,
 )
 
+# These bloat the lazy_extractors, so allow them to passthrough silently
+ALLOWED_CLASSMETHODS = {'get_testcases', 'extract_from_webpage'}
+
 
 class LazyLoadMetaClass(type):
     def __getattr__(cls, name):
-        # "_TESTS" bloat the lazy_extractors
-        if '_real_class' not in cls.__dict__ and name != 'get_testcases':
+        if '_real_class' not in cls.__dict__ and name not in ALLOWED_CLASSMETHODS:
             write_string(
                 'WARNING: Falling back to normal extractor since lazy extractor '
                 f'{cls.__name__} does not have attribute {name}{bug_reports_message()}\n')