commit: 27940ca09c51af23f7fcc6609ab75974576d5b19
parent 3931b84597c3de013ab9ec3455dbcdaf568871e2
Author: Petr Novák <petr.novak@cosmoboy.cz>
Date: Sat, 27 Jan 2018 17:34:31 +0100
[seznamzpravy] Add extractor (closes #14102)
Diffstat:
2 files changed, 173 insertions(+), 0 deletions(-)
diff --git a/youtube_dl/extractor/extractors.py b/youtube_dl/extractor/extractors.py
@@ -933,6 +933,10 @@ from .servingsys import ServingSysIE
from .servus import ServusIE
from .sevenplus import SevenPlusIE
from .sexu import SexuIE
+from .seznamzpravy import (
+ SeznamZpravyIE,
+ SeznamZpravyArticleIE,
+)
from .shahid import (
ShahidIE,
ShahidShowIE,
diff --git a/youtube_dl/extractor/seznamzpravy.py b/youtube_dl/extractor/seznamzpravy.py
@@ -0,0 +1,169 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+from .common import InfoExtractor
+from ..compat import (
+ compat_parse_qs,
+ compat_str,
+ compat_urllib_parse_urlparse,
+)
+from ..utils import (
+ urljoin,
+ int_or_none,
+ try_get,
+ update_url_query,
+)
+
+
+def _raw_id(src_url):
+ return compat_urllib_parse_urlparse(src_url).path.split('/')[-1]
+
+
+class SeznamZpravyIE(InfoExtractor):
+ _VALID_URL = r'https?://(?:www\.)?(?:seznam\.cz/zpravy|seznamzpravy\.cz)/iframe/player\?.*\bsrc='
+ _TESTS = [{
+ 'url': r'https://www.seznamzpravy.cz/iframe/player?duration=241&serviceSlug=zpravy&src=https%3A%2F%2Fv39-a.sdn.szn.cz%2Fv_39%2Fvmd%2F5999c902ea707c67d8e267a9%3Ffl%3Dmdk%2C432f65a0%7C&itemType=video&autoPlay=false&title=Sv%C4%9Bt%20bez%20obalu%3A%20%C4%8Ce%C5%A1t%C3%AD%20voj%C3%A1ci%20na%20mis%C3%ADch%20(kr%C3%A1tk%C3%A1%20verze)&series=Sv%C4%9Bt%20bez%20obalu&serviceName=Seznam%20Zpr%C3%A1vy&poster=%2F%2Fd39-a.sdn.szn.cz%2Fd_39%2Fc_img_F_I%2FR5puJ.jpeg%3Ffl%3Dcro%2C0%2C0%2C1920%2C1080%7Cres%2C1200%2C%2C1%7Cjpg%2C80%2C%2C1&width=1920&height=1080&cutFrom=0&cutTo=0&splVersion=VOD&contentId=170889&contextId=35990&showAdvert=true&collocation=&autoplayPossible=true&embed=&isVideoTooShortForPreroll=false&isVideoTooLongForPostroll=true&videoCommentOpKey=&videoCommentId=&version=4.0.76&dotService=zpravy&gemiusPrismIdentifier=bVc1ZIb_Qax4W2v5xOPGpMeCP31kFfrTzj0SqPTLh_b.Z7&zoneIdPreroll=seznam.pack.videospot&skipOffsetPreroll=5§ionPrefixPreroll=%2Fzpravy',
+ 'params': {'skip_download': True}, # 'file_minsize': 1586 seems to get killed in test_download.py
+ 'info_dict': {
+ 'id': '170889',
+ 'ext': 'mp4',
+ 'title': 'Svět bez obalu: Čeští vojáci na misích (krátká verze)',
+ }
+ }]
+
+ def _extract_sdn_formats(self, sdn_url, video_id):
+ sdn_data = self._download_json(sdn_url, video_id)
+ formats = []
+ mp4_formats = try_get(sdn_data, lambda x: x['data']['mp4'], dict) or {}
+ for fmt, fmtdata in mp4_formats.items():
+ relative_url = fmtdata.get('url')
+ if not relative_url:
+ continue
+
+ try:
+ width, height = fmtdata.get('resolution')
+ except (TypeError, ValueError):
+ width, height = None, None
+
+ formats.append({
+ 'format_id': fmt,
+ 'width': int_or_none(width),
+ 'height': int_or_none(height),
+ 'url': urljoin(sdn_url, relative_url),
+ 'vcodec': fmtdata.get('codec'),
+ 'tbr': int_or_none(fmtdata.get('bandwidth'), scale=1000),
+ 'duration': int_or_none(fmtdata.get('duration'), scale=1000),
+ })
+
+ playlists = sdn_data.get('pls', {})
+ dash_rel_url = try_get(playlists, lambda x: x['dash']['url'], compat_str)
+ if dash_rel_url:
+ formats.extend(self._extract_mpd_formats(urljoin(sdn_url, dash_rel_url), video_id, mpd_id='dash', fatal=False))
+
+ hls_rel_url = try_get(playlists, lambda x: x['hls']['url'], compat_str)
+ if hls_rel_url:
+ formats.extend(self._extract_m3u8_formats(urljoin(sdn_url, hls_rel_url), video_id, ext='mp4', m3u8_id='hls', fatal=False))
+
+ self._sort_formats(formats)
+ return formats
+
+ def _real_extract(self, url):
+ params = compat_parse_qs(compat_urllib_parse_urlparse(url).query)
+ src = params['src'][0]
+ video_id = params.get('contentId', [_raw_id(src)])[0]
+
+ return {
+ 'id': video_id,
+ 'title': params['title'][0],
+ 'formats': self._extract_sdn_formats(src + 'spl2,2,VOD', video_id),
+ }
+
+
+class SeznamZpravyArticleIE(InfoExtractor):
+ _VALID_URL = r'https?://(?:www\.)?(?:seznam\.cz/zpravy|seznamzpravy\.cz)/clanek/(?:[-a-z0-9]+)-(?P<id>[0-9]+)'
+ _API_URL = 'https://apizpravy.seznam.cz/'
+
+ _TESTS = [{
+ # two videos on one page, with SDN URL
+ 'url': 'https://www.seznamzpravy.cz/clanek/jejich-svet-na-nas-utoci-je-lepsi-branit-se-na-jejich-pisecku-rika-reziser-a-major-v-zaloze-marhoul-35990',
+ 'params': {'skip_download': True},
+ # ^ this is here instead of 'file_minsize': 1586, which does not work because
+ # test_download.py forces expected_minsize to at least 10k when test is running
+ 'info_dict': {
+ 'id': '170889',
+ 'ext': 'mp4',
+ 'title': 'Svět bez obalu: Čeští vojáci na misích (krátká verze)',
+ }
+ }, {
+ # video with live stream URL
+ 'url': 'https://www.seznam.cz/zpravy/clanek/znovu-do-vlady-s-ano-pavel-belobradek-ve-volebnim-specialu-seznamu-38489',
+ 'info_dict': {
+ 'id': '185688',
+ 'ext': 'mp4',
+ 'title': 'Předseda KDU-ČSL Pavel Bělobrádek ve volební Výzvě Seznamu',
+ }
+ }]
+
+ def _extract_caption(self, api_data, article_id):
+ title = api_data.get('title') or api_data.get('captionTitle')
+ caption = api_data.get('caption')
+ if not title or not caption:
+ return {}
+
+ if 'sdn' in caption.get('video', {}):
+ src_url = caption['video']['sdn']
+ elif 'liveStreamUrl' in caption:
+ src_url = self._download_json(caption['liveStreamUrl'], article_id)['Location']
+ else:
+ return {}
+
+ return {
+ 'id': caption.get('uid'),
+ 'title': caption.get('title'),
+ 'src': src_url,
+ }
+
+ def _extract_content(self, api_data):
+ entries = []
+ for item in api_data.get('content', []):
+ media = item.get('properties', {}).get('media', {})
+ src_url = media.get('video', {}).get('sdn')
+ title = media.get('title')
+ if not src_url or not title:
+ continue
+
+ entries.append({
+ 'id': media.get('uid'),
+ 'title': title,
+ 'src': src_url,
+ })
+
+ return entries
+
+ def _iframe_result(self, info_dict):
+ video_id = info_dict['id'] or _raw_id(info_dict['src'])
+ url = update_url_query('https://www.seznam.cz/zpravy/iframe/player', {
+ 'src': info_dict['src'],
+ 'title': info_dict['title'],
+ 'contentId': video_id,
+ 'serviceName': 'Seznam Zprávy',
+ })
+ return self.url_result(url, ie='SeznamZpravy', video_id=video_id, video_title=info_dict['title'])
+
+ def _real_extract(self, url):
+ article_id = self._match_id(url)
+ api_data = self._download_json(self._API_URL + 'v1/documents/' + article_id, article_id)
+
+ caption = self._extract_caption(api_data, article_id)
+ content = self._extract_content(api_data)
+
+ if caption and not content:
+ return self._iframe_result(caption)
+ else:
+ if caption:
+ content.insert(0, caption)
+ return self.playlist_result(
+ [self._iframe_result(x) for x in content],
+ playlist_id=article_id,
+ playlist_title=api_data.get('title') or caption.get('title')
+ )
