From 24ce9e5f51478f1e1cd352d7971aade2ffadded2 Mon Sep 17 00:00:00 2001 From: gfabiano Date: Wed, 21 Jun 2017 01:59:45 +0200 Subject: [PATCH] [npr] Add new extractor --- youtube_dl/extractor/extractors.py | 5 +- youtube_dl/extractor/npr.py | 135 ++++++++++++++++++++--------- 2 files changed, 97 insertions(+), 43 deletions(-) diff --git a/youtube_dl/extractor/extractors.py b/youtube_dl/extractor/extractors.py index e97691daa..fddbc9ce6 100644 --- a/youtube_dl/extractor/extractors.py +++ b/youtube_dl/extractor/extractors.py @@ -709,7 +709,10 @@ from .npo import ( VPROIE, WNLIE, ) -from .npr import NprIE +from .npr import ( + NprPlaylistIE, + NprVideoIE, +) from .nrk import ( NRKIE, NRKPlaylistIE, diff --git a/youtube_dl/extractor/npr.py b/youtube_dl/extractor/npr.py index 1777aa10b..a5ff6e8c4 100644 --- a/youtube_dl/extractor/npr.py +++ b/youtube_dl/extractor/npr.py @@ -1,33 +1,79 @@ from __future__ import unicode_literals from .common import InfoExtractor -from ..compat import compat_urllib_parse_urlencode -from ..utils import ( - int_or_none, - qualities, -) +from ..utils import int_or_none -class NprIE(InfoExtractor): - _VALID_URL = r'https?://(?:www\.)?npr\.org/player/v2/mediaPlayer\.html\?.*\bid=(?P\d+)' +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': + # Error 404 for some reason + formats += self._extract_m3u8_formats( + format_url, + media_info['id'], + fatal=False + ) + continue + formats.append({ + 'url': format_url, + 'format_id': format_id, + 'ext': FORMATS_.get(format_id), + }) + return formats + + +class NprPlaylistIE(NprBaseIE): + _VALID_URL = r'https?://(?:www\.)?npr\.org/(?:sections/\w+/\d+/\d+/\d+/|player/v2/mediaPlayer\.html\?.*\bid=)(?P\d+)' _TESTS = [{ - 'url': 'http://www.npr.org/player/v2/mediaPlayer.html?id=449974205', + '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/player/v2/mediaPlayer.html?action=1&t=1&islist=false&id=446928052&m=446929930&live=1', + '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': '12fa60cb2d3ed932f53609d4aeceabf1', + 'md5': 'df2917b738fdd2358a9f0e7e49fcdf2e', 'info_dict': { 'id': '446929930', - 'ext': 'mp3', + # 'ext': 'mp4', 'title': 'Your Mercy is Boundless (Bazum en Qo gtutyunqd)', 'duration': 402, }, @@ -36,47 +82,52 @@ class NprIE(InfoExtractor): def _real_extract(self, url): playlist_id = self._match_id(url) - - config = self._download_json( - 'http://api.npr.org/query?%s' % compat_urllib_parse_urlencode({ - 'id': playlist_id, - 'fields': 'titles,audio,show', - 'format': 'json', - 'apiKey': 'MDAzMzQ2MjAyMDEyMzk4MTU1MDg3ZmM3MQ010', - }), playlist_id) - - story = config['list']['story'][0] - - KNOWN_FORMATS = ('threegp', 'mp4', 'mp3') - quality = qualities(KNOWN_FORMATS) + 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')) - formats = [] - for format_id, formats_entry in audio.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 in KNOWN_FORMATS: - formats.append({ - 'url': format_url, - 'format_id': format_id, - 'ext': formats_entry.get('type'), - 'quality': quality(format_id), - }) - self._sort_formats(formats) entries.append({ 'id': audio['id'], 'title': title, 'duration': duration, - 'formats': formats, + '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\d+)' + + _TEST = { + 'url': 'http://www.npr.org/event/music/533198237/tigers-jaw-tiny-desk-concert', + 'md5': '5b385e0e96c2731261df9a4ed1ff2cba', + 'info_dict': { + 'id': '533201718', + 'display_id': '533198237', + 'title': 'Tigers Jaw: Tiny Desk Concert', + 'ext': 'mp4', + 'width': 624, + 'height': 351, + } + } + + def _real_extract(self, url): + display_id = self._match_id(url) + story = self.extract_info(display_id) + + video = story.get('multimedia')[0] + + return { + 'url': url, + 'display_id': display_id, + 'title': story.get('title', {}).get('$text'), + 'id': video.get('id'), + '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')), + }