]> jfr.im git - yt-dlp.git/blame - yt_dlp/extractor/saitosan.py
[cleanup] Fix infodict returned fields (#8906)
[yt-dlp.git] / yt_dlp / extractor / saitosan.py
CommitLineData
9dee4df5
L
1from .common import InfoExtractor
2from ..utils import ExtractorError, try_get
3
4
5class SaitosanIE(InfoExtractor):
6 IE_NAME = 'Saitosan'
7 _VALID_URL = r'https?://(?:www\.)?saitosan\.net/bview.html\?id=(?P<id>[0-9]+)'
8 _TESTS = [{
9 'url': 'http://www.saitosan.net/bview.html?id=10031846',
10 'info_dict': {
11 'id': '10031846',
12 'ext': 'mp4',
13 'title': '井下原 和弥',
14 'uploader': '井下原 和弥',
15 'thumbnail': 'http://111.171.196.85:8088/921f916f-7f55-4c97-b92e-5d9d0fef8f5f/thumb',
16 'is_live': True,
17 },
18 'params': {
19 # m3u8 download
20 'skip_download': True,
21 },
22 'skip': 'Broadcasts are ephemeral',
23 },
24 {
25 'url': 'http://www.saitosan.net/bview.html?id=10031795',
26 'info_dict': {
27 'id': '10031795',
28 'ext': 'mp4',
29 'title': '橋本',
30 'uploader': '橋本',
31 'thumbnail': 'http://111.171.196.85:8088/1a3933e1-a01a-483b-8931-af15f37f8082/thumb',
32 'is_live': True,
33 },
34 'params': {
35 # m3u8 download
36 'skip_download': True,
37 },
38 'skip': 'Broadcasts are ephemeral',
39 }]
40
41 def _real_extract(self, url):
42 b_id = self._match_id(url)
43
44 base = 'http://hankachi.saitosan-api.net:8002/socket.io/?transport=polling&EIO=3'
45 sid = self._download_socket_json(base, b_id, note='Opening socket').get('sid')
46 base += '&sid=' + sid
47
48 self._download_webpage(base, b_id, note='Polling socket')
49 payload = '420["room_start_join",{"room_id":"%s"}]' % b_id
50 payload = '%s:%s' % (len(payload), payload)
51
52 self._download_webpage(base, b_id, data=payload, note='Polling socket with payload')
53 response = self._download_socket_json(base, b_id, note='Polling socket')
54 if not response.get('ok'):
55 err = response.get('error') or {}
56 raise ExtractorError(
57 '%s said: %s - %s' % (self.IE_NAME, err.get('code', '?'), err.get('msg', 'Unknown')) if err
58 else 'The socket reported that the broadcast could not be joined. Maybe it\'s offline or the URL is incorrect',
59 expected=True, video_id=b_id)
60
61 self._download_webpage(base, b_id, data='26:421["room_finish_join",{}]', note='Polling socket')
62 b_data = self._download_socket_json(base, b_id, note='Getting broadcast metadata from socket')
63 m3u8_url = b_data.get('url')
64
65 self._download_webpage(base, b_id, data='1:1', note='Closing socket', fatal=False)
66
67 return {
68 'id': b_id,
69 'title': b_data.get('name'),
70 'formats': self._extract_m3u8_formats(m3u8_url, b_id, 'mp4', live=True),
71 'thumbnail': m3u8_url.replace('av.m3u8', 'thumb'),
72 'uploader': try_get(b_data, lambda x: x['broadcast_user']['name']), # same as title
73 'is_live': True
74 }