[TRT] Add extractor for TRT (Turkish TV/radio) catch-up and live streams

youtube_dl/extractor/extractors.py

@@ -1293,6 +1293,10 @@ from .trovo import (
     TrovoIE,
     TrovoVodIE,
 )
+from .trt import (
+    TRTIE,
+    TRTLiveIE,
+)
 from .trunews import TruNewsIE
 from .trutv import TruTVIE
 from .tube8 import Tube8IE

youtube_dl/extractor/trt.py

@@ -0,0 +1,325 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+import json
+import re
+import time
+
+from .common import InfoExtractor
+
+from ..compat import (
+    compat_urlparse,
+)
+from ..utils import (
+    determine_ext,
+    dict_get,
+    merge_dicts,
+    parse_iso8601,
+    strip_or_none,
+    try_get,
+    url_or_none,
+    urljoin,
+)
+
+
+class TRTIE(InfoExtractor):
+    IE_DESC = 'TRT (Turkish State TV) programs and series'
+    _VALID_URL = r'''(?x)
+                     (?P<list>https?://(?:www\.)trtizle\.com/
+                         (?:diziler|programlar|belgesel|filmler|cocuk|trtarsiv|engelsiz)/
+                             (?P<show>[\w-]+))(?:/(?P<id>[\w-]+))?'''
+    _TESTS = [{
+        'url': 'https://www.trtizle.com/belgesel/dunya-tarihinin-donum-noktalari/dunya-tarihinin-donum-noktalari-1-bolum-125583',
+        'md5': 'c46dc0b9b53ad372c4ac6b3982805f05',
+        'info_dict': {
+            'id': 'dunya-tarihinin-donum-noktalari-1-bolum-125583',
+            'ext': 'mp4',
+            'title': 'Dünya Tarihinin Dönüm Noktaları 1.Bölüm',
+            'description': 'Bedelini insanların ödeyeceği bir imparatorluk çekişmesinde Persler, Yunanlara karşı...',
+            'timestamp': 1617148800,
+            'upload_date': '20210331',
+            'thumbnail': r're:https?://.+\.jpe?g',
+            'duration': float,
+            'series': 'Dünya Tarihinin Dönüm Noktaları',
+        },
+        'params': {
+            # adaptive download
+            'skip_download': True,
+        }
+    }, {
+        'url': 'https://www.trtizle.com/belgesel/dunya-tarihinin-donum-noktalari',
+        'info_dict': {
+            'id': 'dunya-tarihinin-donum-noktalari',
+            'title': 'Dünya Tarihinin Dönüm Noktaları',
+        },
+        'playlist_mincount': 22,
+    }, {
+        'url': 'https://www.trtizle.com/diziler/yol-ayrimi/yol-ayrimi-1-bolum-5774583',
+        'md5': '67ada6b2020b5dd0d3e24646b2725676',
+        'info_dict': {
+            'id': 'yol-ayrimi-1-bolum-5774583',
+            'ext': 'mp4',
+            'title': 'Yol Ayrımı 1.Bölüm',
+            'description': 'Seyrisefain balosunda, herkes bir haberin akıbetini beklemektedir…',
+            'timestamp': 1623888000,
+            'upload_date': '20210617',
+            'thumbnail': r're:https?://.+\.jpe?g',
+            'duration': float,
+            'series': 'Yol Ayrımı',
+        },
+        'params': {
+            # adaptive download
+            'skip_download': True,
+        },
+    }, {
+        'url': 'https://www.trtizle.com/diziler/yol-ayrimi/',
+        'info_dict': {
+            'id': 'yol-ayrimi',
+            'title': 'Yol Ayrımı',
+        },
+        'playlist_mincount': 5,
+    }, {
+        'url': 'https://www.trtizle.com/programlar/sade-saz/sade-saz-1-bolum-7646201',
+        'md5': '8f416e64379ea4d1d3ea0a65dc922f5c',
+        'info_dict': {
+            'id': 'sade-saz-1-bolum-7646201',
+            'ext': 'mp4',
+            'title': 'Sade Saz 1.Bölüm',
+            'description': 'Sade Saz’ın ilk bölümünün konuğu, tanbur icracısı K. Alper Uzkur.',
+            'timestamp': 1641772800,
+            'upload_date': '20220110',
+            'thumbnail': r're:https?://.+\.jpe?g',
+            'duration': float,
+            'series': 'Sade Saz',
+        },
+        'params': {
+            # adaptive download
+            'skip_download': True,
+        },
+    }, {
+        'url': 'https://www.trtizle.com/programlar/sade-saz',
+        'info_dict': {
+            'id': 'sade-saz',
+            'title': 'Sade Saz',
+        },
+        'playlist_mincount': 6,
+    }, {
+        'url': 'https://www.trtizle.com/filmler/looking-for-eric/looking-for-eric-8414201',
+        'md5': '833d61e4a10606d71b3903295cfa3c63',
+        'info_dict': {
+            'id': 'looking-for-eric-8414201',
+            'ext': 'mp4',
+            'title': 'Looking for Eric',
+            'description': 'Postacı Eric\'in hayatı krize sürüklenirken gerçek ve hayal birbirine karışır...',
+            'upload_date': '20220401',
+            'timestamp': 1648771200,
+            'thumbnail': r're:https?://.+\.jpe?g',
+            'duration': float,
+        },
+        'params': {
+            # adaptive download
+            'skip_download': True,
+        },
+    }, {
+        'url': 'https://www.trtizle.com/cocuk/kaptan-pengu-ve-arkadaslari/kaptan-pengu-ve-arkadaslari-okul-aciliyor-6034815',
+        'md5': '551c479d1a6bc7c538356907d4ea5d19',
+        'info_dict': {
+            'id': 'kaptan-pengu-ve-arkadaslari-okul-aciliyor-6034815',
+            'ext': 'mp4',
+            'title': 'Kaptan Pengu ve Arkadaşları 1.Bölüm',
+            'description': 'Hayvanlar Konseyi\'nden Kaptan Pengu\'ya bir mektup vardır...',
+            'timestamp': 1626134400,
+            'upload_date': '20210713',
+            'thumbnail': r're:https?://.+\.jpe?g',
+            'duration': float,
+            'series': 'Kaptan Pengu ve Arkadaşları',
+        },
+        'params': {
+            # adaptive download
+            'skip_download': True,
+        },
+    }, {
+        'url': 'https://www.trtizle.com/cocuk/kaptan-pengu-ve-arkadaslari',
+        'info_dict': {
+            'id': 'kaptan-pengu-ve-arkadaslari',
+            'title': 'Kaptan Pengu ve Arkadaşları',
+        },
+        'playlist_mincount': 41,
+    },
+    ]
+
+    def _extract_formats(self, fmt_url, video_id):
+        formats = []
+        ext = determine_ext(fmt_url)
+        if ext == 'm3u8':
+            formats.extend(self._extract_m3u8_formats(
+                fmt_url, video_id, 'mp4', entry_protocol='m3u8_native',
+                m3u8_id='hls', fatal=False))
+        elif ext == 'mpd':
+            formats.extend(self._extract_mpd_formats(
+                fmt_url, video_id, mpd_id='dash', fatal=False))
+        else:
+            formats.append({
+                'url': fmt_url,
+            })
+        return formats
+
+    def _extract_list(self, playlist_id, url):
+        webpage = self._download_webpage(url, playlist_id)
+        LIST_RE = (
+            r'''<a\s[^>]*?\b%s\s*=\s*['"](%s(?:(?<=/)|/)[\w-]+)'''
+            % ('data-path' if 'data-path' in webpage else 'href',
+               re.escape(compat_urlparse.urlparse(url).path), ))
+
+        def entries():
+            for item_url in re.finditer(LIST_RE, webpage):
+                item_url = urljoin(url, item_url.group(1))
+                yield self._extract_video(self._match_id(item_url), item_url)
+
+        series = self._search_json_ld(webpage, playlist_id, default={}, expected_type='TVSeries')
+        return self.playlist_result(entries(), playlist_id, series.get('series'))
+
+    def _extract_video(self, video_id, url):
+        webpage = self._download_webpage(url, video_id)
+        result = self._search_json_ld(webpage, video_id, default={})
+        result['id'] = video_id
+        if 'title' not in result:
+            result['title'] = (
+                self._html_search_meta(('title', 'og:title', 'twitter:title'), webpage)
+                or self._html_search_regex(r'<title\b[^>]*>([^<]+)</title\b', webpage, 'title'))
+        fmt_url = result.get('url')
+        formats = []
+        if fmt_url:
+            del result['url']
+            formats = self._extract_formats(fmt_url, video_id)
+        self._sort_formats(formats)
+        result['formats'] = formats
+
+        return merge_dicts(
+            result, {
+                'description': self._html_search_meta(('description', 'og:description'), webpage, 'description'),
+                'thumbnail': url_or_none(self._og_search_thumbnail(webpage)),
+            })
+
+    def _real_extract(self, url):
+        show_id, video_id, playlist_url = re.match(self._VALID_URL, url).group('show', 'id', 'list')
+        # TODO: adapt --yes/no-playlist to make this work properly
+        # if not video_id or self._downloader.params.get('noplaylist') is False:
+        if not video_id:
+            return self._extract_list(show_id, playlist_url)
+
+        return self._extract_video(video_id, url)
+
+
+class TRTLiveIE(TRTIE):
+    IE_DESC = 'TRT (Turkish State TV and radio) live channels'
+    _VALID_URL = r'https?://(?:www\.)?trtizle\.com/canli/(?:tv/trt-|radyo/(?:radyo-)?)(?P<id>[\w-]+)'
+    _TESTS = [{
+        'url': 'https://www.trtizle.com/canli/tv/trt-world',
+        'info_dict': {
+            'id': 'trtworld',
+            'ext': 'mp4',
+            'title': r're:TRT WORLD .+',
+            'description': 'TRT World',
+            'is_live': True,
+        },
+        'params': {
+            # adaptive download
+            'skip_download': True,
+        }
+    },
+    ]
+
+    def _real_extract(self, url):
+        chan_id = self._match_id(url)
+        webpage = self._download_webpage(url, chan_id)
+        chan_id = self._search_regex(
+            r'\blivePlayer\s*\.\s*openPlayer\s*\([^)]*?\btrt\.com\.tr/trtportal/(?:[^/]+/)+thumbnails/([\w-]+)\.(?:jp|png)',
+            webpage, 'slug', fatal=False) or chan_id
+        chan_url = self._search_regex(
+            r'''\blivePlayerConfig\s*\.\s*baseEpgUrl\s*=\s*(?P<q>'|")(?P<url>https?://(?:(?!(?P=q)).)+)(?P=q)''',
+            webpage, 'player config', group='url')
+        chan_url = '%s%s.json' % (chan_url, chan_id)
+
+        def maybe_xml2json(src):
+            """Turn unexpected XML returned from an API URL into JSON"""
+            m = re.match(r'''^\s*<\?xml\b(?:[^/>]*?\bencoding\s*=\s*['"](?P<enc>[\w-]+))?[^/>]*\?>\s*(?P<xml><.+>)$''', src)
+            if m:
+
+                # Thanks https://stackoverflow.com/a/63556250 for inspiration
+                ATTR_RE = (
+                    r"""(?s)(?P<avr>\S ?)(?:\s*=\s*(?P<q>['"])(?P<avl>.*?)(?<!\\)(?P=q))?"""
+                )
+
+                def elt_value(attr_str, val_dict):
+                    v = {}
+                    attrs = dict((j.group("avr"), j.groupdict(True).get("avl"))
+                                 for j in re.finditer(ATTR_RE, attr_str.strip()))
+                    if attrs:
+                        v['@attributes'] = attrs
+                    v['@values'] = val_dict
+                    return v
+
+                def xml2dict(xml_str):
+                    elts = re.findall(
+                        r"(?s)<(?P<var>\S )(?P<attr>[^/>]*)(?:(?:>(?P<val>.*?)</(?P=var)>)|(?:/>))",
+                        xml_str,
+                    )
+
+                    if elts:
+                        elts = [{i[0]: elt_value(i[1], xml2dict(i[2]))} for i in elts]
+                        if len(elts) == 1:
+                            return elts[0]
+                        return elts
+                    return xml_str
+
+                try:
+                    return json.dumps(xml2dict(m.group('xml').encode(m.group('enc') or 'utf-8')))
+                except Exception:
+                    pass
+            return src
+
+        chan_info = self._download_json(
+            chan_url, chan_id, fatal=False,
+            note='Downloading player EPG JSON',
+            query={'_': int(time.time() * 1000)},
+            expected_status=403,
+            # errors are returned as XML
+            transform_source=maybe_xml2json)
+        if not isinstance(chan_info, dict) or 'Error' in chan_info:
+            chan_info = self._download_json(
+                'https://trtizle-api.cdn.wp.trt.com.tr/trttv/v3/livestream',
+                chan_id, fatal=False,
+                note='Downloading livestream API JSON',
+                query={'path': compat_urlparse.urlparse(url).path}) or {}
+
+        title = chan_info['channel']['title']
+
+        current = try_get(chan_info, lambda x: x['current'], dict) or {}
+        if current.get('geo_block'):
+            self._downloader.report_warning(
+                '[%s] %s' % (self.IE_NAME, 'Stream is geo-blocked'))
+
+        chan_info = chan_info['channel']
+        fmt_url = dict_get(chan_info, ('url', 'noneDvrUrl'))
+        formats = []
+        if fmt_url:
+            formats = self._extract_formats(fmt_url, chan_id)
+        self._sort_formats(formats)
+
+        start_end = [parse_iso8601(current.get(x)) for x in ('starttime', 'endtime')]
+        if None in start_end:
+            start_end = None
+
+        return {
+            'id': chan_id,
+            'title': self._live_title(current.get('title') or title),
+            'is_live': True,
+            'formats': formats,
+            'description': strip_or_none(chan_info.get('description')),
+            'thumbnail': next((url_or_none(chan_info.get(x))
+                              for x in ('thumbnail', 'thumbnailYoutubeUrl', 'square_logo', 'livestreamLogoUrl')),
+                              None),
+            'timestamp': start_end and start_end[0],
+            'duration': start_end and (start_end[1] - time.time()),
+        }