|
|
- # coding: utf-8
- from __future__ import unicode_literals
-
- import re
- import time
-
- from .common import InfoExtractor
- from ..compat import compat_str
- from ..utils import (
- ExtractorError,
- js_to_json,
- try_get,
- update_url_query,
- urlencode_postdata,
- )
-
-
- class PicartoIE(InfoExtractor):
- _VALID_URL = r'https?://(?:www.)?picarto\.tv/(?P<id>[a-zA-Z0-9]+)(?:/(?P<token>[a-zA-Z0-9]+))?'
- _TEST = {
- 'url': 'https://picarto.tv/Setz',
- 'info_dict': {
- 'id': 'Setz',
- 'ext': 'mp4',
- 'title': 're:^Setz [0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}$',
- 'timestamp': int,
- 'is_live': True
- },
- 'skip': 'Stream is offline',
- }
-
- @classmethod
- def suitable(cls, url):
- return False if PicartoVodIE.suitable(url) else super(PicartoIE, cls).suitable(url)
-
- def _real_extract(self, url):
- mobj = re.match(self._VALID_URL, url)
- channel_id = mobj.group('id')
-
- metadata = self._download_json(
- 'https://api.picarto.tv/v1/channel/name/' + channel_id,
- channel_id)
-
- if metadata.get('online') is False:
- raise ExtractorError('Stream is offline', expected=True)
-
- cdn_data = self._download_json(
- 'https://picarto.tv/process/channel', channel_id,
- data=urlencode_postdata({'loadbalancinginfo': channel_id}),
- note='Downloading load balancing info')
-
- token = mobj.group('token') or 'public'
- params = {
- 'con': int(time.time() * 1000),
- 'token': token,
- }
-
- prefered_edge = cdn_data.get('preferedEdge')
- formats = []
-
- for edge in cdn_data['edges']:
- edge_ep = edge.get('ep')
- if not edge_ep or not isinstance(edge_ep, compat_str):
- continue
- edge_id = edge.get('id')
- for tech in cdn_data['techs']:
- tech_label = tech.get('label')
- tech_type = tech.get('type')
- preference = 0
- if edge_id == prefered_edge:
- preference += 1
- format_id = []
- if edge_id:
- format_id.append(edge_id)
- if tech_type == 'application/x-mpegurl' or tech_label == 'HLS':
- format_id.append('hls')
- formats.extend(self._extract_m3u8_formats(
- update_url_query(
- 'https://%s/hls/%s/index.m3u8'
- % (edge_ep, channel_id), params),
- channel_id, 'mp4', preference=preference,
- m3u8_id='-'.join(format_id), fatal=False))
- continue
- elif tech_type == 'video/mp4' or tech_label == 'MP4':
- format_id.append('mp4')
- formats.append({
- 'url': update_url_query(
- 'https://%s/mp4/%s.mp4' % (edge_ep, channel_id),
- params),
- 'format_id': '-'.join(format_id),
- 'preference': preference,
- })
- else:
- # rtmp format does not seem to work
- continue
- self._sort_formats(formats)
-
- mature = metadata.get('adult')
- if mature is None:
- age_limit = None
- else:
- age_limit = 18 if mature is True else 0
-
- return {
- 'id': channel_id,
- 'title': self._live_title(metadata.get('title') or channel_id),
- 'is_live': True,
- 'thumbnail': try_get(metadata, lambda x: x['thumbnails']['web']),
- 'channel': channel_id,
- 'channel_url': 'https://picarto.tv/%s' % channel_id,
- 'age_limit': age_limit,
- 'formats': formats,
- }
-
-
- class PicartoVodIE(InfoExtractor):
- _VALID_URL = r'https?://(?:www.)?picarto\.tv/videopopout/(?P<id>[^/?#&]+)'
- _TESTS = [{
- 'url': 'https://picarto.tv/videopopout/ArtofZod_2017.12.12.00.13.23.flv',
- 'md5': '3ab45ba4352c52ee841a28fb73f2d9ca',
- 'info_dict': {
- 'id': 'ArtofZod_2017.12.12.00.13.23.flv',
- 'ext': 'mp4',
- 'title': 'ArtofZod_2017.12.12.00.13.23.flv',
- 'thumbnail': r're:^https?://.*\.jpg'
- },
- }, {
- 'url': 'https://picarto.tv/videopopout/Plague',
- 'only_matching': True,
- }]
-
- def _real_extract(self, url):
- video_id = self._match_id(url)
-
- webpage = self._download_webpage(url, video_id)
-
- vod_info = self._parse_json(
- self._search_regex(
- r'(?s)#vod-player["\']\s*,\s*(\{.+?\})\s*\)', webpage,
- video_id),
- video_id, transform_source=js_to_json)
-
- formats = self._extract_m3u8_formats(
- vod_info['vod'], video_id, 'mp4', entry_protocol='m3u8_native',
- m3u8_id='hls')
- self._sort_formats(formats)
-
- return {
- 'id': video_id,
- 'title': video_id,
- 'thumbnail': vod_info.get('vodThumb'),
- 'formats': formats,
- }
|
