[seznamzpravy] Add extractor (closes #14102)

2 files changed, 173 insertions, 0 deletions

diff --git a/youtube_dl/extractor/extractors.py b/youtube_dl/extractor/extractors.py
index 57e74ba62..b442256fe 100644
--- 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
new file mode 100644
index 000000000..0d1e7668e
--- /dev/null
+++ 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&sectionPrefixPreroll=%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')
+            )