]>
Commit | Line | Data |
---|---|---|
a9c7198a PH |
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): | |
dfb1b146 | 10 | """ Allows addressing of the test cases as test:yout.*be_1 """ |
a9c7198a PH |
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) | |
4d756a9c PH |
42 | else: |
43 | extractor = matching_extractors[0] | |
a9c7198a PH |
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 | ||
d226c560 | 64 | return self.url_result(tc['url'], video_id=video_id) |