[itv] extract subtitles
authorRemita Amine <remitamine@gmail.com>
Sat, 28 Jan 2017 16:25:15 +0000 (17:25 +0100)
committerRemita Amine <remitamine@gmail.com>
Sat, 28 Jan 2017 16:26:14 +0000 (17:26 +0100)
youtube_dl/extractor/itv.py

index d029609c3fdc56a5372e938baa5e90acb5290c45..d65cdc6af3fc87c13b6e799ddfdb1e138d6b8c0c 100644 (file)
@@ -20,7 +20,7 @@ from ..utils import (
 
 
 class ITVIE(InfoExtractor):
-    _VALID_URL = r'https?://(?:www\.)?itv\.com/hub/[^/]+/(?P<id>[0-9a-z]+)'
+    _VALID_URL = r'https?://(?:www\.)?itv\.com/hub/[^/]+/(?P<id>[0-9a-zA-Z]+)'
     _TEST = {
         'url': 'http://www.itv.com/hub/mr-bean-animated-series/2a2936a0053',
         'info_dict': {
@@ -98,7 +98,8 @@ class ITVIE(InfoExtractor):
             fault_string = xpath_text(resp_env, './/faultstring')
             raise ExtractorError('%s said: %s' % (self.IE_NAME, fault_string))
         title = xpath_text(playlist, 'EpisodeTitle', fatal=True)
-        media_files = xpath_element(playlist, 'VideoEntries/Video/MediaFiles', fatal=True)
+        video_element = xpath_element(playlist, 'VideoEntries/Video', fatal=True)
+        media_files = xpath_element(video_element, 'MediaFiles', fatal=True)
         rtmp_url = media_files.attrib['base']
 
         formats = []
@@ -170,10 +171,21 @@ class ITVIE(InfoExtractor):
                         })
         self._sort_formats(formats)
 
+        subtitles = {}
+        for caption_url in video_element.findall('ClosedCaptioningURIs/URL'):
+            if not caption_url.text:
+                continue
+            ext = determine_ext(caption_url.text, 'ttml')
+            subtitles.setdefault('en', []).append({
+                'url': caption_url,
+                'ext': 'ttml' if ext == 'xml' else ext,
+            })
+
         return {
             'id': video_id,
             'title': title,
             'formats': formats,
+            'subtitles': subtitles,
             'episode_title': title,
             'episode_number': int_or_none(xpath_text(playlist, 'EpisodeNumber')),
             'series': xpath_text(playlist, 'ProgrammeTitle'),