mirror of
https://github.com/l1ving/youtube-dl
synced 2020-11-18 19:53:54 -08:00
107 lines
3.7 KiB
Python
107 lines
3.7 KiB
Python
# coding: utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
import json
|
|
import re
|
|
|
|
from .common import InfoExtractor
|
|
from ..utils import urljoin
|
|
|
|
|
|
class TelevizeSeznamIE(InfoExtractor):
|
|
_VALID_URL = r'https?://(?:www\.)?televizeseznam\.cz/.+/(?P<display_id>.+)-(?P<id>[0-9]+)'
|
|
|
|
_GRAPHQL_URL = 'https://www.televizeseznam.cz/api/graphql'
|
|
_GRAPHQL_QUERY = '''query LoadEpisode($urlName : String){ episode(urlName: $urlName){ ...VideoDetailFragmentOnEpisode } }
|
|
fragment VideoDetailFragmentOnEpisode on Episode {
|
|
id
|
|
spl
|
|
urlName
|
|
name
|
|
perex
|
|
}
|
|
'''
|
|
|
|
_TEST = {
|
|
'url': 'https://www.televizeseznam.cz/video/lajna/buh-57953890',
|
|
'md5': '40c41ade1464a390a0b447e333df4239',
|
|
'info_dict': {
|
|
'id': '57953890',
|
|
'display_id': 'buh',
|
|
'title': 'Bůh',
|
|
'description': 'Trenér Hrouzek je plný rozporů. Na pomoc si povolá i toho nejvyššího. Kdo to ale je? Pomůže mu vyřešit několik dilemat, která se mu v poslední době v životě nahromadila?',
|
|
'ext': 'mp4',
|
|
}
|
|
}
|
|
|
|
def extract_subtitles(self, spl_url, play_list):
|
|
|
|
if not play_list:
|
|
return None
|
|
|
|
subtitles = {}
|
|
for k, v in play_list.items():
|
|
if v.get('language'):
|
|
for ext in v.get('urls'):
|
|
subtitles.setdefault(v['language'], []).append({
|
|
'ext': ext,
|
|
'url': urljoin(spl_url, v['urls'].get(ext))
|
|
})
|
|
return subtitles
|
|
|
|
def _extract(self, ext, spl_url, play_list):
|
|
formats = []
|
|
for r, v in play_list.items():
|
|
format = {
|
|
'format_id': r,
|
|
'url': urljoin(spl_url, v.get('url')),
|
|
'protocol': 'https',
|
|
'ext': ext
|
|
}
|
|
if v.get('resolution'):
|
|
format.update({'width': v['resolution'][0], 'height': v['resolution'][1]})
|
|
|
|
formats.append(format)
|
|
return formats
|
|
|
|
def extract_formats(self, spl_url, play_list):
|
|
formats = []
|
|
|
|
if play_list.get('http_stream') and play_list['http_stream'].get('qualities'):
|
|
formats.extend(self._extract(None, spl_url, play_list['http_stream']['qualities']))
|
|
|
|
if play_list.get('mp4'):
|
|
formats.extend(self._extract('mp4', spl_url, play_list['mp4']))
|
|
|
|
self._sort_formats(formats)
|
|
return formats
|
|
|
|
def _real_extract(self, url):
|
|
display_id, video_id = re.match(self._VALID_URL, url).groups()
|
|
|
|
data = self._download_json(
|
|
self._GRAPHQL_URL, video_id, 'Downloading GraphQL result',
|
|
data=json.dumps({
|
|
'query': self._GRAPHQL_QUERY,
|
|
'variables': {'urlName': video_id}
|
|
}).encode('utf-8'),
|
|
headers={'Content-Type': 'application/json;charset=UTF-8'}
|
|
)['data']
|
|
|
|
spl_url = data['episode']['spl'] + 'spl2,3'
|
|
metadata = self._download_json(spl_url, video_id, 'Downloading playlist')
|
|
if 'Location' in metadata and 'data' not in metadata:
|
|
# they sometimes wants to redirect
|
|
spl_url = metadata['Location']
|
|
metadata = self._download_json(spl_url, video_id, 'Redirected -> Downloading playlist')
|
|
play_list = metadata['data']
|
|
|
|
return {
|
|
'id': video_id,
|
|
'display_id': display_id,
|
|
'title': data['episode'].get('name'),
|
|
'description': data['episode'].get('perex'),
|
|
'subtitles': self.extract_subtitles(spl_url, play_list.get('subtitles')),
|
|
'formats': self.extract_formats(spl_url, play_list)
|
|
}
|