From 93412126422b1324e920dc5097ee57c3ad11371b Mon Sep 17 00:00:00 2001 From: =?utf8?q?Jaime=20Marqui=CC=81nez=20Ferra=CC=81ndiz?= Date: Sat, 20 Apr 2013 12:42:57 +0200 Subject: [PATCH] Create a function in InfoExtractors that returns the InfoExtractor class with the given name --- test/test_download.py | 2 +- youtube_dl/InfoExtractors.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) 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'] -- 2.22.2