1
0
mirror of https://github.com/l1ving/youtube-dl synced 2025-03-14 06:37:17 +08:00

edit __init__.py to handle suggestions

This commit is contained in:
Filippo Valsorda 2013-06-05 12:18:14 +02:00
parent d47e2518f0
commit 1452a3a959

View File

@ -12,11 +12,13 @@ IEs = [
IE_list = []
IE_dict = {}
for module, IE_names in IEs:
_mod = __import__('youtube_dl.InfoExtractors.' + module,
globals(), locals(), IE_names)
_mod = __import__('youtube_dl.InfoExtractors.' + module, globals(), IE_names)
for IE_name in IE_names:
IE_list.append(getattr(_mod, IE_name))
IE = getattr(_mod, IE_name)
IE_list.append(IE)
IE_dict[IE_name] = IE
def gen_extractors():
@ -25,4 +27,4 @@ def gen_extractors():
def get_info_extractor(IE_name):
"""Returns the info extractor class with the given ie_name"""
return next(IE for IE in IE_list if IE.__name__ == IE_name + 'IE')
return IE_dict[IE_name]