Initiate new release.

[youtubedl] / youtube_dl / extractor / ceskatelevize.py

diff --git a/youtube_dl/extractor/ceskatelevize.py b/youtube_dl/extractor/ceskatelevize.py
index f70e090bb5b01942713149493e48bc0e51f7f74b..dda583680a03ba3cb420beb74a99af2ec60cbc83 100644 (file)
--- a/youtube_dl/extractor/ceskatelevize.py
+++ b/youtube_dl/extractor/ceskatelevize.py
@@ -3,10 +3,11 @@ from __future__ import unicode_literals
 
 import re
 
 
 import re
 

-from .subtitles import SubtitlesInfoExtractor
+from .common import InfoExtractor

 from ..compat import (
     compat_urllib_request,
     compat_urllib_parse,
 from ..compat import (
     compat_urllib_request,
     compat_urllib_parse,

+    compat_urllib_parse_unquote,

     compat_urllib_parse_urlparse,
 )
 from ..utils import (
     compat_urllib_parse_urlparse,
 )
 from ..utils import (


@@ -15,7 +16,7 @@ from ..utils import (
 )
 
 
 )
 
 

-class CeskaTelevizeIE(SubtitlesInfoExtractor):
+class CeskaTelevizeIE(InfoExtractor):

     _VALID_URL = r'https?://www\.ceskatelevize\.cz/(porady|ivysilani)/(.+/)?(?P<id>[^?#]+)'
 
     _TESTS = [
     _VALID_URL = r'https?://www\.ceskatelevize\.cz/(porady|ivysilani)/(.+/)?(?P<id>[^?#]+)'
 
     _TESTS = [


@@ -88,7 +89,7 @@ class CeskaTelevizeIE(SubtitlesInfoExtractor):
         if playlist_url == 'error_region':
             raise ExtractorError(NOT_AVAILABLE_STRING, expected=True)
 
         if playlist_url == 'error_region':
             raise ExtractorError(NOT_AVAILABLE_STRING, expected=True)
 

-        req = compat_urllib_request.Request(compat_urllib_parse.unquote(playlist_url))
+        req = compat_urllib_request.Request(compat_urllib_parse_unquote(playlist_url))

         req.add_header('Referer', url)
 
         playlist = self._download_json(req, video_id)
         req.add_header('Referer', url)
 
         playlist = self._download_json(req, video_id)


@@ -107,13 +108,7 @@ class CeskaTelevizeIE(SubtitlesInfoExtractor):
         subtitles = {}
         subs = item.get('subtitles')
         if subs:
         subtitles = {}
         subs = item.get('subtitles')
         if subs:

-            subtitles['cs'] = subs[0]['url']
-
-        if self._downloader.params.get('listsubtitles', False):
-            self._list_available_subtitles(video_id, subtitles)
-            return
-
-        subtitles = self._fix_subtitles(self.extract_subtitles(video_id, subtitles))
+            subtitles = self.extract_subtitles(episode_id, subs)

 
         return {
             'id': episode_id,
 
         return {
             'id': episode_id,


@@ -125,11 +120,20 @@ class CeskaTelevizeIE(SubtitlesInfoExtractor):
             'subtitles': subtitles,
         }
 
             'subtitles': subtitles,
         }
 

+    def _get_subtitles(self, episode_id, subs):
+        original_subtitles = self._download_webpage(
+            subs[0]['url'], episode_id, 'Downloading subtitles')
+        srt_subs = self._fix_subtitles(original_subtitles)
+        return {
+            'cs': [{
+                'ext': 'srt',
+                'data': srt_subs,
+            }]
+        }
+

     @staticmethod
     def _fix_subtitles(subtitles):
         """ Convert millisecond-based subtitles to SRT """
     @staticmethod
     def _fix_subtitles(subtitles):
         """ Convert millisecond-based subtitles to SRT """

-        if subtitles is None:
-            return subtitles  # subtitles not requested

 
         def _msectotimecode(msec):
             """ Helper utility to convert milliseconds to timecode """
 
         def _msectotimecode(msec):
             """ Helper utility to convert milliseconds to timecode """


@@ -149,7 +153,4 @@ class CeskaTelevizeIE(SubtitlesInfoExtractor):
                 else:
                     yield line
 
                 else:
                     yield line
 

-        fixed_subtitles = {}
-        for k, v in subtitles.items():
-            fixed_subtitles[k] = "\r\n".join(_fix_subtitle(v))
-        return fixed_subtitles
+        return "\r\n".join(_fix_subtitle(subtitles))