[youtube] Fix extraction.

[youtube-dl] / youtube_dl / extractor / kusi.py

diff --git a/youtube_dl/extractor/kusi.py b/youtube_dl/extractor/kusi.py
index 5e87c5f79934f1531874524facbcf40e60b88bd1..6a7e3baa70cc019b497828ab84d176e55216356f 100644 (file)
--- a/youtube_dl/extractor/kusi.py
+++ b/youtube_dl/extractor/kusi.py
@@ -1,66 +1,80 @@
 # coding: utf-8
 from __future__ import unicode_literals
 
+import random
 import re
 
 from .common import InfoExtractor
 from ..compat import compat_urllib_parse_unquote_plus
-from ..utils import int_or_none
+from ..utils import (
+    int_or_none,
+    float_or_none,
+    timeconvert,
+    update_url_query,
+    xpath_text,
+)
 
 
 class KUSIIE(InfoExtractor):
-    _VALID_URL = r'http://(?:www\.)?kusi\.com/(?P<path>story/.+|video\?clipId=(?P<clipId>\d+))'
+    _VALID_URL = r'https?://(?:www\.)?kusi\.com/(?P<path>story/.+|video\?clipId=(?P<clipId>\d+))'
     _TESTS = [{
-        'url': 'http://www.kusi.com/story/31183873/turko-files-case-closed-put-on-hold',
-        'md5': 'f926e7684294cf8cb7bdf8858e1b3988',
+        'url': 'http://www.kusi.com/story/32849881/turko-files-refused-to-help-it-aint-right',
+        'md5': '4e76ce8e53660ce9697d06c0ba6fc47d',
         'info_dict': {
-            'id': '12203019',
+            'id': '12689020',
             'ext': 'mp4',
-            'title': 'Turko Files: Case Closed! & Put On Hold!',
-            'duration': 231000,
+            'title': "Turko Files: Refused to Help, It Ain't Right!",
+            'duration': 223.586,
+            'upload_date': '20160826',
+            'timestamp': 1472233118,
+            'thumbnail': r're:^https?://.*\.jpg$'
         },
     }, {
         'url': 'http://kusi.com/video?clipId=12203019',
-        'info_dict': {
-            'id': '12203019',
-            'ext': 'mp4',
-            'title': 'Turko Files: Case Closed! & Put On Hold!',
-            'duration': 231000,
-        },
-        'params': {
-            'skip_download': True,  # Same as previous one
-        },
+        'only_matching': True,
     }]
 
     def _real_extract(self, url):
         mobj = re.match(self._VALID_URL, url)
+        clip_id = mobj.group('clipId')
+        video_id = clip_id or mobj.group('path')
+
+        webpage = self._download_webpage(url, video_id)
+
+        if clip_id is None:
+            video_id = clip_id = self._html_search_regex(
+                r'"clipId"\s*,\s*"(\d+)"', webpage, 'clip id')
+
+        affiliate_id = self._search_regex(
+            r'affiliateId\s*:\s*\'([^\']+)\'', webpage, 'affiliate id')
 
-        if mobj.group('clipId') is not None:
-            video_id = mobj.group('clipId')
-        else:
-            webpage = self._download_webpage(url, mobj.group('path'))
-            video_id = self._html_search_regex(r'"clipId", "(\d+)"', webpage,
-                                               'clipId')
+        # See __Packages/worldnow/model/GalleryModel.as of WNGallery.swf
+        xml_url = update_url_query('http://www.kusi.com/build.asp', {
+            'buildtype': 'buildfeaturexmlrequest',
+            'featureType': 'Clip',
+            'featureid': clip_id,
+            'affiliateno': affiliate_id,
+            'clientgroupid': '1',
+            'rnd': int(round(random.random() * 1000000)),
+        })
 
-        xml_url = 'http://www.kusi.com/build.asp?buildtype=buildfeaturexml'\
-                  'request&featureType=Clip&featureid={0}&affiliateno=956&'\
-                  'clientgroupid=1&rnd=562461'.format(video_id)
-        doc = self._download_xml(xml_url, video_id,
-                                 note='Downloading video info',
-                                 errnote='Failed to download video info')
+        doc = self._download_xml(xml_url, video_id)
 
-        video_title = doc.find('HEADLINE').text
-        duration = int_or_none(doc.find('DURATION'), get_attr='text')
-        description = doc.find('ABSTRACT')
+        video_title = xpath_text(doc, 'HEADLINE', fatal=True)
+        duration = float_or_none(xpath_text(doc, 'DURATION'), scale=1000)
+        description = xpath_text(doc, 'ABSTRACT')
+        thumbnail = xpath_text(doc, './THUMBNAILIMAGE/FILENAME')
+        createtion_time = timeconvert(xpath_text(doc, 'rfc822creationdate'))
 
         quality_options = doc.find('{http://search.yahoo.com/mrss/}group').findall('{http://search.yahoo.com/mrss/}content')
         formats = []
         for quality in quality_options:
-            if 'height' in quality.attrib:
-                formats.append({
-                    'url': compat_urllib_parse_unquote_plus(quality.attrib['url']),
-                    'height': quality.attrib['height'],
-                })
+            formats.append({
+                'url': compat_urllib_parse_unquote_plus(quality.attrib['url']),
+                'height': int_or_none(quality.attrib.get('height')),
+                'width': int_or_none(quality.attrib.get('width')),
+                'vbr': float_or_none(quality.attrib.get('bitratebits'), scale=1000),
+            })
         self._sort_formats(formats)
 
         return {
@@ -69,4 +83,6 @@ class KUSIIE(InfoExtractor):
             'description': description,
             'duration': duration,
             'formats': formats,
+            'thumbnail': thumbnail,
+            'timestamp': createtion_time,
         }