summaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorJaime Marquínez Ferrándiz <[email protected]>2013-11-03 12:14:44 +0100
committerJaime Marquínez Ferrándiz <[email protected]>2013-11-03 12:14:44 +0100
commitbe97abc247d26bc36d1ef8cad5c17fc2a99d9101 (patch)
treee08910ea76394e06104e77812d9df13190da2f02
parent9103bbc5cd11957de2e906e4401dcf4df9511d28 (diff)
downloadyoutube-dl-be97abc247d26bc36d1ef8cad5c17fc2a99d9101.tar.gz
youtube-dl-be97abc247d26bc36d1ef8cad5c17fc2a99d9101.zip
Set the 'extractor_key' field in the info_dict
It's the string returned by the class method 'ie_key', which allows to retrieve the extractor with 'get_info_extractor'
-rw-r--r--test/test_download.py2
-rw-r--r--youtube_dl/YoutubeDL.py5
2 files changed, 5 insertions, 2 deletions
diff --git a/test/test_download.py b/test/test_download.py
index d6cc9ec33..73379beb1 100644
--- a/test/test_download.py
+++ b/test/test_download.py
@@ -149,7 +149,7 @@ def generator(test_case):
for key in ('id', 'url', 'title', 'ext'):
self.assertTrue(key in info_dict.keys() and info_dict[key])
# 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)
finally:
try_rm_tcs_files()
diff --git a/youtube_dl/YoutubeDL.py b/youtube_dl/YoutubeDL.py
index 8938a2cd3..86a6fd043 100644
--- a/youtube_dl/YoutubeDL.py
+++ b/youtube_dl/YoutubeDL.py
@@ -357,7 +357,8 @@ class YoutubeDL(object):
self.add_extra_info(ie_result,
{
'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)
except ExtractorError as de: # An error we somewhat expected
@@ -421,6 +422,7 @@ class YoutubeDL(object):
'playlist_index': i + playliststart,
'extractor': ie_result['extractor'],
'webpage_url': ie_result['webpage_url'],
+ 'extractor_key': ie_result['extractor_key'],
}
entry_result = self.process_ie_result(entry,
download=download,
@@ -434,6 +436,7 @@ class YoutubeDL(object):
{
'extractor': ie_result['extractor'],
'webpage_url': ie_result['webpage_url'],
+ 'extractor_key': ie_result['extractor_key'],
})
return r
ie_result['entries'] = [