]> Raphaƫl G. Git Repositories - youtubedl/blob - youtube_dl/extractor/showroomlive.py
Update README.md
[youtubedl] / youtube_dl / extractor / showroomlive.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 from .common import InfoExtractor
5 from ..compat import compat_str
6 from ..utils import (
7 ExtractorError,
8 int_or_none,
9 urljoin,
10 )
11
12
13 class ShowRoomLiveIE(InfoExtractor):
14 _VALID_URL = r'https?://(?:www\.)?showroom-live\.com/(?!onlive|timetable|event|campaign|news|ranking|room)(?P<id>[^/?#&]+)'
15 _TEST = {
16 'url': 'https://www.showroom-live.com/48_Nana_Okada',
17 'only_matching': True,
18 }
19
20 def _real_extract(self, url):
21 broadcaster_id = self._match_id(url)
22
23 webpage = self._download_webpage(url, broadcaster_id)
24
25 room_id = self._search_regex(
26 (r'SrGlobal\.roomId\s*=\s*(\d+)',
27 r'(?:profile|room)\?room_id\=(\d+)'), webpage, 'room_id')
28
29 room = self._download_json(
30 urljoin(url, '/api/room/profile?room_id=%s' % room_id),
31 broadcaster_id)
32
33 is_live = room.get('is_onlive')
34 if is_live is not True:
35 raise ExtractorError('%s is offline' % broadcaster_id, expected=True)
36
37 uploader = room.get('performer_name') or broadcaster_id
38 title = room.get('room_name') or room.get('main_name') or uploader
39
40 streaming_url_list = self._download_json(
41 urljoin(url, '/api/live/streaming_url?room_id=%s' % room_id),
42 broadcaster_id)['streaming_url_list']
43
44 formats = []
45 for stream in streaming_url_list:
46 stream_url = stream.get('url')
47 if not stream_url:
48 continue
49 stream_type = stream.get('type')
50 if stream_type == 'hls':
51 m3u8_formats = self._extract_m3u8_formats(
52 stream_url, broadcaster_id, ext='mp4', m3u8_id='hls',
53 live=True)
54 for f in m3u8_formats:
55 f['quality'] = int_or_none(stream.get('quality', 100))
56 formats.extend(m3u8_formats)
57 elif stream_type == 'rtmp':
58 stream_name = stream.get('stream_name')
59 if not stream_name:
60 continue
61 formats.append({
62 'url': stream_url,
63 'play_path': stream_name,
64 'page_url': url,
65 'player_url': 'https://www.showroom-live.com/assets/swf/v3/ShowRoomLive.swf',
66 'rtmp_live': True,
67 'ext': 'flv',
68 'format_id': 'rtmp',
69 'format_note': stream.get('label'),
70 'quality': int_or_none(stream.get('quality', 100)),
71 })
72 self._sort_formats(formats)
73
74 return {
75 'id': compat_str(room.get('live_id') or broadcaster_id),
76 'title': self._live_title(title),
77 'description': room.get('description'),
78 'timestamp': int_or_none(room.get('current_live_started_at')),
79 'uploader': uploader,
80 'uploader_id': broadcaster_id,
81 'view_count': int_or_none(room.get('view_num')),
82 'formats': formats,
83 'is_live': True,
84 }