mirror of
https://github.com/l1ving/youtube-dl
synced 2025-03-15 01:57:16 +08:00
edit __init__.py to handle suggestions
This commit is contained in:
parent
d47e2518f0
commit
1452a3a959
@ -12,11 +12,13 @@ IEs = [
|
|||||||
|
|
||||||
|
|
||||||
IE_list = []
|
IE_list = []
|
||||||
|
IE_dict = {}
|
||||||
for module, IE_names in IEs:
|
for module, IE_names in IEs:
|
||||||
_mod = __import__('youtube_dl.InfoExtractors.' + module,
|
_mod = __import__('youtube_dl.InfoExtractors.' + module, globals(), IE_names)
|
||||||
globals(), locals(), IE_names)
|
|
||||||
for IE_name in 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():
|
def gen_extractors():
|
||||||
@ -25,4 +27,4 @@ def gen_extractors():
|
|||||||
|
|
||||||
def get_info_extractor(IE_name):
|
def get_info_extractor(IE_name):
|
||||||
"""Returns the info extractor class with the given 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]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user