external/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-10 09:07:58 +01:00
Code Issues Projects Releases Packages Wiki Activity

Set the 'extractor_key' field in the info_dict

Browse source 
It's the string returned by the class method 'ie_key', which allows to retrieve the extractor with 'get_info_extractor'
This commit is contained in:
Jaime Marquínez Ferrándiz 2013-11-03 12:14:44 +01:00
parent 9103bbc5cd
commit be97abc247
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
test/test_download.py
View file

@ -149,7 +149,7 @@ def try_rm_tcs_files():
for key in ('id', 'url', 'title', 'ext'): for key in ('id', 'url', 'title', 'ext'):
self.assertTrue(key in info_dict.keys() and info_dict[key]) self.assertTrue(key in info_dict.keys() and info_dict[key])
# Check for mandatory fields that are automatically set by YoutubeDL # Check for mandatory fields that are automatically set by YoutubeDL
for key in ['webpage_url', 'extractor']: for key in ['webpage_url', 'extractor', 'extractor_key']:
self.assertTrue(info_dict.get(key), u'Missing field: %s' % key) self.assertTrue(info_dict.get(key), u'Missing field: %s' % key)
finally: finally:
try_rm_tcs_files() try_rm_tcs_files()

5
youtube_dl/YoutubeDL.py
View file

@ -357,7 +357,8 @@ def extract_info(self, url, download=True, ie_key=None, extra_info={}):
self.add_extra_info(ie_result, self.add_extra_info(ie_result,
{ {
'extractor': ie.IE_NAME, 'extractor': ie.IE_NAME,
'webpage_url': url 'webpage_url': url,
'extractor_key': ie.ie_key(),
}) })
return self.process_ie_result(ie_result, download, extra_info) return self.process_ie_result(ie_result, download, extra_info)
except ExtractorError as de: # An error we somewhat expected except ExtractorError as de: # An error we somewhat expected
@ -421,6 +422,7 @@ def process_ie_result(self, ie_result, download=True, extra_info={}):
'playlist_index': i + playliststart, 'playlist_index': i + playliststart,
'extractor': ie_result['extractor'], 'extractor': ie_result['extractor'],
'webpage_url': ie_result['webpage_url'], 'webpage_url': ie_result['webpage_url'],
'extractor_key': ie_result['extractor_key'],
} }
entry_result = self.process_ie_result(entry, entry_result = self.process_ie_result(entry,
download=download, download=download,
@ -434,6 +436,7 @@ def _fixup(r):
{ {
'extractor': ie_result['extractor'], 'extractor': ie_result['extractor'],
'webpage_url': ie_result['webpage_url'], 'webpage_url': ie_result['webpage_url'],
'extractor_key': ie_result['extractor_key'],
}) })
return r return r
ie_result['entries'] = [ ie_result['entries'] = [