Browse Source

[cnn] Improve thumbnail extraction

master
Philipp Hagemeister 10 years ago
parent
commit
be6d722904
2 changed files with 8 additions and 4 deletions
  1. 2
      youtube_dl/YoutubeDL.py
  2. 10
      youtube_dl/extractor/cnn.py

2
youtube_dl/YoutubeDL.py

@ -719,6 +719,8 @@ class YoutubeDL(object):
thumbnails = info_dict.get('thumbnails') thumbnails = info_dict.get('thumbnails')
if thumbnails: if thumbnails:
thumbnails.sort(key=lambda t: (
t.get('width'), t.get('height'), t.get('url')))
for t in thumbnails: for t in thumbnails:
if 'width' in t and 'height' in t: if 'width' in t and 'height' in t:
t['resolution'] = '%dx%d' % (t['width'], t['height']) t['resolution'] = '%dx%d' % (t['width'], t['height'])

10
youtube_dl/extractor/cnn.py

@ -79,8 +79,11 @@ class CNNIE(InfoExtractor):
self._sort_formats(formats) self._sort_formats(formats)
thumbnails = sorted([((int(t.attrib['height']),int(t.attrib['width'])), t.text) for t in info.findall('images/image')])
thumbs_dict = [{'resolution': res, 'url': t_url} for (res, t_url) in thumbnails]
thumbnails = [{
'height': int(t.attrib['height']),
'width': int(t.attrib['width']),
'url': t.text,
} for t in info.findall('images/image')]
metas_el = info.find('metas') metas_el = info.find('metas')
upload_date = ( upload_date = (
@ -93,8 +96,7 @@ class CNNIE(InfoExtractor):
'id': info.attrib['id'], 'id': info.attrib['id'],
'title': info.find('headline').text, 'title': info.find('headline').text,
'formats': formats, 'formats': formats,
'thumbnail': thumbnails[-1][1],
'thumbnails': thumbs_dict,
'thumbnails': thumbnails,
'description': info.find('description').text, 'description': info.find('description').text,
'duration': duration, 'duration': duration,
'upload_date': upload_date, 'upload_date': upload_date,

Loading…
Cancel
Save