mirror of
				https://github.com/yt-dlp/yt-dlp.git
				synced 2025-11-03 16:15:14 +00:00 
			
		
		
		
	Using https://github.com/asottile/pyupgrade 1. `__future__` imports and `coding: utf-8` were removed 2. Files were rewritten with `pyupgrade --py36-plus --keep-percent-format` 3. f-strings were cherry-picked from `pyupgrade --py36-plus` Extractors are left untouched (except removing header) to avoid unnecessary merge conflicts
		
			
				
	
	
		
			106 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			106 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
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,
 | 
						|
            },
 | 
						|
            '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 _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 _perform_login(self, username, password):
 | 
						|
        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 = self._match_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 = []
 | 
						|
        subtitles = {}
 | 
						|
        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, subtitles = self._extract_m3u8_formats(
 | 
						|
                    src, video_id, 'mp4', 'm3u8_native',
 | 
						|
                    m3u8_id='hls', fatal=False)
 | 
						|
            elif src_type == 'application/dash+xml':
 | 
						|
                formats, subtitles = 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')),
 | 
						|
            'subtitles': subtitles,
 | 
						|
        }
 |