# coding: utf-8 from __future__ import unicode_literals import re from .common import InfoExtractor from ..compat import compat_HTTPError from ..utils import ( ExtractorError, int_or_none, urlencode_postdata, ) class AtresPlayerIE(InfoExtractor): _VALID_URL = r'https?://(?:www\.)?atresplayer\.com/[^/]+/[^/]+/[^/]+/[^/]+/(?P<display_id>.+?)_(?P<id>[0-9a-f]{24})' _NETRC_MACHINE = 'atresplayer' _TESTS = [ { 'url': 'https://www.atresplayer.com/antena3/series/pequenas-coincidencias/temporada-1/capitulo-7-asuntos-pendientes_5d4aa2c57ed1a88fc715a615/', 'info_dict': { 'id': '5d4aa2c57ed1a88fc715a615', 'ext': 'mp4', 'title': 'CapĂtulo 7: Asuntos pendientes', 'description': 'md5:7634cdcb4d50d5381bedf93efb537fbc', 'duration': 3413, }, 'params': { 'format': 'bestvideo', }, 'skip': 'This video is only available for registered users' }, { 'url': 'https://www.atresplayer.com/lasexta/programas/el-club-de-la-comedia/temporada-4/capitulo-10-especial-solidario-nochebuena_5ad08edf986b2855ed47adc4/', 'only_matching': True, }, { 'url': 'https://www.atresplayer.com/antena3/series/el-secreto-de-puente-viejo/el-chico-de-los-tres-lunares/capitulo-977-29-12-14_5ad51046986b2886722ccdea/', 'only_matching': True, }, ] _API_BASE = 'https://api.atresplayer.com/' def _real_initialize(self): self._login() def _handle_error(self, e, code): if isinstance(e.cause, compat_HTTPError) and e.cause.code == code: error = self._parse_json(e.cause.read(), None) if error.get('error') == 'required_registered': self.raise_login_required() raise ExtractorError(error['error_description'], expected=True) raise def _login(self): username, password = self._get_login_info() if username is None: return self._request_webpage( self._API_BASE + 'login', None, 'Downloading login page') try: target_url = self._download_json( 'https://account.atresmedia.com/api/login', None, 'Logging in', headers={ 'Content-Type': 'application/x-www-form-urlencoded' }, data=urlencode_postdata({ 'username': username, 'password': password, }))['targetUrl'] except ExtractorError as e: self._handle_error(e, 400) self._request_webpage(target_url, None, 'Following Target URL') def _real_extract(self, url): display_id, video_id = re.match(self._VALID_URL, url).groups() try: episode = self._download_json( self._API_BASE + 'client/v1/player/episode/' + video_id, video_id) except ExtractorError as e: self._handle_error(e, 403) title = episode['titulo'] formats = [] for source in episode.get('sources', []): src = source.get('src') if not src: continue src_type = source.get('type') if src_type == 'application/vnd.apple.mpegurl': formats.extend(self._extract_m3u8_formats( src, video_id, 'mp4', 'm3u8_native', m3u8_id='hls', fatal=False)) elif src_type == 'application/dash+xml': formats.extend(self._extract_mpd_formats( src, video_id, mpd_id='dash', fatal=False)) self._sort_formats(formats) heartbeat = episode.get('heartbeat') or {} omniture = episode.get('omniture') or {} get_meta = lambda x: heartbeat.get(x) or omniture.get(x) return { 'display_id': display_id, 'id': video_id, 'title': title, 'description': episode.get('descripcion'), 'thumbnail': episode.get('imgPoster'), 'duration': int_or_none(episode.get('duration')), 'formats': formats, 'channel': get_meta('channel'), 'season': get_meta('season'), 'episode_number': int_or_none(get_meta('episodeNumber')), }