youtube-dl

Another place where youtube-dl lives on
git clone git://git.oshgnacknak.de/youtube-dl.git
Log | Files | Refs | README | LICENSE

testurl.py (2111B)

      1 from __future__ import unicode_literals
      2 
      3 import re
      4 
      5 from .common import InfoExtractor
      6 from ..utils import ExtractorError
      7 
      8 
      9 class TestURLIE(InfoExtractor):
     10     """ Allows addressing of the test cases as test:yout.*be_1 """
     11 
     12     IE_DESC = False  # Do not list
     13     _VALID_URL = r'test(?:url)?:(?P<id>(?P<extractor>.+?)(?:_(?P<num>[0-9]+))?)$'
     14 
     15     def _real_extract(self, url):
     16         from ..extractor import gen_extractors
     17 
     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()
     22 
     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)]
     26 
     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]
     44 
     45         num_str = mobj.group('num')
     46         num = int(num_str) if num_str else 0
     47 
     48         testcases = []
     49         t = getattr(extractor, '_TEST', None)
     50         if t:
     51             testcases.append(t)
     52         testcases.extend(getattr(extractor, '_TESTS', []))
     53 
     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)
     61 
     62         self.to_screen('Test URL: %s' % tc['url'])
     63 
     64         return self.url_result(tc['url'], video_id=video_id)