logo

youtube-dl

[mirror] Download/Watch videos from video hostersgit clone https://hacktivis.me/git/mirror/youtube-dl.git

testurl.py (2111B)

    

  1. from __future__ import unicode_literals
  3. import re
  5. from .common import InfoExtractor
  6. from ..utils import ExtractorError
  9. class TestURLIE(InfoExtractor):
  10.     """ Allows addressing of the test cases as test:yout.*be_1 """
  12.     IE_DESC = False  # Do not list
  13.     _VALID_URL = r'test(?:url)?:(?P<id>(?P<extractor>.+?)(?:_(?P<num>[0-9]+))?)$'
  15.     def _real_extract(self, url):
  16.         from ..extractor import gen_extractors
  18.         mobj = re.match(self._VALID_URL, url)
  19.         video_id = mobj.group('id')
  20.         extractor_id = mobj.group('extractor')
  21.         all_extractors = gen_extractors()
  23.         rex = re.compile(extractor_id, flags=re.IGNORECASE)
  24.         matching_extractors = [
  25.             e for e in all_extractors if rex.search(e.IE_NAME)]
  27.         if len(matching_extractors) == 0:
  28.             raise ExtractorError(
  29.                 'No extractors matching %r found' % extractor_id,
  30.                 expected=True)
  31.         elif len(matching_extractors) > 1:
  32.             # Is it obvious which one to pick?
  33.             try:
  34.                 extractor = next(
  35.                     ie for ie in matching_extractors
  36.                     if ie.IE_NAME.lower() == extractor_id.lower())
  37.             except StopIteration:
  38.                 raise ExtractorError(
  39.                     ('Found multiple matching extractors: %s' %
  40.                         ' '.join(ie.IE_NAME for ie in matching_extractors)),
  41.                     expected=True)
  42.         else:
  43.             extractor = matching_extractors[0]
  45.         num_str = mobj.group('num')
  46.         num = int(num_str) if num_str else 0
  48.         testcases = []
  49.         t = getattr(extractor, '_TEST', None)
  50.         if t:
  51.             testcases.append(t)
  52.         testcases.extend(getattr(extractor, '_TESTS', []))
  54.         try:
  55.             tc = testcases[num]
  56.         except IndexError:
  57.             raise ExtractorError(
  58.                 ('Test case %d not found, got only %d tests' %
  59.                     (num, len(testcases))),
  60.                 expected=True)
  62.         self.to_screen('Test URL: %s' % tc['url'])
  64.         return self.url_result(tc['url'], video_id=video_id)