mirror of https://github.com/yt-dlp/yt-dlp
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
from .._globals import extractors as _extractors_context
|
|
from .._globals import plugin_ies as _plugin_ies_context
|
|
from ..compat.compat_utils import passthrough_module
|
|
from ..plugins import PluginSpec, register_plugin_spec
|
|
|
|
passthrough_module(__name__, '.extractors')
|
|
del passthrough_module
|
|
|
|
register_plugin_spec(PluginSpec(
|
|
module_name='extractor',
|
|
suffix='IE',
|
|
destination=_extractors_context,
|
|
plugin_destination=_plugin_ies_context,
|
|
))
|
|
|
|
|
|
def gen_extractor_classes():
|
|
""" Return a list of supported extractors.
|
|
The order does matter; the first extractor matched is the one handling the URL.
|
|
"""
|
|
import_extractors()
|
|
return list(_extractors_context.value.values())
|
|
|
|
|
|
def gen_extractors():
|
|
""" Return a list of an instance of every supported extractor.
|
|
The order does matter; the first extractor matched is the one handling the URL.
|
|
"""
|
|
return [klass() for klass in gen_extractor_classes()]
|
|
|
|
|
|
def list_extractor_classes(age_limit=None):
|
|
"""Return a list of extractors that are suitable for the given age, sorted by extractor name"""
|
|
from .generic import GenericIE
|
|
|
|
yield from sorted(filter(
|
|
lambda ie: ie.is_suitable(age_limit) and ie != GenericIE,
|
|
gen_extractor_classes()), key=lambda ie: ie.IE_NAME.lower())
|
|
yield GenericIE
|
|
|
|
|
|
def list_extractors(age_limit=None):
|
|
"""Return a list of extractor instances that are suitable for the given age, sorted by extractor name"""
|
|
return [ie() for ie in list_extractor_classes(age_limit)]
|
|
|
|
|
|
def get_info_extractor(ie_name):
|
|
"""Returns the info extractor class with the given ie_name"""
|
|
import_extractors()
|
|
return _extractors_context.value[f'{ie_name}IE']
|
|
|
|
|
|
def import_extractors():
|
|
from . import extractors # noqa: F401
|