# coding: utf-8 from __future__ import unicode_literals import json from .common import InfoExtractor from ..utils import ( ExtractorError, int_or_none, str_or_none, try_get, ) class TrovoBaseIE(InfoExtractor): _VALID_URL_BASE = r'https?://(?:www\.)?trovo\.live/' def _extract_streamer_info(self, data): streamer_info = data.get('streamerInfo') or {} username = streamer_info.get('userName') return { 'uploader': streamer_info.get('nickName'), 'uploader_id': str_or_none(streamer_info.get('uid')), 'uploader_url': 'https://trovo.live/' + username if username else None, } class TrovoIE(TrovoBaseIE): _VALID_URL = TrovoBaseIE._VALID_URL_BASE + r'(?!(?:clip|video)/)(?P<id>[^/?&#]+)' def _real_extract(self, url): username = self._match_id(url) live_info = self._download_json( 'https://gql.trovo.live/', username, query={ 'query': '''{ getLiveInfo(params: {userName: "%s"}) { isLive programInfo { coverUrl id streamInfo { desc playUrl } title } streamerInfo { nickName uid userName } } }''' % username, })['data']['getLiveInfo'] if live_info.get('isLive') == 0: raise ExtractorError('%s is offline' % username, expected=True) program_info = live_info['programInfo'] program_id = program_info['id'] title = self._live_title(program_info['title']) formats = [] for stream_info in (program_info.get('streamInfo') or []): play_url = stream_info.get('playUrl') if not play_url: continue format_id = stream_info.get('desc') formats.append({ 'format_id': format_id, 'height': int_or_none(format_id[:-1]) if format_id else None, 'url': play_url, }) self._sort_formats(formats) info = { 'id': program_id, 'title': title, 'formats': formats, 'thumbnail': program_info.get('coverUrl'), 'is_live': True, } info.update(self._extract_streamer_info(live_info)) return info class TrovoVodIE(TrovoBaseIE): _VALID_URL = TrovoBaseIE._VALID_URL_BASE + r'(?:clip|video)/(?P<id>[^/?&#]+)' _TESTS = [{ 'url': 'https://trovo.live/video/ltv-100095501_100095501_1609596043', 'info_dict': { 'id': 'ltv-100095501_100095501_1609596043', 'ext': 'mp4', 'title': 'Spontaner 12 Stunden Stream! - Ok Boomer!', 'uploader': 'Exsl', 'timestamp': 1609640305, 'upload_date': '20210103', 'uploader_id': '100095501', 'duration': 43977, 'view_count': int, 'like_count': int, 'comment_count': int, 'comments': 'mincount:8', 'categories': ['Grand Theft Auto V'], }, }, { 'url': 'https://trovo.live/clip/lc-5285890810184026005', 'only_matching': True, }] def _real_extract(self, url): vid = self._match_id(url) resp = self._download_json( 'https://gql.trovo.live/', vid, data=json.dumps([{ 'query': '''{ batchGetVodDetailInfo(params: {vids: ["%s"]}) { VodDetailInfos } }''' % vid, }, { 'query': '''{ getCommentList(params: {appInfo: {postID: "%s"}, pageSize: 1000000000, preview: {}}) { commentList { author { nickName uid } commentID content createdAt parentID } } }''' % vid, }]).encode(), headers={ 'Content-Type': 'application/json', }) vod_detail_info = resp[0]['data']['batchGetVodDetailInfo']['VodDetailInfos'][vid] vod_info = vod_detail_info['vodInfo'] title = vod_info['title'] language = vod_info.get('languageName') formats = [] for play_info in (vod_info.get('playInfos') or []): play_url = play_info.get('playUrl') if not play_url: continue format_id = play_info.get('desc') formats.append({ 'ext': 'mp4', 'filesize': int_or_none(play_info.get('fileSize')), 'format_id': format_id, 'height': int_or_none(format_id[:-1]) if format_id else None, 'language': language, 'protocol': 'm3u8_native', 'tbr': int_or_none(play_info.get('bitrate')), 'url': play_url, }) self._sort_formats(formats) category = vod_info.get('categoryName') 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 = { 'id': vid, 'title': title, 'formats': formats, 'thumbnail': vod_info.get('coverUrl'), 'timestamp': int_or_none(vod_info.get('publishTs')), 'duration': int_or_none(vod_info.get('duration')), 'view_count': get_count('watch'), 'like_count': get_count('like'), 'comment_count': get_count('comment'), 'comments': comments, 'categories': [category] if category else None, } info.update(self._extract_streamer_info(vod_detail_info)) return info