[youtube] fix hd720 format position
[youtube-dl] / youtube_dl / extractor / pandatv.py
index 84014f3c531480e3428abbe4a0c1301fbd0f9c2d..13a2e7efc76e4f0fbd9e778644781c5218770f61 100644 (file)
@@ -2,46 +2,47 @@
 from __future__ import unicode_literals
 
 from .common import InfoExtractor
-from ..compat import compat_str
 from ..utils import (
     ExtractorError,
-    qualities
+    qualities,
 )
 
+
 class PandaTVIE(InfoExtractor):
     IE_DESC = '熊猫TV'
-    _VALID_URL = r'http://(?:www\.)?panda\.tv/(?P<id>[0-9]+)'
+    _VALID_URL = r'https?://(?:www\.)?panda\.tv/(?P<id>[0-9]+)'
     _TESTS = [{
-        'url': 'http://www.panda.tv/10091',
+        'url': 'http://www.panda.tv/66666',
         'info_dict': {
-            'id': '10091',
+            'id': '66666',
             'title': 're:.+',
-            'uploader': 'å\9b\9aå¾\92',
+            'uploader': 'å\88\98æ\9d\80鸡',
             'ext': 'flv',
             'is_live': True,
         },
         'params': {
             'skip_download': True,
         },
+        'skip': 'Live stream is offline',
+    }, {
+        'url': 'https://www.panda.tv/66666',
+        'only_matching': True,
     }]
 
     def _real_extract(self, url):
         video_id = self._match_id(url)
 
         config = self._download_json(
-            'http://www.panda.tv/api_room?roomid=%s' % video_id,
-            video_id
-        )
-
-        data = config['data']
+            'https://www.panda.tv/api_room_v2?roomid=%s' % video_id, video_id)
 
         error_code = config.get('errno', 0)
         if error_code is not 0:
-            error_desc = 'Server reported error %i' % error_code
-            if isinstance(data, compat_str):
-                error_desc += ': ' + data
-            raise ExtractorError(error_desc, expected=True)
+            raise ExtractorError(
+                '%s returned error %s: %s'
+                % (self.IE_NAME, error_code, config['errmsg']),
+                expected=True)
 
+        data = config['data']
         video_info = data['videoinfo']
 
         # 2 = live, 3 = offline
@@ -52,8 +53,12 @@ class PandaTVIE(InfoExtractor):
         title = data['roominfo']['name']
         uploader = data.get('hostinfo', {}).get('name')
         room_key = video_info['room_key']
-        stream_addr = video_info.get('stream_addr', {'OD': '1', 'HD': '1', 'SD': '1'})
+        stream_addr = video_info.get(
+            'stream_addr', {'OD': '1', 'HD': '1', 'SD': '1'})
 
+        # Reverse engineered from web player swf
+        # (http://s6.pdim.gs/static/07153e425f581151.swf at the moment of
+        # writing).
         plflag0, plflag1 = video_info['plflag'].split('_')
         plflag0 = int(plflag0) - 1
         if plflag1 == '21':
@@ -61,18 +66,28 @@ class PandaTVIE(InfoExtractor):
             plflag1 = '4'
         live_panda = 'live_panda' if plflag0 < 1 else ''
 
+        plflag_auth = self._parse_json(video_info['plflag_list'], video_id)
+        sign = plflag_auth['auth']['sign']
+        ts = plflag_auth['auth']['time']
+        rid = plflag_auth['auth']['rid']
+
         quality_key = qualities(['OD', 'HD', 'SD'])
         suffix = ['_small', '_mid', '']
         formats = []
         for k, v in stream_addr.items():
-            if v == '1':
-                quality = quality_key(k)
-                if quality >= 0:
-                    formats.append({
-                        'url': 'http://pl%s.live.panda.tv/live_panda/%s%s%s.flv' % (plflag1, room_key, live_panda, suffix[quality]),
-                        'format_id': k,
-                        'quality': quality,
-                    })
+            if v != '1':
+                continue
+            quality = quality_key(k)
+            if quality <= 0:
+                continue
+            for pref, (ext, pl) in enumerate((('m3u8', '-hls'), ('flv', ''))):
+                formats.append({
+                    'url': 'https://pl%s%s.live.panda.tv/live_panda/%s%s%s.%s?sign=%s&ts=%s&rid=%s'
+                    % (pl, plflag1, room_key, live_panda, suffix[quality], ext, sign, ts, rid),
+                    'format_id': '%s-%s' % (k, ext),
+                    'quality': quality,
+                    'source_preference': pref,
+                })
         self._sort_formats(formats)
 
         return {