_ Git - youtube-dl/blob - youtube_dl/extractor/amp.py

   1 # coding: utf-8
   2 from __future__ import unicode_literals
   3
   4 from .common import InfoExtractor
   5 from ..utils import (
   6     int_or_none,
   7     parse_iso8601,
   8     mimetype2ext,
   9     determine_ext,
  10     ExtractorError,
  11 )
  12
  13
  14 class AMPIE(InfoExtractor):
  15     # parse Akamai Adaptive Media Player feed
  16     def _extract_feed_info(self, url):
  17         feed = self._download_json(
  18             url, None, 'Downloading Akamai AMP feed',
  19             'Unable to download Akamai AMP feed')
  20         item = feed.get('channel', {}).get('item')
  21         if not item:
  22             raise ExtractorError('%s said: %s' % (self.IE_NAME, feed['error']))
  23
  24         video_id = item['guid']
  25
  26         def get_media_node(name, default=None):
  27             media_name = 'media-%s' % name
  28             media_group = item.get('media-group') or item
  29             return media_group.get(media_name) or item.get(media_name) or item.get(name, default)
  30
  31         thumbnails = []
  32         media_thumbnail = get_media_node('thumbnail')
  33         if media_thumbnail:
  34             if isinstance(media_thumbnail, dict):
  35                 media_thumbnail = [media_thumbnail]
  36             for thumbnail_data in media_thumbnail:
  37                 thumbnail = thumbnail_data['@attributes']
  38                 thumbnails.append({
  39                     'url': self._proto_relative_url(thumbnail['url'], 'http:'),
  40                     'width': int_or_none(thumbnail.get('width')),
  41                     'height': int_or_none(thumbnail.get('height')),
  42                 })
  43
  44         subtitles = {}
  45         media_subtitle = get_media_node('subTitle')
  46         if media_subtitle:
  47             if isinstance(media_subtitle, dict):
  48                 media_subtitle = [media_subtitle]
  49             for subtitle_data in media_subtitle:
  50                 subtitle = subtitle_data['@attributes']
  51                 lang = subtitle.get('lang') or 'en'
  52                 subtitles[lang] = [{'url': subtitle['href']}]
  53
  54         formats = []
  55         media_content = get_media_node('content')
  56         if isinstance(media_content, dict):
  57             media_content = [media_content]
  58         for media_data in media_content:
  59             media = media_data.get('@attributes', {})
  60             media_url = media.get('url')
  61             if not media_url:
  62                 continue
  63             ext = mimetype2ext(media.get('type')) or determine_ext(media_url)
  64             if ext == 'f4m':
  65                 formats.extend(self._extract_f4m_formats(
  66                     media_url + '?hdcore=3.4.0&plugin=aasp-3.4.0.132.124',
  67                     video_id, f4m_id='hds', fatal=False))
  68             elif ext == 'm3u8':
  69                 formats.extend(self._extract_m3u8_formats(
  70                     media_url, video_id, 'mp4', m3u8_id='hls', fatal=False))
  71             else:
  72                 formats.append({
  73                     'format_id': media_data.get('media-category', {}).get('@attributes', {}).get('label'),
  74                     'url': media['url'],
  75                     'tbr': int_or_none(media.get('bitrate')),
  76                     'filesize': int_or_none(media.get('fileSize')),
  77                     'ext': ext,
  78                 })
  79
  80         self._sort_formats(formats)
  81
  82         timestamp = parse_iso8601(item.get('pubDate'), ' ') or parse_iso8601(item.get('dc-date'))
  83
  84         return {
  85             'id': video_id,
  86             'title': get_media_node('title'),
  87             'description': get_media_node('description'),
  88             'thumbnails': thumbnails,
  89             'timestamp': timestamp,
  90             'duration': int_or_none(media_content[0].get('@attributes', {}).get('duration')),
  91             'subtitles': subtitles,
  92             'formats': formats,
  93         }