mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-09-04 21:45:22 +02:00
parent
bb58c9ed5c
commit
6ef5ad9e29
@ -1,5 +1,7 @@
|
|||||||
import itertools
|
import itertools
|
||||||
import json
|
import json
|
||||||
|
import random
|
||||||
|
import string
|
||||||
|
|
||||||
from .common import InfoExtractor
|
from .common import InfoExtractor
|
||||||
from ..utils import (
|
from ..utils import (
|
||||||
@ -15,10 +17,20 @@ class TrovoBaseIE(InfoExtractor):
|
|||||||
_VALID_URL_BASE = r'https?://(?:www\.)?trovo\.live/'
|
_VALID_URL_BASE = r'https?://(?:www\.)?trovo\.live/'
|
||||||
_HEADERS = {'Origin': 'https://trovo.live'}
|
_HEADERS = {'Origin': 'https://trovo.live'}
|
||||||
|
|
||||||
def _call_api(self, video_id, query=None, data=None):
|
def _call_api(self, video_id, data):
|
||||||
return self._download_json(
|
if 'persistedQuery' in data.get('extensions', {}):
|
||||||
'https://gql.trovo.live/', video_id, query=query, data=data,
|
url = 'https://gql.trovo.live'
|
||||||
headers={'Accept': 'application/json'})
|
else:
|
||||||
|
url = 'https://api-web.trovo.live/graphql'
|
||||||
|
|
||||||
|
resp = self._download_json(
|
||||||
|
url, video_id, data=json.dumps([data]).encode(), headers={'Accept': 'application/json'},
|
||||||
|
query={
|
||||||
|
'qid': ''.join(random.choices(string.ascii_uppercase + string.digits, k=10)),
|
||||||
|
})[0]
|
||||||
|
if 'errors' in resp:
|
||||||
|
raise ExtractorError(f'Trovo said: {resp["errors"][0]["message"]}')
|
||||||
|
return resp['data'][data['operationName']]
|
||||||
|
|
||||||
def _extract_streamer_info(self, data):
|
def _extract_streamer_info(self, data):
|
||||||
streamer_info = data.get('streamerInfo') or {}
|
streamer_info = data.get('streamerInfo') or {}
|
||||||
@ -35,27 +47,14 @@ class TrovoIE(TrovoBaseIE):
|
|||||||
|
|
||||||
def _real_extract(self, url):
|
def _real_extract(self, url):
|
||||||
username = self._match_id(url)
|
username = self._match_id(url)
|
||||||
live_info = self._call_api(username, query={
|
live_info = self._call_api(username, data={
|
||||||
'query': '''{
|
'operationName': 'live_LiveReaderService_GetLiveInfo',
|
||||||
getLiveInfo(params: {userName: "%s"}) {
|
'variables': {
|
||||||
isLive
|
'params': {
|
||||||
programInfo {
|
'userName': username,
|
||||||
coverUrl
|
},
|
||||||
id
|
},
|
||||||
streamInfo {
|
})
|
||||||
desc
|
|
||||||
playUrl
|
|
||||||
}
|
|
||||||
title
|
|
||||||
}
|
|
||||||
streamerInfo {
|
|
||||||
nickName
|
|
||||||
uid
|
|
||||||
userName
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}''' % username,
|
|
||||||
})['data']['getLiveInfo']
|
|
||||||
if live_info.get('isLive') == 0:
|
if live_info.get('isLive') == 0:
|
||||||
raise ExtractorError('%s is offline' % username, expected=True)
|
raise ExtractorError('%s is offline' % username, expected=True)
|
||||||
program_info = live_info['programInfo']
|
program_info = live_info['programInfo']
|
||||||
@ -90,56 +89,61 @@ def _real_extract(self, url):
|
|||||||
class TrovoVodIE(TrovoBaseIE):
|
class TrovoVodIE(TrovoBaseIE):
|
||||||
_VALID_URL = TrovoBaseIE._VALID_URL_BASE + r'(?:clip|video)/(?P<id>[^/?&#]+)'
|
_VALID_URL = TrovoBaseIE._VALID_URL_BASE + r'(?:clip|video)/(?P<id>[^/?&#]+)'
|
||||||
_TESTS = [{
|
_TESTS = [{
|
||||||
'url': 'https://trovo.live/video/ltv-100095501_100095501_1609596043',
|
'url': 'https://trovo.live/clip/lc-5285890818705062210?ltab=videos',
|
||||||
|
'params': {'getcomments': True},
|
||||||
'info_dict': {
|
'info_dict': {
|
||||||
'id': 'ltv-100095501_100095501_1609596043',
|
'id': 'lc-5285890818705062210',
|
||||||
'ext': 'mp4',
|
'ext': 'mp4',
|
||||||
'title': 'Spontaner 12 Stunden Stream! - Ok Boomer!',
|
'title': 'fatal moaning for a super good🤣🤣',
|
||||||
'uploader': 'Exsl',
|
'uploader': 'OneTappedYou',
|
||||||
'timestamp': 1609640305,
|
'timestamp': 1621628019,
|
||||||
'upload_date': '20210103',
|
'upload_date': '20210521',
|
||||||
'uploader_id': '100095501',
|
'uploader_id': '100719456',
|
||||||
'duration': 43977,
|
'duration': 31,
|
||||||
'view_count': int,
|
'view_count': int,
|
||||||
'like_count': int,
|
'like_count': int,
|
||||||
'comment_count': int,
|
'comment_count': int,
|
||||||
'comments': 'mincount:8',
|
'comments': 'mincount:1',
|
||||||
'categories': ['Grand Theft Auto V'],
|
'categories': ['Call of Duty: Mobile'],
|
||||||
|
'uploader_url': 'https://trovo.live/OneTappedYou',
|
||||||
|
'thumbnail': r're:^https?://.*\.jpg',
|
||||||
},
|
},
|
||||||
'skip': '404'
|
|
||||||
}, {
|
}, {
|
||||||
'url': 'https://trovo.live/clip/lc-5285890810184026005',
|
'url': 'https://trovo.live/video/ltv-100095501_100095501_1609596043',
|
||||||
'only_matching': True,
|
'only_matching': True,
|
||||||
}]
|
}]
|
||||||
|
|
||||||
def _real_extract(self, url):
|
def _real_extract(self, url):
|
||||||
vid = self._match_id(url)
|
vid = self._match_id(url)
|
||||||
resp = self._call_api(vid, data=json.dumps([{
|
|
||||||
'query': '''{
|
# NOTE: It is also possible to extract this info from the Nuxt data on the website,
|
||||||
batchGetVodDetailInfo(params: {vids: ["%s"]}) {
|
# however that seems unreliable - sometimes it randomly doesn't return the data,
|
||||||
VodDetailInfos
|
# at least when using a non-residential IP.
|
||||||
}
|
resp = self._call_api(vid, data={
|
||||||
}''' % vid,
|
'operationName': 'batchGetVodDetailInfo',
|
||||||
}, {
|
'variables': {
|
||||||
'query': '''{
|
'params': {
|
||||||
getCommentList(params: {appInfo: {postID: "%s"}, pageSize: 1000000000, preview: {}}) {
|
'vids': [vid],
|
||||||
commentList {
|
},
|
||||||
author {
|
},
|
||||||
nickName
|
'extensions': {
|
||||||
uid
|
'persistedQuery': {
|
||||||
}
|
'version': 1,
|
||||||
commentID
|
'sha256Hash': 'ceae0355d66476e21a1dd8e8af9f68de95b4019da2cda8b177c9a2255dad31d0',
|
||||||
content
|
},
|
||||||
createdAt
|
},
|
||||||
parentID
|
})
|
||||||
}
|
vod_detail_info = resp['VodDetailInfos'][vid]
|
||||||
}
|
|
||||||
}''' % vid,
|
|
||||||
}]).encode())
|
|
||||||
vod_detail_info = resp[0]['data']['batchGetVodDetailInfo']['VodDetailInfos'][vid]
|
|
||||||
vod_info = vod_detail_info['vodInfo']
|
vod_info = vod_detail_info['vodInfo']
|
||||||
title = vod_info['title']
|
title = vod_info['title']
|
||||||
|
|
||||||
|
if try_get(vod_info, lambda x: x['playbackRights']['playbackRights'] != 'Normal'):
|
||||||
|
playback_rights_setting = vod_info['playbackRights']['playbackRightsSetting']
|
||||||
|
if playback_rights_setting == 'SubscriberOnly':
|
||||||
|
raise ExtractorError('This video is only available for subscribers', expected=True)
|
||||||
|
else:
|
||||||
|
raise ExtractorError(f'This video is not available ({playback_rights_setting})', expected=True)
|
||||||
|
|
||||||
language = vod_info.get('languageName')
|
language = vod_info.get('languageName')
|
||||||
formats = []
|
formats = []
|
||||||
for play_info in (vod_info.get('playInfos') or []):
|
for play_info in (vod_info.get('playInfos') or []):
|
||||||
@ -163,23 +167,6 @@ def _real_extract(self, url):
|
|||||||
category = vod_info.get('categoryName')
|
category = vod_info.get('categoryName')
|
||||||
get_count = lambda x: int_or_none(vod_info.get(x + 'Num'))
|
get_count = lambda x: int_or_none(vod_info.get(x + 'Num'))
|
||||||
|
|
||||||
comment_list = try_get(resp, lambda x: x[1]['data']['getCommentList']['commentList'], list) or []
|
|
||||||
comments = []
|
|
||||||
for comment in comment_list:
|
|
||||||
content = comment.get('content')
|
|
||||||
if not content:
|
|
||||||
continue
|
|
||||||
author = comment.get('author') or {}
|
|
||||||
parent = comment.get('parentID')
|
|
||||||
comments.append({
|
|
||||||
'author': author.get('nickName'),
|
|
||||||
'author_id': str_or_none(author.get('uid')),
|
|
||||||
'id': str_or_none(comment.get('commentID')),
|
|
||||||
'text': content,
|
|
||||||
'timestamp': int_or_none(comment.get('createdAt')),
|
|
||||||
'parent': 'root' if parent == 0 else str_or_none(parent),
|
|
||||||
})
|
|
||||||
|
|
||||||
info = {
|
info = {
|
||||||
'id': vid,
|
'id': vid,
|
||||||
'title': title,
|
'title': title,
|
||||||
@ -190,12 +177,51 @@ def _real_extract(self, url):
|
|||||||
'view_count': get_count('watch'),
|
'view_count': get_count('watch'),
|
||||||
'like_count': get_count('like'),
|
'like_count': get_count('like'),
|
||||||
'comment_count': get_count('comment'),
|
'comment_count': get_count('comment'),
|
||||||
'comments': comments,
|
|
||||||
'categories': [category] if category else None,
|
'categories': [category] if category else None,
|
||||||
|
'__post_extractor': self.extract_comments(vid),
|
||||||
}
|
}
|
||||||
info.update(self._extract_streamer_info(vod_detail_info))
|
info.update(self._extract_streamer_info(vod_detail_info))
|
||||||
return info
|
return info
|
||||||
|
|
||||||
|
def _get_comments(self, vid):
|
||||||
|
for page in itertools.count(1):
|
||||||
|
comments_json = self._call_api(vid, data={
|
||||||
|
'operationName': 'getCommentList',
|
||||||
|
'variables': {
|
||||||
|
'params': {
|
||||||
|
'appInfo': {
|
||||||
|
'postID': vid,
|
||||||
|
},
|
||||||
|
'preview': {},
|
||||||
|
'pageSize': 99,
|
||||||
|
'page': page,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
'extensions': {
|
||||||
|
'persistedQuery': {
|
||||||
|
'version': 1,
|
||||||
|
'sha256Hash': 'be8e5f9522ddac7f7c604c0d284fd22481813263580849926c4c66fb767eed25',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
for comment in comments_json['commentList']:
|
||||||
|
content = comment.get('content')
|
||||||
|
if not content:
|
||||||
|
continue
|
||||||
|
author = comment.get('author') or {}
|
||||||
|
parent = comment.get('parentID')
|
||||||
|
yield {
|
||||||
|
'author': author.get('nickName'),
|
||||||
|
'author_id': str_or_none(author.get('uid')),
|
||||||
|
'id': str_or_none(comment.get('commentID')),
|
||||||
|
'text': content,
|
||||||
|
'timestamp': int_or_none(comment.get('createdAt')),
|
||||||
|
'parent': 'root' if parent == 0 else str_or_none(parent),
|
||||||
|
}
|
||||||
|
|
||||||
|
if comments_json['lastPage']:
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
class TrovoChannelBaseIE(TrovoBaseIE):
|
class TrovoChannelBaseIE(TrovoBaseIE):
|
||||||
def _get_vod_json(self, page, uid):
|
def _get_vod_json(self, page, uid):
|
||||||
@ -215,9 +241,15 @@ def _entries(self, uid):
|
|||||||
|
|
||||||
def _real_extract(self, url):
|
def _real_extract(self, url):
|
||||||
id = self._match_id(url)
|
id = self._match_id(url)
|
||||||
uid = str(self._call_api(id, query={
|
live_info = self._call_api(id, data={
|
||||||
'query': '{getLiveInfo(params:{userName:"%s"}){streamerInfo{uid}}}' % id
|
'operationName': 'live_LiveReaderService_GetLiveInfo',
|
||||||
})['data']['getLiveInfo']['streamerInfo']['uid'])
|
'variables': {
|
||||||
|
'params': {
|
||||||
|
'userName': id,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
uid = str(live_info['streamerInfo']['uid'])
|
||||||
return self.playlist_result(self._entries(uid), playlist_id=uid)
|
return self.playlist_result(self._entries(uid), playlist_id=uid)
|
||||||
|
|
||||||
|
|
||||||
@ -233,13 +265,25 @@ class TrovoChannelVodIE(TrovoChannelBaseIE):
|
|||||||
},
|
},
|
||||||
}]
|
}]
|
||||||
|
|
||||||
_QUERY = '{getChannelLtvVideoInfos(params:{pageSize:99,currPage:%d,channelID:%s}){hasMore,vodInfos{vid}}}'
|
|
||||||
_TYPE = 'video'
|
_TYPE = 'video'
|
||||||
|
|
||||||
def _get_vod_json(self, page, uid):
|
def _get_vod_json(self, page, uid):
|
||||||
return self._call_api(uid, query={
|
return self._call_api(uid, data={
|
||||||
'query': self._QUERY % (page, uid)
|
'operationName': 'getChannelLtvVideoInfos',
|
||||||
})['data']['getChannelLtvVideoInfos']
|
'variables': {
|
||||||
|
'params': {
|
||||||
|
'channelID': int(uid),
|
||||||
|
'pageSize': 99,
|
||||||
|
'currPage': page,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
'extensions': {
|
||||||
|
'persistedQuery': {
|
||||||
|
'version': 1,
|
||||||
|
'sha256Hash': '78fe32792005eab7e922cafcdad9c56bed8bbc5f5df3c7cd24fcb84a744f5f78',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
class TrovoChannelClipIE(TrovoChannelBaseIE):
|
class TrovoChannelClipIE(TrovoChannelBaseIE):
|
||||||
@ -254,10 +298,22 @@ class TrovoChannelClipIE(TrovoChannelBaseIE):
|
|||||||
},
|
},
|
||||||
}]
|
}]
|
||||||
|
|
||||||
_QUERY = '{getChannelClipVideoInfos(params:{pageSize:99,currPage:%d,channelID:%s,albumType:VOD_CLIP_ALBUM_TYPE_LATEST}){hasMore,vodInfos{vid}}}'
|
|
||||||
_TYPE = 'clip'
|
_TYPE = 'clip'
|
||||||
|
|
||||||
def _get_vod_json(self, page, uid):
|
def _get_vod_json(self, page, uid):
|
||||||
return self._call_api(uid, query={
|
return self._call_api(uid, data={
|
||||||
'query': self._QUERY % (page, uid)
|
'operationName': 'getChannelClipVideoInfos',
|
||||||
})['data']['getChannelClipVideoInfos']
|
'variables': {
|
||||||
|
'params': {
|
||||||
|
'channelID': int(uid),
|
||||||
|
'pageSize': 99,
|
||||||
|
'currPage': page,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
'extensions': {
|
||||||
|
'persistedQuery': {
|
||||||
|
'version': 1,
|
||||||
|
'sha256Hash': 'e7924bfe20059b5c75fc8ff9e7929f43635681a7bdf3befa01072ed22c8eff31',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
Loading…
Reference in New Issue
Block a user