mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-07 20:30:41 -05:00
111 lines
4.3 KiB
Python
111 lines
4.3 KiB
Python
# coding: utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
import re
|
|
|
|
from .common import InfoExtractor
|
|
|
|
from ..utils import (
|
|
js_to_json,
|
|
try_get,
|
|
int_or_none,
|
|
str_or_none,
|
|
url_or_none,
|
|
)
|
|
from ..compat import compat_str
|
|
|
|
|
|
class TrovoLiveIE(InfoExtractor):
|
|
_VALID_URL = r'https?://(?:www\.)?trovo\.live/video/(?P<id>[\w-]+)'
|
|
_TEST = {
|
|
'url': 'https://trovo.live/video/ltv-100759829_100759829_1610625308',
|
|
'md5': 'ea7b58427910e9af66a462d895201a30',
|
|
'info_dict': {
|
|
'id': 'ltv-100759829_100759829_1610625308',
|
|
'ext': 'ts',
|
|
'title': 'GTA RP ASTERIX doa najjaca',
|
|
'uploader': 'Peroo42',
|
|
'duration': 5872,
|
|
'view_count': int,
|
|
'like_count': int,
|
|
'comment_count': int,
|
|
'categories': list,
|
|
'is_live': False,
|
|
'thumbnail': r're:^https?://.*\.jpg$',
|
|
'uploader_id': '100759829',
|
|
}
|
|
}
|
|
|
|
def _real_extract(self, url):
|
|
video_id = self._match_id(url)
|
|
webpage = self._download_webpage(url, video_id)
|
|
nuxt = self._search_regex(r'\bwindow\.__NUXT__\s*=\s*(.+?);?\s*</script>', webpage, 'nuxt', default='')
|
|
mobj = re.search(r'\((?P<arg_names>[^(]+)\)\s*{\s*return\s+(?P<json>{.+})\s*\((?P<args>.+?)\)\s*\)$', nuxt)
|
|
|
|
vod_details = vod_info = {}
|
|
if mobj:
|
|
vod_details = self._parse_json(
|
|
js_to_json(
|
|
self._search_regex(r'VodDetailInfos\s*:({.+?}),\s*_', webpage, 'VodDetailInfos'),
|
|
dict(zip(
|
|
(i.strip() for i in mobj.group('arg_names').split(',')),
|
|
(i.strip() for i in mobj.group('args').split(','))))),
|
|
video_id)
|
|
vod_info = try_get(vod_details, lambda x: x['json'][video_id]['vodInfo'], dict) or {}
|
|
|
|
player_info = self._parse_json(
|
|
self._search_regex(
|
|
r'_playerInfo\s*=\s*({.+?})\s*</script>', webpage, 'player info'),
|
|
video_id)
|
|
|
|
title = (
|
|
vod_info.get('title')
|
|
or self._html_search_regex(r'<h3>(.+?)</h3>', webpage, 'title', fatal=False)
|
|
or self._og_search_title(webpage))
|
|
uploader = (
|
|
try_get(vod_details, lambda x: x['json'][video_id]['streamerInfo']['userName'], compat_str)
|
|
or self._search_regex(r'<div[^>]+userName\s=\s[\'"](.+?)[\'"]', webpage, 'uploader', fatal=False))
|
|
|
|
format_dicts = vod_info.get('playInfos') or player_info.get('urlArray') or []
|
|
|
|
def _extract_format_data(format_dict):
|
|
res = format_dict.get('desc')
|
|
enc = str_or_none(format_dict.get('encodeType'))
|
|
if enc:
|
|
notes = [enc.replace('VOD_ENCODE_TYPE_', '')]
|
|
level = str_or_none(format_dict.get('levelType'))
|
|
if level:
|
|
notes.append('level %s' % level)
|
|
height = int_or_none(res[:-1]) if res else None
|
|
bitrate = format_dict.get('bitrate')
|
|
fid = res or ('%sk' % str_or_none(bitrate) if bitrate else None) or ' '.join(notes)
|
|
|
|
return {
|
|
'url': format_dict['playUrl'],
|
|
'format_id': fid,
|
|
'format_note': ' '.join(notes),
|
|
'height': height,
|
|
'resolution': str_or_none(res),
|
|
'tbr': int_or_none(bitrate),
|
|
'filesize': int_or_none(format_dict.get('fileSize')),
|
|
'vcodec': 'avc3',
|
|
'acodec': 'aac',
|
|
'ext': 'ts'
|
|
}
|
|
|
|
formats = [_extract_format_data(f) for f in format_dicts]
|
|
self._sort_formats(formats)
|
|
return {
|
|
'id': video_id,
|
|
'title': title,
|
|
'uploader': uploader,
|
|
'duration': int_or_none(vod_info.get('duration')),
|
|
'formats': formats,
|
|
'view_count': int_or_none(vod_info.get('watchNum')),
|
|
'like_count': int_or_none(vod_info.get('likeNum')),
|
|
'comment_count': int_or_none(vod_info.get('commentNum')),
|
|
'categories': [str_or_none(vod_info.get('categoryName'))],
|
|
'is_live': try_get(player_info, lambda x: x['isLive'], bool),
|
|
'thumbnail': url_or_none(vod_info.get('coverUrl')),
|
|
'uploader_id': str_or_none(try_get(vod_details, lambda x: x['json'][video_id]['streamerInfo']['uid'])),
|
|
}
|