summaryrefslogtreecommitdiff
path: root/youtube_dl/extractor/showroomlive.py
blob: efd9d561ffcf24bc922fd65f0eef0f4aff5ba091 (plain)
    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         }

Generated by cgit