From d7fc56318b72607758e7484c22076ec2999f10b0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sergey=20M=E2=80=A4?= Date: Sun, 4 Oct 2015 20:41:57 +0600 Subject: [PATCH] [limelight] Fix python 2.6, simplify, make more robust (Closes #6734) --- youtube_dl/extractor/__init__.py | 6 +- youtube_dl/extractor/limelight.py | 185 ++++++++++++++++++------------ 2 files changed, 113 insertions(+), 78 deletions(-) diff --git a/youtube_dl/extractor/__init__.py b/youtube_dl/extractor/__init__.py index 20cc3660c..3ace1cc2c 100644 --- a/youtube_dl/extractor/__init__.py +++ b/youtube_dl/extractor/__init__.py @@ -296,9 +296,9 @@ LifeEmbedIE, ) from .limelight import ( - LimeLightMediaIE, - LimeLightChannelIE, - LimeLightChannelListIE, + LimelightMediaIE, + LimelightChannelIE, + LimelightChannelListIE, ) from .liveleak import LiveLeakIE from .livestream import ( diff --git a/youtube_dl/extractor/limelight.py b/youtube_dl/extractor/limelight.py index dcfc215c7..599d8413d 100644 --- a/youtube_dl/extractor/limelight.py +++ b/youtube_dl/extractor/limelight.py @@ -5,65 +5,105 @@ from .common import InfoExtractor from ..utils import ( - int_or_none, determine_ext, + float_or_none, + int_or_none, ) -class LimeLightBaseIE(InfoExtractor): +class LimelightBaseIE(InfoExtractor): + _PLAYLIST_SERVICE_URL = 'http://production-ps.lvp.llnw.net/r/PlaylistService/%s/%s/%s' + _API_URL = 'http://api.video.limelight.com/rest/organizations/%s/%s/%s/%s.json' - def get_playlist_service(self, id, method): - return self._download_json(self.PLAYLIST_SERVICE_URL % (id, method), id) + def _call_playlist_service(self, item_id, method, fatal=True): + return self._download_json( + self._PLAYLIST_SERVICE_URL % (self._PLAYLIST_SERVICE_PATH, item_id, method), + item_id, 'Downloading PlaylistService %s JSON' % method, fatal=fatal) - def get_api(self, orgId, id, method): - return self._download_json(self.API_URL % (orgId, id, method), id) + def _call_api(self, organization_id, item_id, method): + return self._download_json( + self._API_URL % (organization_id, self._API_PATH, item_id, method), + item_id, 'Downloading API %s JSON' % method) - def process_data(self, mobileUrls, streams, properties): + def _extract(self, item_id, pc_method, mobile_method, meta_method): + pc = self._call_playlist_service(item_id, pc_method) + metadata = self._call_api(pc['orgId'], item_id, meta_method) + mobile = self._call_playlist_service(item_id, mobile_method, fatal=False) + return pc, mobile, metadata + + def _extract_info(self, streams, mobile_urls, properties): video_id = properties['media_id'] formats = [] - for mobileUrl in mobileUrls: - if '.m3u8' in mobileUrl['mobileUrl']: - formats.extend(self._extract_m3u8_formats(mobileUrl['mobileUrl'], video_id)) - else: - formats.append({'url': mobileUrl['mobileUrl']}) - for stream in streams: - if '.f4m' in stream['url']: - formats.extend(self._extract_f4m_formats(stream['url'], video_id)) + stream_url = stream.get('url') + if not stream_url: + continue + if '.f4m' in stream_url: + formats.extend(self._extract_f4m_formats(stream_url, video_id)) else: fmt = { - 'url': stream.get('url'), - 'abr': stream.get('audioBitRate'), - 'vbr': stream.get('videoBitRate'), - 'fps': stream.get('videoFrameRate'), - 'width': stream.get('videoWidthInPixels'), - 'height': stream.get('videoHeightInPixels'), - 'ext': determine_ext(stream.get('url')) + 'url': stream_url, + 'abr': float_or_none(stream.get('audioBitRate')), + 'vbr': float_or_none(stream.get('videoBitRate')), + 'fps': float_or_none(stream.get('videoFrameRate')), + 'width': int_or_none(stream.get('videoWidthInPixels')), + 'height': int_or_none(stream.get('videoHeightInPixels')), + 'ext': determine_ext(stream_url) } - rtmp = re.search(r'^(?Prtmp://[^/]+/(?P.+))/(?Pmp4:.+)$', stream['url']) + rtmp = re.search(r'^(?Prtmpe?://[^/]+/(?P.+))/(?Pmp4:.+)$', stream_url) if rtmp: + format_id = 'rtmp' + if stream.get('videoBitRate'): + format_id += '-%d' % int_or_none(stream['videoBitRate']) fmt.update({ 'url': rtmp.group('url'), 'play_path': rtmp.group('playpath'), 'app': rtmp.group('app'), + 'ext': 'flv', + 'format_id': format_id, }) formats.append(fmt) + for mobile_url in mobile_urls: + media_url = mobile_url.get('mobileUrl') + if not media_url: + continue + format_id = mobile_url.get('targetMediaPlatform') + if determine_ext(media_url) == 'm3u8': + formats.extend(self._extract_m3u8_formats( + media_url, video_id, 'mp4', entry_protocol='m3u8_native', + preference=-1, m3u8_id=format_id)) + else: + formats.append({ + 'url': media_url, + 'format_id': format_id, + 'preference': -1, + }) + self._sort_formats(formats) title = properties['title'] description = properties.get('description') - timestamp = properties.get('create_date') - duration = int_or_none(properties.get('duration_in_milliseconds')) - filesize = properties.get('total_storage_in_bytes') + timestamp = int_or_none(properties.get('publish_date') or properties.get('create_date')) + duration = float_or_none(properties.get('duration_in_milliseconds'), 1000) + filesize = int_or_none(properties.get('total_storage_in_bytes')) categories = [properties.get('category')] + tags = properties.get('tags', []) thumbnails = [{ - 'url': thumbnail.get('url'), + 'url': thumbnail['url'], 'width': int_or_none(thumbnail.get('width')), 'height': int_or_none(thumbnail.get('height')), - } for thumbnail in properties.get('thumbnails')] - subtitles = {caption.get('language_code'): [{'url': caption.get('url')}] for caption in properties.get('captions')} + } for thumbnail in properties.get('thumbnails', []) if thumbnail.get('url')] + + subtitles = {} + for caption in properties.get('captions', {}): + lang = caption.get('language_code') + subtitles_url = caption.get('url') + if lang and subtitles_url: + subtitles[lang] = [{ + 'url': subtitles_url, + }] return { 'id': video_id, @@ -74,44 +114,50 @@ def process_data(self, mobileUrls, streams, properties): 'duration': duration, 'filesize': filesize, 'categories': categories, + 'tags': tags, 'thumbnails': thumbnails, 'subtitles': subtitles, } -class LimeLightMediaIE(LimeLightBaseIE): +class LimelightMediaIE(LimelightBaseIE): IE_NAME = 'limelight' - _VALID_URL = r'http://link\.videoplatform\.limelight\.com/media/?.*mediaId=(?P[a-z0-9]{32})' + _VALID_URL = r'(?:limelight:media:|http://link\.videoplatform\.limelight\.com/media/\??\bmediaId=)(?P[a-z0-9]{32})' _TEST = { 'url': 'http://link.videoplatform.limelight.com/media/?mediaId=3ffd040b522b4485b6d84effc750cd86', - 'md5': '3213605088be599705677ef785db6972', 'info_dict': { 'id': '3ffd040b522b4485b6d84effc750cd86', - 'ext': 'mp4', + 'ext': 'flv', 'title': 'HaP and the HB Prince Trailer', 'description': 'As Harry Potter begins his 6th year at Hogwarts School of Witchcraft and Wizardry, he discovers an old book marked mysteriously "This book is the property of the Half-Blood Prince" and begins to learn more about Lord Voldemort\'s dark past.', 'thumbnail': 're:^https?://.*\.jpeg$', - 'duration': 144230, + 'duration': 144.23, 'timestamp': 1244136834, - "upload_date": "20090604", - } + 'upload_date': '20090604', + }, + 'params': { + # rtmp download + 'skip_download': True, + }, } - PLAYLIST_SERVICE_URL = 'http://production-ps.lvp.llnw.net/r/PlaylistService/media/%s/%s' - API_URL = 'http://api.video.limelight.com/rest/organizations/%s/media/%s/%s.json' + _PLAYLIST_SERVICE_PATH = 'media' + _API_PATH = 'media' def _real_extract(self, url): video_id = self._match_id(url) - mobile_json_data = self.get_playlist_service(video_id, 'getMobilePlaylistByMediaId') - pc_json_data = self.get_playlist_service(video_id, 'getPlaylistByMediaId') - properties = self.get_api(pc_json_data['orgId'], video_id, 'properties') + pc, mobile, metadata = self._extract( + video_id, 'getPlaylistByMediaId', 'getMobilePlaylistByMediaId', 'properties') - return self.process_data(mobile_json_data['mediaList'][0]['mobileUrls'], pc_json_data['playlistItems'][0]['streams'], properties) + return self._extract_info( + pc['playlistItems'][0].get('streams', []), + mobile['mediaList'][0].get('mobileUrls', []) if mobile else [], + metadata) -class LimeLightChannelIE(LimeLightBaseIE): +class LimelightChannelIE(LimelightBaseIE): IE_NAME = 'limelight:channel' - _VALID_URL = r'http://link\.videoplatform\.limelight\.com/media/?.*channelId=(?P[a-z0-9]{32})' + _VALID_URL = r'(?:limelight:channel:|http://link\.videoplatform\.limelight\.com/media/\??\bchannelId=)(?P[a-z0-9]{32})' _TEST = { 'url': 'http://link.videoplatform.limelight.com/media/?channelId=ab6a524c379342f9b23642917020c082', 'info_dict': { @@ -120,31 +166,29 @@ class LimeLightChannelIE(LimeLightBaseIE): }, 'playlist_mincount': 3, } - PLAYLIST_SERVICE_URL = 'http://production-ps.lvp.llnw.net/r/PlaylistService/channel/%s/%s' - API_URL = 'http://api.video.limelight.com/rest/organizations/%s/channels/%s/%s.json' + _PLAYLIST_SERVICE_PATH = 'channel' + _API_PATH = 'channels' def _real_extract(self, url): channel_id = self._match_id(url) - mobile_json_data = self.get_playlist_service(channel_id, 'getMobilePlaylistWithNItemsByChannelId?begin=0&count=-1') - pc_json_data = self.get_playlist_service(channel_id, 'getPlaylistByChannelId') - medias = self.get_api(pc_json_data['orgId'], channel_id, 'media') + pc, mobile, medias = self._extract( + channel_id, 'getPlaylistByChannelId', + 'getMobilePlaylistWithNItemsByChannelId?begin=0&count=-1', 'media') - entries = [] - for i in range(len(medias['media_list'])): - entries.append(self.process_data(mobile_json_data['mediaList'][i]['mobileUrls'], pc_json_data['playlistItems'][i]['streams'], medias['media_list'][i])) + entries = [ + self._extract_info( + pc['playlistItems'][i].get('streams', []), + mobile['mediaList'][i].get('mobileUrls', []) if mobile else [], + medias['media_list'][i]) + for i in range(len(medias['media_list']))] - return { - 'id': channel_id, - 'title': pc_json_data['title'], - 'entries': entries, - '_type': 'playlist', - } + return self.playlist_result(entries, channel_id, pc['title']) -class LimeLightChannelListIE(LimeLightBaseIE): +class LimelightChannelListIE(LimelightBaseIE): IE_NAME = 'limelight:channel_list' - _VALID_URL = r'http://link\.videoplatform\.limelight\.com/media/?.*channelListId=(?P[a-z0-9]{32})' + _VALID_URL = r'(?:limelight:channel_list:|http://link\.videoplatform\.limelight\.com/media/\?.*?\bchannelListId=)(?P[a-z0-9]{32})' _TEST = { 'url': 'http://link.videoplatform.limelight.com/media/?channelListId=301b117890c4465c8179ede21fd92e2b', 'info_dict': { @@ -153,24 +197,15 @@ class LimeLightChannelListIE(LimeLightBaseIE): }, 'playlist_mincount': 2, } - PLAYLIST_SERVICE_URL = 'http://production-ps.lvp.llnw.net/r/PlaylistService/channel_list/%s/%s' + _PLAYLIST_SERVICE_PATH = 'channel_list' def _real_extract(self, url): channel_list_id = self._match_id(url) - json_data = self.get_playlist_service(channel_list_id, 'getMobileChannelListById') + channel_list = self._call_playlist_service(channel_list_id, 'getMobileChannelListById') - entries = [] - for channel in json_data['channelList']: - entries.append({ - 'url': 'http://link.videoplatform.limelight.com/media/?channelId=%s' % channel['id'], - '_type': 'url', - 'ie_key': 'LimeLightChannel', - }) + entries = [ + self.url_result('limelight:channel:%s' % channel['id'], 'LimelightChannel') + for channel in channel_list['channelList']] - return { - 'id': channel_list_id, - 'title': json_data['title'], - 'entries': entries, - '_type': 'playlist', - } + return self.playlist_result(entries, channel_list_id, channel_list['title'])