diff options
Diffstat (limited to 'youtube_dlc/extractor/trovolive.py')
| -rw-r--r-- | youtube_dlc/extractor/trovolive.py | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/youtube_dlc/extractor/trovolive.py b/youtube_dlc/extractor/trovolive.py new file mode 100644 index 000000000..8ad3ebeca --- /dev/null +++ b/youtube_dlc/extractor/trovolive.py @@ -0,0 +1,111 @@ +# 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'])), + }
\ No newline at end of file |