logo

youtube-dl

[mirror] Download/Watch videos from video hostersgit clone https://hacktivis.me/git/mirror/youtube-dl.git

atresplayer.py (4382B)

    

  1. # coding: utf-8
  2. from __future__ import unicode_literals
  4. import re
  6. from .common import InfoExtractor
  7. from ..compat import compat_HTTPError
  8. from ..utils import (
  9.     ExtractorError,
  10.     int_or_none,
  11.     urlencode_postdata,
  12. )
  15. class AtresPlayerIE(InfoExtractor):
  16.     _VALID_URL = r'https?://(?:www\.)?atresplayer\.com/[^/]+/[^/]+/[^/]+/[^/]+/(?P<display_id>.+?)_(?P<id>[0-9a-f]{24})'
  17.     _NETRC_MACHINE = 'atresplayer'
  18.     _TESTS = [
  19.         {
  20.             'url': 'https://www.atresplayer.com/antena3/series/pequenas-coincidencias/temporada-1/capitulo-7-asuntos-pendientes_5d4aa2c57ed1a88fc715a615/',
  21.             'info_dict': {
  22.                 'id': '5d4aa2c57ed1a88fc715a615',
  23.                 'ext': 'mp4',
  24.                 'title': 'Capítulo 7: Asuntos pendientes',
  25.                 'description': 'md5:7634cdcb4d50d5381bedf93efb537fbc',
  26.                 'duration': 3413,
  27.             },
  28.             'params': {
  29.                 'format': 'bestvideo',
  30.             },
  31.             'skip': 'This video is only available for registered users'
  32.         },
  33.         {
  34.             'url': 'https://www.atresplayer.com/lasexta/programas/el-club-de-la-comedia/temporada-4/capitulo-10-especial-solidario-nochebuena_5ad08edf986b2855ed47adc4/',
  35.             'only_matching': True,
  36.         },
  37.         {
  38.             'url': 'https://www.atresplayer.com/antena3/series/el-secreto-de-puente-viejo/el-chico-de-los-tres-lunares/capitulo-977-29-12-14_5ad51046986b2886722ccdea/',
  39.             'only_matching': True,
  40.         },
  41.     ]
  42.     _API_BASE = 'https://api.atresplayer.com/'
  44.     def _real_initialize(self):
  45.         self._login()
  47.     def _handle_error(self, e, code):
  48.         if isinstance(e.cause, compat_HTTPError) and e.cause.code == code:
  49.             error = self._parse_json(e.cause.read(), None)
  50.             if error.get('error') == 'required_registered':
  51.                 self.raise_login_required()
  52.             raise ExtractorError(error['error_description'], expected=True)
  53.         raise
  55.     def _login(self):
  56.         username, password = self._get_login_info()
  57.         if username is None:
  58.             return
  60.         self._request_webpage(
  61.             self._API_BASE + 'login', None, 'Downloading login page')
  63.         try:
  64.             target_url = self._download_json(
  65.                 'https://account.atresmedia.com/api/login', None,
  66.                 'Logging in', headers={
  67.                     'Content-Type': 'application/x-www-form-urlencoded'
  68.                 }, data=urlencode_postdata({
  69.                     'username': username,
  70.                     'password': password,
  71.                 }))['targetUrl']
  72.         except ExtractorError as e:
  73.             self._handle_error(e, 400)
  75.         self._request_webpage(target_url, None, 'Following Target URL')
  77.     def _real_extract(self, url):
  78.         display_id, video_id = re.match(self._VALID_URL, url).groups()
  80.         try:
  81.             episode = self._download_json(
  82.                 self._API_BASE + 'client/v1/player/episode/' + video_id, video_id)
  83.         except ExtractorError as e:
  84.             self._handle_error(e, 403)
  86.         title = episode['titulo']
  88.         formats = []
  89.         for source in episode.get('sources', []):
  90.             src = source.get('src')
  91.             if not src:
  92.                 continue
  93.             src_type = source.get('type')
  94.             if src_type == 'application/vnd.apple.mpegurl':
  95.                 formats.extend(self._extract_m3u8_formats(
  96.                     src, video_id, 'mp4', 'm3u8_native',
  97.                     m3u8_id='hls', fatal=False))
  98.             elif src_type == 'application/dash+xml':
  99.                 formats.extend(self._extract_mpd_formats(
  100.                     src, video_id, mpd_id='dash', fatal=False))
  101.         self._sort_formats(formats)
  103.         heartbeat = episode.get('heartbeat') or {}
  104.         omniture = episode.get('omniture') or {}
  105.         get_meta = lambda x: heartbeat.get(x) or omniture.get(x)
  107.         return {
  108.             'display_id': display_id,
  109.             'id': video_id,
  110.             'title': title,
  111.             'description': episode.get('descripcion'),
  112.             'thumbnail': episode.get('imgPoster'),
  113.             'duration': int_or_none(episode.get('duration')),
  114.             'formats': formats,
  115.             'channel': get_meta('channel'),
  116.             'season': get_meta('season'),
  117.             'episode_number': int_or_none(get_meta('episodeNumber')),
  118.         }