mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-17 15:25:49 +00:00
86e5f3ed2e
Using https://github.com/asottile/pyupgrade 1. `__future__` imports and `coding: utf-8` were removed 2. Files were rewritten with `pyupgrade --py36-plus --keep-percent-format` 3. f-strings were cherry-picked from `pyupgrade --py36-plus` Extractors are left untouched (except removing header) to avoid unnecessary merge conflicts
101 lines
3.9 KiB
Python
101 lines
3.9 KiB
Python
from .common import InfoExtractor
|
|
from ..utils import (
|
|
determine_ext,
|
|
ExtractorError,
|
|
int_or_none,
|
|
mimetype2ext,
|
|
parse_iso8601,
|
|
unified_timestamp,
|
|
url_or_none,
|
|
)
|
|
|
|
|
|
class AMPIE(InfoExtractor):
|
|
# parse Akamai Adaptive Media Player feed
|
|
def _extract_feed_info(self, url):
|
|
feed = self._download_json(
|
|
url, None, 'Downloading Akamai AMP feed',
|
|
'Unable to download Akamai AMP feed')
|
|
item = feed.get('channel', {}).get('item')
|
|
if not item:
|
|
raise ExtractorError('%s said: %s' % (self.IE_NAME, feed['error']))
|
|
|
|
video_id = item['guid']
|
|
|
|
def get_media_node(name, default=None):
|
|
media_name = 'media-%s' % name
|
|
media_group = item.get('media-group') or item
|
|
return media_group.get(media_name) or item.get(media_name) or item.get(name, default)
|
|
|
|
thumbnails = []
|
|
media_thumbnail = get_media_node('thumbnail')
|
|
if media_thumbnail:
|
|
if isinstance(media_thumbnail, dict):
|
|
media_thumbnail = [media_thumbnail]
|
|
for thumbnail_data in media_thumbnail:
|
|
thumbnail = thumbnail_data.get('@attributes', {})
|
|
thumbnail_url = url_or_none(thumbnail.get('url'))
|
|
if not thumbnail_url:
|
|
continue
|
|
thumbnails.append({
|
|
'url': self._proto_relative_url(thumbnail_url, 'http:'),
|
|
'width': int_or_none(thumbnail.get('width')),
|
|
'height': int_or_none(thumbnail.get('height')),
|
|
})
|
|
|
|
subtitles = {}
|
|
media_subtitle = get_media_node('subTitle')
|
|
if media_subtitle:
|
|
if isinstance(media_subtitle, dict):
|
|
media_subtitle = [media_subtitle]
|
|
for subtitle_data in media_subtitle:
|
|
subtitle = subtitle_data.get('@attributes', {})
|
|
subtitle_href = url_or_none(subtitle.get('href'))
|
|
if not subtitle_href:
|
|
continue
|
|
subtitles.setdefault(subtitle.get('lang') or 'en', []).append({
|
|
'url': subtitle_href,
|
|
'ext': mimetype2ext(subtitle.get('type')) or determine_ext(subtitle_href),
|
|
})
|
|
|
|
formats = []
|
|
media_content = get_media_node('content')
|
|
if isinstance(media_content, dict):
|
|
media_content = [media_content]
|
|
for media_data in media_content:
|
|
media = media_data.get('@attributes', {})
|
|
media_url = url_or_none(media.get('url'))
|
|
if not media_url:
|
|
continue
|
|
ext = mimetype2ext(media.get('type')) or determine_ext(media_url)
|
|
if ext == 'f4m':
|
|
formats.extend(self._extract_f4m_formats(
|
|
media_url + '?hdcore=3.4.0&plugin=aasp-3.4.0.132.124',
|
|
video_id, f4m_id='hds', fatal=False))
|
|
elif ext == 'm3u8':
|
|
formats.extend(self._extract_m3u8_formats(
|
|
media_url, video_id, 'mp4', m3u8_id='hls', fatal=False))
|
|
else:
|
|
formats.append({
|
|
'format_id': media_data.get('media-category', {}).get('@attributes', {}).get('label'),
|
|
'url': media_url,
|
|
'tbr': int_or_none(media.get('bitrate')),
|
|
'filesize': int_or_none(media.get('fileSize')),
|
|
'ext': ext,
|
|
})
|
|
|
|
self._sort_formats(formats)
|
|
|
|
timestamp = unified_timestamp(item.get('pubDate'), ' ') or parse_iso8601(item.get('dc-date'))
|
|
|
|
return {
|
|
'id': video_id,
|
|
'title': get_media_node('title'),
|
|
'description': get_media_node('description'),
|
|
'thumbnails': thumbnails,
|
|
'timestamp': timestamp,
|
|
'duration': int_or_none(media_content[0].get('@attributes', {}).get('duration')),
|
|
'subtitles': subtitles,
|
|
'formats': formats,
|
|
}
|