[ceskatelevize] Simplify, restore bonus video test and skip georestri…

…cted test (Closes #9431)
pmenzel · May 9, 2016 · 3951e7e · 3951e7e
1 parent f1f6f5a
commit 3951e7e
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/youtube_dl/extractor/ceskatelevize.py b/youtube_dl/extractor/ceskatelevize.py
@@ -32,19 +32,34 @@ class CeskaTelevizeIE(InfoExtractor):
             # m3u8 download
             'skip_download': True,
         },
+    }, {
+        'url': 'http://www.ceskatelevize.cz/ivysilani/10441294653-hyde-park-civilizace/215411058090502/bonus/20641-bonus-01-en',
+        'info_dict': {
+            'id': '61924494877028507',
+            'ext': 'mp4',
+            'title': 'Hyde Park Civilizace: Bonus 01 - En',
+            'description': 'English Subtittles',
+            'thumbnail': 're:^https?://.*\.jpg',
+            'duration': 81.3,
+        },
+        'params': {
+            # m3u8 download
+            'skip_download': True,
+        },
     }, {
         # live stream
         'url': 'http://www.ceskatelevize.cz/ivysilani/zive/ct4/',
         'info_dict': {
             'id': 402,
             'ext': 'mp4',
-            'title': 're:ČT Sport.*',
+            'title': 're:^ČT Sport \d{4}-\d{2}-\d{2} \d{2}:\d{2}$',
             'is_live': True,
         },
         'params': {
             # m3u8 download
             'skip_download': True,
         },
+        'skip': 'Georestricted to Czech Republic',
     }, {
         # video with 18+ caution trailer
         'url': 'http://www.ceskatelevize.cz/porady/10520528904-queer/215562210900007-bogotart/',
@@ -125,7 +140,7 @@ def _real_extract(self, url):
 
         entries = []
         for item in playlist:
-            is_live = item['type'] == 'LIVE'
+            is_live = item.get('type') == 'LIVE'
             formats = []
             for format_id, stream_url in item['streamUrls'].items():
                 formats.extend(self._extract_m3u8_formats(
@@ -147,15 +162,9 @@ def _real_extract(self, url):
                     subtitles = self.extract_subtitles(episode_id, subs)
 
             if playlist_len == 1:
+                final_title = playlist_title or title
                 if is_live:
-                    # live streams has channel name in title
-                    final_title = self._live_title(title)
-                elif playlist_title:
-                    # title is always set (no KeyError caught)
-                    # and gives good fallback
-                    final_title = title
-                else:
-                    final_title = playlist_title
+                    final_title = self._live_title(final_title)
             else:
                 final_title = '%s (%s)' % (playlist_title, title)