SolarFM/plugins/youtube_download/yt_dlp/extractor/testurl.py

51 lines
2.0 KiB
Python
Raw Normal View History

import re
from .common import InfoExtractor
from ..utils import ExtractorError
class TestURLIE(InfoExtractor):
""" Allows addressing of the test cases as test:yout.*be_1 """
IE_DESC = False # Do not list
2023-08-14 01:13:21 +00:00
_VALID_URL = r'test(?:url)?:(?P<extractor>.*?)(?:_(?P<num>\d+|all))?$'
def _real_extract(self, url):
2023-02-21 01:18:45 +00:00
from . import gen_extractor_classes
2023-02-21 01:18:45 +00:00
extractor_id, num = self._match_valid_url(url).group('extractor', 'num')
if not extractor_id:
return {'id': ':test', 'title': '', 'url': url}
rex = re.compile(extractor_id, flags=re.IGNORECASE)
2023-02-21 01:18:45 +00:00
matching_extractors = [e for e in gen_extractor_classes() if rex.search(e.IE_NAME)]
if len(matching_extractors) == 0:
2023-02-21 01:18:45 +00:00
raise ExtractorError(f'No extractors matching {extractor_id!r} found', expected=True)
elif len(matching_extractors) > 1:
2023-02-21 01:18:45 +00:00
extractor = next(( # Check for exact match
ie for ie in matching_extractors if ie.IE_NAME.lower() == extractor_id.lower()
), None) or next(( # Check for exact match without plugin suffix
ie for ie in matching_extractors if ie.IE_NAME.split('+')[0].lower() == extractor_id.lower()
), None)
if not extractor:
raise ExtractorError(
2023-02-21 01:18:45 +00:00
'Found multiple matching extractors: %s' % ' '.join(ie.IE_NAME for ie in matching_extractors),
expected=True)
else:
extractor = matching_extractors[0]
2023-02-21 01:18:45 +00:00
testcases = tuple(extractor.get_testcases(True))
2023-08-14 01:13:21 +00:00
if num == 'all':
return self.playlist_result(
[self.url_result(tc['url'], extractor) for tc in testcases],
url, f'{extractor.IE_NAME} tests')
try:
2023-02-21 01:18:45 +00:00
tc = testcases[int(num or 0)]
except IndexError:
raise ExtractorError(
2023-02-21 01:18:45 +00:00
f'Test case {num or 0} not found, got only {len(testcases)} tests', expected=True)
2023-02-21 01:18:45 +00:00
self.to_screen(f'Test URL: {tc["url"]}')
2023-08-14 01:13:21 +00:00
return self.url_result(tc['url'], extractor)