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

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']