yt-dlp/yt_dlp/extractor/__init__.py

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