1
0
mirror of https://github.com/l1ving/youtube-dl synced 2025-03-13 18:40:01 +08:00
2017-01-04 01:51:08 +03:00

85 lines
2.9 KiB
Python

# coding: utf-8
from __future__ import unicode_literals
from .common import InfoExtractor
from ..utils import (
int_or_none,
ExtractorError,
compat_str,
clean_html,
parse_iso8601,
)
class BeamProLiveIE(InfoExtractor):
IE_NAME = 'Beam:live'
_VALID_URL = r'https?://(?:\w+.)?beam.pro/(?P<id>[^?]+)$'
_API_CHANNEL = 'https://beam.pro/api/v1/channels/{0}'
_API_MANIFEST = 'https://beam.pro/api/v1/channels/{0}/manifest.{1}'
_VALID_MANIFESTS = ('smil', 'm3u8', 'light', 'light2', 'ftl', 'ftlOld')
_RATINGS = {'family': 0, 'teen': 13, '18+': 18}
_TEST = {
'url': 'http://www.beam.pro/niterhayven',
'info_dict': {
'id': '261562',
'ext': 'mp4',
'uploader': 'niterhayven',
'timestamp': 1483477281,
'age_limit': 18,
'title': 'Introducing The Witcher 3 // The Grind Starts Now!',
'thumbnail': r're:https://.*\.jpg$',
'upload_date': '20170103',
'uploader_id': 373396,
'description': 'md5:0b161ac080f15fe05d18a07adb44a74d',
'is_live': True,
},
# 'skip': r're:.* is offline$',
'params': {
'skip_download': True,
},
}
def _real_extract(self, url):
channel_id = self._match_id(url)
chan_data = self._download_json(self._API_CHANNEL.format(channel_id), channel_id)
if not chan_data.get('online'):
raise ExtractorError('{0} is offline'.format(channel_id), expected=True)
formats = self._extract_m3u8_formats(
self._API_MANIFEST.format(
chan_data.get('id'),
self._VALID_MANIFESTS[1]), channel_id, ext='mp4',
)
self._sort_formats(formats, 'vbr')
info = {}
info['formats'] = formats
if chan_data:
info.update(self._extract_info(chan_data))
return info
def _rating_to_age(self, rating):
return self._RATINGS[rating] if rating in self._RATINGS else None
def _extract_info(self, info):
thumbnail = info['thumbnail'].get('url') if info.get('thumbnail') else None
username = info['user'].get('url') if info.get('username') else None
return {
'id': compat_str(info['id']),
'title': info.get('name') or 'Untitled Broadcast',
'description': clean_html(info.get('description')),
'age_limit': self._rating_to_age(info.get('audience')),
'is_live': True if info.get('online') else False,
'timestamp': parse_iso8601(info.get('updatedAt')),
# 'release_date': info.get('createdAt'),
# 'upload_date': info.get('updatedAt'),
# 'formats': formats,
'uploader': info.get('token') or username,
'uploader_id': int_or_none(info.get('userId')),
'view_count': int_or_none(info.get('viewersTotal')),
'thumbnail': thumbnail,
}