[soopglobal] Add extractor

This commit is contained in:
hui1601 2024-06-08 23:48:01 +09:00
parent 8b46ad4d8b
commit d73bce4928
No known key found for this signature in database
GPG key ID: 4A9906BFF1F5CB4A
2 changed files with 129 additions and 0 deletions

View file

@ -1874,6 +1874,10 @@
SonyLIVIE,
SonyLIVSeriesIE,
)
from .soopglobal import (
SoopGlobalLiveIE,
SoopGlobalVodIE,
)
from .soundcloud import (
SoundcloudEmbedIE,
SoundcloudIE,

View file

@ -0,0 +1,125 @@
import uuid
from yt_dlp import int_or_none, traverse_obj
from yt_dlp.compat import functools
from yt_dlp.extractor.common import InfoExtractor
from yt_dlp.utils import UserNotLive, bool_or_none, parse_iso8601
class SoopGlobalLiveIE(InfoExtractor):
IE_NAME = 'soopglobal:live'
_VALID_URL = r'https?://www\.sooplive\.com/(?P<id>[\w]+$)'
_TESTS = [{
'url': 'https://www.sooplive.com/soopbowl',
'info_dict': {
'id': 'soopbowl',
'ext': 'mp4',
'title': str,
'thumbnail': r're:^https?://.*\.jpg$',
'channel': 'SoopBowl',
'channel_id': 'soopbowl',
'concurrent_view_count': int,
'channel_follower_count': int,
'timestamp': 1717852526,
'upload_date': '20240608',
'live_status': 'is_live',
'view_count': int,
'age_limit': False,
},
}]
def _real_extract(self, url):
channel_id = self._match_id(url)
client_id = str(uuid.uuid4())
live_detail = self._download_json(
f'https://api.sooplive.com/stream/info/{channel_id}', channel_id,
headers={'client-id': client_id},
note='Downloading live info', errnote='Unable to download live info')
if not live_detail.get('isStream'):
raise UserNotLive(video_id=channel_id)
age_limit = 0
if traverse_obj(live_detail, ('data', 'isAdult', {bool_or_none})):
age_limit = 19
live_statistic = self._download_json(
f'https://api.sooplive.com/stream/info/{channel_id}/live', channel_id,
headers={'client-id': client_id},
note='Downloading live statistics', errnote='Unable to download live statistics')
channel_info = self._download_json(
f'https://api.sooplive.com/channel/info/{channel_id}', channel_id,
headers={'client-id': client_id},
note='Downloading channel information', errnote='Unable to download channel information')
formats, subtitles = self._extract_m3u8_formats_and_subtitles(
f'https://api.sooplive.com/media/live/{channel_id}/master.m3u8', channel_id,
headers={'client-id': client_id},
note='Downloading live stream', errnote='Unable to download live stream')
return {
'id': channel_id,
'channel_id': channel_id,
'is_live': True,
'formats': formats,
'subtitles': subtitles,
'view_count': live_statistic.get('viewer'),
'age_limit': age_limit,
**traverse_obj(channel_info.get('streamerChannelInfo'), {
'channel': ('nickname', {str}),
'channel_id': ('channelId', {str}),
'channel_follower_count': ('totalFollowerCount', {int_or_none}),
}),
**traverse_obj(live_detail.get('data'), {
'title': ('title', {str}),
'timestamp': ('streamStartDate', {functools.partial(parse_iso8601)}),
'concurrent_view_count': ('totalStreamCumulativeViewer', {int_or_none}),
'thumbnail': ('thumbnailUrl', {str}),
}),
}
class SoopGlobalVodIE(InfoExtractor):
IE_NAME = 'soopglobal:vod'
_VALID_URL = r'https?://www\.sooplive\.com/video/(?P<id>[\d]+)'
_TESTS = [{
'url': 'https://www.sooplive.com/video/607',
'info_dict': {
'id': '607',
'ext': 'mp4',
'title': str,
'thumbnail': r're:^https?://.*\.jpg$',
'channel': '샤미요',
'channel_id': 'shamiyo',
'timestamp': 1717051284,
'upload_date': '20240530',
'view_count': int,
'age_limit': False,
},
'params': {'skip_download': 'm3u8'},
}]
def _real_extract(self, url):
video_id = self._match_id(url)
client_id = str(uuid.uuid4())
video_info = self._download_json(
f'https://api.sooplive.com/vod/info/{video_id}', video_id,
headers={'client-id': client_id},
note='Downloading video info', errnote='Unable to download video info')
channel_id = video_info.get('channelId')
formats, subtitles = self._extract_m3u8_formats_and_subtitles(
f'https://api.sooplive.com/media/vod/{channel_id}/{video_id}/master.m3u8', video_id,
headers={'client-id': client_id},
note='Downloading video stream', errnote='Unable to download video stream')
return {
'id': video_id,
'channel': video_info.get('nickName'),
'channel_id': channel_id,
'title': video_info.get('titleName'),
'thumbnail': video_info.get('thumb'),
'timestamp': parse_iso8601(video_info.get('createDate')),
'view_count': video_info.get('readCnt'),
'age_limit': 0 if not video_info.get('isAdult') else 19,
'formats': formats,
'subtitles': subtitles,
}