Create a function in InfoExtractors that returns the InfoExtractor class with the given name

This commit is contained in:
Jaime Marquínez Ferrándiz 2013-04-20 12:42:57 +02:00
parent f7a9721e16
commit 9341212642
2 changed files with 5 additions and 1 deletions

View file

@ -67,7 +67,7 @@ class TestDownload(unittest.TestCase):
def generator(test_case): def generator(test_case):
def test_template(self): def test_template(self):
ie = getattr(youtube_dl.InfoExtractors, test_case['name'] + 'IE') ie = youtube_dl.InfoExtractors.get_info_extractor(test_case['name'])#getattr(youtube_dl.InfoExtractors, test_case['name'] + 'IE')
if not ie._WORKING: if not ie._WORKING:
print('Skipping: IE marked as not _WORKING') print('Skipping: IE marked as not _WORKING')
return return

View file

@ -4469,3 +4469,7 @@ def gen_extractors():
ARDIE(), ARDIE(),
GenericIE() GenericIE()
] ]
def get_info_extractor(ie_name):
"""Returns the info extractor class with the given ie_name"""
return globals()[ie_name+'IE']