mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-18 15:11:45 +00:00
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
# coding: utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
from .common import InfoExtractor
|
|
from ..compat import compat_str
|
|
from ..utils import (
|
|
ExtractorError,
|
|
qualities
|
|
)
|
|
|
|
class PandaTVIE(InfoExtractor):
|
|
IE_DESC = '熊猫TV'
|
|
_VALID_URL = r'http://(?:www\.)?panda\.tv/(?P<id>[0-9]+)'
|
|
_TESTS = [{
|
|
'url': 'http://www.panda.tv/10091',
|
|
'info_dict': {
|
|
'id': '10091',
|
|
'title': 're:.+',
|
|
'uploader': '囚徒',
|
|
'ext': 'flv',
|
|
'is_live': True,
|
|
},
|
|
'params': {
|
|
'skip_download': 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']
|
|
|
|
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)
|
|
|
|
video_info = data['videoinfo']
|
|
|
|
# 2 = live, 3 = offline
|
|
if video_info.get('status') != '2':
|
|
raise ExtractorError(
|
|
'Live stream is offline', expected=True)
|
|
|
|
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'})
|
|
|
|
plflag0, plflag1 = video_info['plflag'].split('_')
|
|
plflag0 = int(plflag0) - 1
|
|
if plflag1 == '21':
|
|
plflag0 = 10
|
|
plflag1 = '4'
|
|
live_panda = 'live_panda' if plflag0 < 1 else ''
|
|
|
|
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,
|
|
})
|
|
self._sort_formats(formats)
|
|
|
|
return {
|
|
'id': video_id,
|
|
'title': self._live_title(title),
|
|
'uploader': uploader,
|
|
'formats': formats,
|
|
'is_live': True,
|
|
}
|