youtube-dl/youtube_dl/extractor/kick.py
2023-04-11 11:22:19 +02:00

54 lines
2.2 KiB
Python

# coding: utf-8
from __future__ import unicode_literals
from .common import InfoExtractor
from ..utils import url_or_none, traverse_obj, strip_or_none, float_or_none, parse_iso8601, int_or_none
class KickIE(InfoExtractor):
_VALID_URL = r'https?://(?:www\.)?kick\.com/video/(?P<id>[0-9a-zA-Z-]+)'
_TEST = {
'url': 'https://kick.com/video/82a3c11d-7a17-4747-aecb-2e61413eb11f',
'md5': 'f052bc1046cd9ca6751925dd12420010',
'info_dict': {
'id': '82a3c11d-7a17-4747-aecb-2e61413eb11f',
'ext': 'mp4',
'title': 'Weekly Stake Stream',
'uploader': 'Eddie',
'thumbnail': r're:^https?://.*\.jpg.*$',
'timestamp': 1676890314,
'upload_date': '20230220',
}
}
def _real_extract(self, url):
id = self._match_id(url)
headers = {
'Accept': 'application/json',
}
data = self._download_json('https://kick.com/api/v1/video/%s' % id, id, headers=headers)
formats = self._extract_m3u8_formats(
data['source'], id, 'mp4')
self._sort_formats(formats)
livestream = data['livestream']
strip_lambda = lambda x: strip_or_none(x) or None
return {
'id': id,
'formats': formats,
'title': livestream.get('session_title'),
'uploader': traverse_obj(livestream, ('channel', 'user', 'username'), expected_type=strip_lambda),
'thumbnail': url_or_none(livestream.get('thumbnail')),
'duration': float_or_none(data['livestream'].get('duration'), scale=1000),
'timestamp': traverse_obj(data, 'updated_at', 'created_at', expected_type=parse_iso8601),
'release_timestamp': parse_iso8601(data.get('created_at')),
'view_count': int_or_none(data.get('views')),
'is_live': data['livestream'].get('is_live'),
'channel': traverse_obj(data['livestream'], ('channel', 'slug'), expected_type=strip_lambda),
'categories': traverse_obj(data, ('categories', Ellipsis, 'name'), expected_type=strip_lambda) or None,
'tags': traverse_obj(data, ('categories', Ellipsis, 'tags', Ellipsis), expected_type=strip_lambda) or None,
}