]> git.bitcoin.ninja Git - youtube-dl/blob - youtube_dl/extractor/pandatv.py
[vlive] Add support for channels
[youtube-dl] / youtube_dl / extractor / pandatv.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 from .common import InfoExtractor
5 from ..utils import (
6     ExtractorError,
7     qualities,
8 )
9
10
11 class PandaTVIE(InfoExtractor):
12     IE_DESC = '熊猫TV'
13     _VALID_URL = r'http://(?:www\.)?panda\.tv/(?P<id>[0-9]+)'
14     _TEST = {
15         'url': 'http://www.panda.tv/10091',
16         'info_dict': {
17             'id': '10091',
18             'title': 're:.+',
19             'uploader': '囚徒',
20             'ext': 'flv',
21             'is_live': True,
22         },
23         'params': {
24             'skip_download': True,
25         },
26         'skip': 'Live stream is offline',
27     }
28
29     def _real_extract(self, url):
30         video_id = self._match_id(url)
31
32         config = self._download_json(
33             'http://www.panda.tv/api_room?roomid=%s' % video_id, video_id)
34
35         error_code = config.get('errno', 0)
36         if error_code is not 0:
37             raise ExtractorError(
38                 '%s returned error %s: %s'
39                 % (self.IE_NAME, error_code, config['errmsg']),
40                 expected=True)
41
42         data = config['data']
43         video_info = data['videoinfo']
44
45         # 2 = live, 3 = offline
46         if video_info.get('status') != '2':
47             raise ExtractorError(
48                 'Live stream is offline', expected=True)
49
50         title = data['roominfo']['name']
51         uploader = data.get('hostinfo', {}).get('name')
52         room_key = video_info['room_key']
53         stream_addr = video_info.get(
54             'stream_addr', {'OD': '1', 'HD': '1', 'SD': '1'})
55
56         # Reverse engineered from web player swf
57         # (http://s6.pdim.gs/static/07153e425f581151.swf at the moment of
58         # writing).
59         plflag0, plflag1 = video_info['plflag'].split('_')
60         plflag0 = int(plflag0) - 1
61         if plflag1 == '21':
62             plflag0 = 10
63             plflag1 = '4'
64         live_panda = 'live_panda' if plflag0 < 1 else ''
65
66         quality_key = qualities(['OD', 'HD', 'SD'])
67         suffix = ['_small', '_mid', '']
68         formats = []
69         for k, v in stream_addr.items():
70             if v != '1':
71                 continue
72             quality = quality_key(k)
73             if quality <= 0:
74                 continue
75             for pref, (ext, pl) in enumerate((('m3u8', '-hls'), ('flv', ''))):
76                 formats.append({
77                     'url': 'http://pl%s%s.live.panda.tv/live_panda/%s%s%s.%s'
78                     % (pl, plflag1, room_key, live_panda, suffix[quality], ext),
79                     'format_id': '%s-%s' % (k, ext),
80                     'quality': quality,
81                     'source_preference': pref,
82                 })
83         self._sort_formats(formats)
84
85         return {
86             'id': video_id,
87             'title': self._live_title(title),
88             'uploader': uploader,
89             'formats': formats,
90             'is_live': True,
91         }