2017-06-21 23:19:02 +02:00

151 lines
5.1 KiB
Python

from __future__ import unicode_literals
import re
from .common import InfoExtractor
from ..utils import (
int_or_none,
ExtractorError
)
class NprBaseIE(InfoExtractor):
def _extract_info(self, id_):
json_data = self._download_json(
'http://api.npr.org/query', id_, query={
'id': id_,
'fields': 'titles,audio,show,multimedia,text',
'format': 'json',
'apiKey': 'MDAzMzQ2MjAyMDEyMzk4MTU1MDg3ZmM3MQ010',
})
return json_data['list']['story'][0]
def _extract_formats(self, media_info):
FORMATS_ = {
'hls': 'm3u8', 'hlsOnDemand': 'm3u8', 'mediastream': 'mp3',
'mp3': 'mp3', 'mp4': 'mp4', 'wm': 'wm', 'threegp': '3gp'
}
formats = []
for format_id, formats_entry in media_info.get('format', {}).items():
if not formats_entry:
continue
if isinstance(formats_entry, list):
formats_entry = formats_entry[0]
format_url = formats_entry.get('$text')
if not format_url:
continue
if format_id == 'smil':
formats += self._extract_smil_formats(
format_url,
media_info['id'],
fatal=False
)
continue
elif format_id == 'm3u8':
mobj = re.match(
r'(?P<baseurl>https?:.+?akamaihd.+?-n-)(?P<qualities>(?:,*\d+)+)',
format_url
)
if mobj is None:
continue
m3u8_base = mobj.group('baseurl')
qualities = re.findall(r'(\d+)', mobj.group('qualities'))
for quality in qualities:
formats += self._extract_m3u8_formats(
m3u8_base + '%s.mp4/master.m3u8' % quality,
media_info['id'],
fatal=False
)
continue
formats.append({
'url': format_url,
'format_id': format_id,
'ext': FORMATS_.get(format_id),
})
self._sort_formats(formats)
return formats
class NprIE(NprBaseIE):
_VALID_URL = r'https?://(?:www\.)?npr\.org/(?:sections/\w+/\d+/\d+/\d+/|player/v2/mediaPlayer\.html\?.*\bid=)(?P<id>\d+)'
_TESTS = [{
'url': 'http://www.npr.org/sections/allsongs/2015/10/21/449974205/new-music-from-beach-house-chairlift-cmj-discoveries-and-more',
'info_dict': {
'id': '449974205',
'title': 'New Music From Beach House, Chairlift, CMJ Discoveries And More'
},
'playlist_count': 7,
}, {
'url': 'http://www.npr.org/sections/deceptivecadence/2015/10/09/446928052/music-from-the-shadows-ancient-armenian-hymns-and-piano-jazz',
'info_dict': {
'id': '446928052',
'title': "Songs We Love: Tigran Hamasyan, 'Your Mercy is Boundless'"
},
'playlist': [{
'md5': 'df2917b738fdd2358a9f0e7e49fcdf2e',
'info_dict': {
'id': '446929930',
'ext': 'mp4',
'title': 'Your Mercy is Boundless (Bazum en Qo gtutyunqd)',
'duration': 402,
},
}],
}]
def _real_extract(self, url):
playlist_id = self._match_id(url)
story = self._extract_info(playlist_id)
entries = []
for audio in story.get('audio', []):
title = audio.get('title', {}).get('$text')
duration = int_or_none(audio.get('duration', {}).get('$text'))
entries.append({
'id': audio['id'],
'title': title,
'duration': duration,
'formats': self._extract_formats(audio),
})
playlist_title = story.get('title', {}).get('$text')
return self.playlist_result(entries, playlist_id, playlist_title)
class NprVideoIE(NprBaseIE):
_VALID_URL = r'https?://(?:www\.)?npr\.org/event/music/(?P<id>\d+)'
_TEST = {
'url': 'http://www.npr.org/event/music/533198237/tigers-jaw-tiny-desk-concert',
'md5': '5b385e0e96c2731261df9a4ed1ff2cba',
'info_dict': {
'id': '533198237',
'title': 'Tigers Jaw: Tiny Desk Concert',
'ext': 'mp4',
'width': 624,
'height': 351,
},
'expected_warnings': ['HTTP Error 404'],
}
def _real_extract(self, url):
video_id = self._match_id(url)
story = self._extract_info(video_id)
title = story.get('title', {}).get('$text')
if title is None:
ExtractorError('Fail extracting title')
video = story.get('multimedia')[0]
return {
'id': video_id,
'title': title,
'duration': int_or_none(video.get('duration', {}).get('$text')),
'formats': self._extract_formats(video),
'width': int_or_none(video.get('width', {}).get('$text')),
'height': int_or_none(video.get('height', {}).get('$text')),
}