# coding: utf-8 from __future__ import unicode_literals import re from .common import InfoExtractor from ..compat import compat_str from ..utils import ( xpath_text, xpath_element, int_or_none, parse_duration, ) class HBOBaseIE(InfoExtractor): _FORMATS_INFO = { 'pro7': { 'width': 1280, 'height': 720, }, '1920': { 'width': 1280, 'height': 720, }, 'pro6': { 'width': 768, 'height': 432, }, '640': { 'width': 768, 'height': 432, }, 'pro5': { 'width': 640, 'height': 360, }, 'highwifi': { 'width': 640, 'height': 360, }, 'high3g': { 'width': 640, 'height': 360, }, 'medwifi': { 'width': 400, 'height': 224, }, 'med3g': { 'width': 400, 'height': 224, }, } def _extract_from_id(self, video_id): video_data = self._download_xml( 'http://render.lv3.hbo.com/data/content/global/videos/data/%s.xml' % video_id, video_id) title = xpath_text(video_data, 'title', 'title', True) formats = [] for source in xpath_element(video_data, 'videos', 'sources', True): if source.tag == 'size': path = xpath_text(source, './/path') if not path: continue width = source.attrib.get('width') format_info = self._FORMATS_INFO.get(width, {}) height = format_info.get('height') fmt = { 'url': path, 'format_id': 'http%s' % ('-%dp' % height if height else ''), 'width': format_info.get('width'), 'height': height, } rtmp = re.search(r'^(?P<url>rtmpe?://[^/]+/(?P<app>.+))/(?P<playpath>mp4:.+)$', path) if rtmp: fmt.update({ 'url': rtmp.group('url'), 'play_path': rtmp.group('playpath'), 'app': rtmp.group('app'), 'ext': 'flv', 'format_id': fmt['format_id'].replace('http', 'rtmp'), }) formats.append(fmt) else: video_url = source.text if not video_url: continue if source.tag == 'tarball': formats.extend(self._extract_m3u8_formats( video_url.replace('.tar', '/base_index_w8.m3u8'), video_id, 'mp4', 'm3u8_native', m3u8_id='hls', fatal=False)) elif source.tag == 'hls': m3u8_formats = self._extract_m3u8_formats( video_url.replace('.tar', '/base_index.m3u8'), video_id, 'mp4', 'm3u8_native', m3u8_id='hls', fatal=False) for f in m3u8_formats: if f.get('vcodec') == 'none' and not f.get('tbr'): f['tbr'] = int_or_none(self._search_regex( r'-(\d+)k/', f['url'], 'tbr', default=None)) formats.extend(m3u8_formats) elif source.tag == 'dash': formats.extend(self._extract_mpd_formats( video_url.replace('.tar', '/manifest.mpd'), video_id, mpd_id='dash', fatal=False)) else: format_info = self._FORMATS_INFO.get(source.tag, {}) formats.append({ 'format_id': 'http-%s' % source.tag, 'url': video_url, 'width': format_info.get('width'), 'height': format_info.get('height'), }) self._sort_formats(formats) thumbnails = [] card_sizes = xpath_element(video_data, 'titleCardSizes') if card_sizes is not None: for size in card_sizes: path = xpath_text(size, 'path') if not path: continue width = int_or_none(size.get('width')) thumbnails.append({ 'id': width, 'url': path, 'width': width, }) return { 'id': video_id, 'title': title, 'duration': parse_duration(xpath_text(video_data, 'duration/tv14')), 'formats': formats, 'thumbnails': thumbnails, } class HBOIE(HBOBaseIE): IE_NAME = 'hbo' _VALID_URL = r'https?://(?:www\.)?hbo\.com/video/video\.html\?.*vid=(?P<id>[0-9]+)' _TEST = { 'url': 'http://www.hbo.com/video/video.html?autoplay=true&g=u&vid=1437839', 'md5': '2c6a6bc1222c7e91cb3334dad1746e5a', 'info_dict': { 'id': '1437839', 'ext': 'mp4', 'title': 'Ep. 64 Clip: Encryption', 'thumbnail': r're:https?://.*\.jpg$', 'duration': 1072, } } def _real_extract(self, url): video_id = self._match_id(url) return self._extract_from_id(video_id) class HBOEpisodeIE(HBOBaseIE): IE_NAME = 'hbo:episode' _VALID_URL = r'https?://(?:www\.)?hbo\.com/(?P<path>(?!video)(?:(?:[^/]+/)+video|watch-free-episodes)/(?P<id>[0-9a-z-]+))(?:\.html)?' _TESTS = [{ 'url': 'http://www.hbo.com/girls/episodes/5/52-i-love-you-baby/video/ep-52-inside-the-episode.html?autoplay=true', 'md5': '61ead79b9c0dfa8d3d4b07ef4ac556fb', 'info_dict': { 'id': '1439518', 'display_id': 'ep-52-inside-the-episode', 'ext': 'mp4', 'title': 'Ep. 52: Inside the Episode', 'thumbnail': r're:https?://.*\.jpg$', 'duration': 240, }, }, { 'url': 'http://www.hbo.com/game-of-thrones/about/video/season-5-invitation-to-the-set.html?autoplay=true', 'only_matching': True, }, { 'url': 'http://www.hbo.com/watch-free-episodes/last-week-tonight-with-john-oliver', 'only_matching': True, }] def _real_extract(self, url): path, display_id = re.match(self._VALID_URL, url).groups() content = self._download_json( 'http://www.hbo.com/api/content/' + path, display_id)['content'] video_id = compat_str((content.get('parsed', {}).get( 'common:FullBleedVideo', {}) or content['selectedEpisode'])['videoId']) info_dict = self._extract_from_id(video_id) info_dict['display_id'] = display_id return info_dict