yt-dlp/yt_dlp/extractor/trovolive.py

# 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'])),
        }