diff options
| -rw-r--r-- | test/test_download.py | 2 | ||||
| -rwxr-xr-x | youtube_dl/InfoExtractors.py | 4 |
2 files changed, 5 insertions, 1 deletions
diff --git a/test/test_download.py b/test/test_download.py index e3513efba..cf8028718 100644 --- a/test/test_download.py +++ b/test/test_download.py @@ -67,7 +67,7 @@ class TestDownload(unittest.TestCase): def generator(test_case): 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: print('Skipping: IE marked as not _WORKING') return diff --git a/youtube_dl/InfoExtractors.py b/youtube_dl/InfoExtractors.py index 8bfb2809b..eeedcf792 100755 --- a/youtube_dl/InfoExtractors.py +++ b/youtube_dl/InfoExtractors.py @@ -4469,3 +4469,7 @@ def gen_extractors(): ARDIE(), GenericIE() ] + +def get_info_extractor(ie_name): + """Returns the info extractor class with the given ie_name""" + return globals()[ie_name+'IE'] |