mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-08-17 12:50:40 +02:00
209 lines
8.1 KiB
Python
209 lines
8.1 KiB
Python
from .common import InfoExtractor
|
|
from ..utils import (
|
|
try_get,
|
|
unified_timestamp
|
|
)
|
|
|
|
|
|
class SovietsClosetBaseIE(InfoExtractor):
|
|
MEDIADELIVERY_REFERER = {'Referer': 'https://iframe.mediadelivery.net/'}
|
|
|
|
def parse_nuxt_jsonp(self, nuxt_jsonp_url, video_id, name):
|
|
nuxt_jsonp = self._download_webpage(nuxt_jsonp_url, video_id, note=f'Downloading {name} __NUXT_JSONP__')
|
|
return self._search_nuxt_data(nuxt_jsonp, video_id, '__NUXT_JSONP__')
|
|
|
|
def video_meta(self, video_id, game_name, category_name, episode_number, stream_date):
|
|
title = game_name
|
|
if category_name and category_name != 'Misc':
|
|
title += f' - {category_name}'
|
|
if episode_number:
|
|
title += f' #{episode_number}'
|
|
|
|
timestamp = unified_timestamp(stream_date)
|
|
|
|
return {
|
|
'id': video_id,
|
|
'title': title,
|
|
'http_headers': self.MEDIADELIVERY_REFERER,
|
|
'uploader': 'SovietWomble',
|
|
'creator': 'SovietWomble',
|
|
'release_timestamp': timestamp,
|
|
'timestamp': timestamp,
|
|
'uploader_id': 'SovietWomble',
|
|
'uploader_url': 'https://www.twitch.tv/SovietWomble',
|
|
'was_live': True,
|
|
'availability': 'public',
|
|
'series': game_name,
|
|
'season': category_name,
|
|
'episode_number': episode_number,
|
|
}
|
|
|
|
|
|
class SovietsClosetIE(SovietsClosetBaseIE):
|
|
_VALID_URL = r'https?://(?:www\.)?sovietscloset\.com/video/(?P<id>[0-9]+)/?'
|
|
_TESTS = [
|
|
{
|
|
'url': 'https://sovietscloset.com/video/1337',
|
|
'md5': 'bd012b04b261725510ca5383074cdd55',
|
|
'info_dict': {
|
|
'id': '1337',
|
|
'ext': 'mp4',
|
|
'title': 'The Witcher #13',
|
|
'thumbnail': r're:^https?://.*\.b-cdn\.net/2f0cfbf4-3588-43a9-a7d6-7c9ea3755e67/thumbnail\.jpg$',
|
|
'uploader': 'SovietWomble',
|
|
'creator': 'SovietWomble',
|
|
'release_timestamp': 1492091580,
|
|
'release_date': '20170413',
|
|
'timestamp': 1492091580,
|
|
'upload_date': '20170413',
|
|
'uploader_id': 'SovietWomble',
|
|
'uploader_url': 'https://www.twitch.tv/SovietWomble',
|
|
'duration': 7007,
|
|
'was_live': True,
|
|
'availability': 'public',
|
|
'series': 'The Witcher',
|
|
'season': 'Misc',
|
|
'episode_number': 13,
|
|
'episode': 'Episode 13',
|
|
},
|
|
},
|
|
{
|
|
'url': 'https://sovietscloset.com/video/1105',
|
|
'md5': '89fa928f183893cb65a0b7be846d8a90',
|
|
'info_dict': {
|
|
'id': '1105',
|
|
'ext': 'mp4',
|
|
'title': 'Arma 3 - Zeus Games #5',
|
|
'uploader': 'SovietWomble',
|
|
'thumbnail': r're:^https?://.*\.b-cdn\.net/c0e5e76f-3a93-40b4-bf01-12343c2eec5d/thumbnail\.jpg$',
|
|
'uploader': 'SovietWomble',
|
|
'creator': 'SovietWomble',
|
|
'release_timestamp': 1461157200,
|
|
'release_date': '20160420',
|
|
'timestamp': 1461157200,
|
|
'upload_date': '20160420',
|
|
'uploader_id': 'SovietWomble',
|
|
'uploader_url': 'https://www.twitch.tv/SovietWomble',
|
|
'duration': 8804,
|
|
'was_live': True,
|
|
'availability': 'public',
|
|
'series': 'Arma 3',
|
|
'season': 'Zeus Games',
|
|
'episode_number': 5,
|
|
'episode': 'Episode 5',
|
|
},
|
|
},
|
|
]
|
|
|
|
def _extract_bunnycdn_iframe(self, video_id, bunnycdn_id):
|
|
iframe = self._download_webpage(
|
|
f'https://iframe.mediadelivery.net/embed/5105/{bunnycdn_id}',
|
|
video_id, note='Downloading BunnyCDN iframe', headers=self.MEDIADELIVERY_REFERER)
|
|
|
|
m3u8_url = self._search_regex(r'(https?://.*?\.m3u8)', iframe, 'm3u8 url')
|
|
thumbnail_url = self._search_regex(r'(https?://.*?thumbnail\.jpg)', iframe, 'thumbnail url')
|
|
|
|
m3u8_formats = self._extract_m3u8_formats(m3u8_url, video_id, headers=self.MEDIADELIVERY_REFERER)
|
|
|
|
if not m3u8_formats:
|
|
duration = None
|
|
else:
|
|
duration = self._extract_m3u8_vod_duration(
|
|
m3u8_formats[0]['url'], video_id, headers=self.MEDIADELIVERY_REFERER)
|
|
|
|
return {
|
|
'formats': m3u8_formats,
|
|
'thumbnail': thumbnail_url,
|
|
'duration': duration,
|
|
}
|
|
|
|
def _real_extract(self, url):
|
|
video_id = self._match_id(url)
|
|
webpage = self._download_webpage(url, video_id)
|
|
|
|
static_assets_base = self._search_regex(r'(/_nuxt/static/\d+)', webpage, 'staticAssetsBase')
|
|
static_assets_base = f'https://sovietscloset.com{static_assets_base}'
|
|
|
|
stream = self.parse_nuxt_jsonp(f'{static_assets_base}/video/{video_id}/payload.js', video_id, 'video')['stream']
|
|
|
|
return {
|
|
**self.video_meta(
|
|
video_id=video_id, game_name=stream['game']['name'],
|
|
category_name=try_get(stream, lambda x: x['subcategory']['name'], str),
|
|
episode_number=stream.get('number'), stream_date=stream.get('date')),
|
|
**self._extract_bunnycdn_iframe(video_id, stream['bunnyId']),
|
|
}
|
|
|
|
|
|
class SovietsClosetPlaylistIE(SovietsClosetBaseIE):
|
|
_VALID_URL = r'https?://(?:www\.)?sovietscloset\.com/(?!video)(?P<id>[^#?]+)'
|
|
_TESTS = [
|
|
|
|
{
|
|
'url': 'https://sovietscloset.com/The-Witcher',
|
|
'info_dict': {
|
|
'id': 'The-Witcher',
|
|
'title': 'The Witcher',
|
|
},
|
|
'playlist_mincount': 31,
|
|
},
|
|
{
|
|
'url': 'https://sovietscloset.com/Arma-3/Zeus-Games',
|
|
'info_dict': {
|
|
'id': 'Arma-3/Zeus-Games',
|
|
'title': 'Arma 3 - Zeus Games',
|
|
},
|
|
'playlist_mincount': 3,
|
|
},
|
|
{
|
|
'url': 'https://sovietscloset.com/arma-3/zeus-games/',
|
|
'info_dict': {
|
|
'id': 'arma-3/zeus-games',
|
|
'title': 'Arma 3 - Zeus Games',
|
|
},
|
|
'playlist_mincount': 3,
|
|
},
|
|
{
|
|
'url': 'https://sovietscloset.com/Total-War-Warhammer',
|
|
'info_dict': {
|
|
'id': 'Total-War-Warhammer',
|
|
'title': 'Total War: Warhammer - Greenskins',
|
|
},
|
|
'playlist_mincount': 33,
|
|
},
|
|
]
|
|
|
|
def _real_extract(self, url):
|
|
playlist_id = self._match_id(url)
|
|
if playlist_id.endswith('/'):
|
|
playlist_id = playlist_id[:-1]
|
|
|
|
webpage = self._download_webpage(url, playlist_id)
|
|
|
|
static_assets_base = self._search_regex(r'(/_nuxt/static/\d+)', webpage, 'staticAssetsBase')
|
|
static_assets_base = f'https://sovietscloset.com{static_assets_base}'
|
|
|
|
sovietscloset = self.parse_nuxt_jsonp(f'{static_assets_base}/payload.js', playlist_id, 'global')['games']
|
|
|
|
if '/' in playlist_id:
|
|
game_slug, category_slug = playlist_id.lower().split('/')
|
|
else:
|
|
game_slug = playlist_id.lower()
|
|
category_slug = 'misc'
|
|
|
|
game = next(game for game in sovietscloset if game['slug'].lower() == game_slug)
|
|
category = next((cat for cat in game['subcategories'] if cat.get('slug', '').lower() == category_slug),
|
|
game['subcategories'][0])
|
|
category_slug = category.get('slug', '').lower() or category_slug
|
|
playlist_title = game.get('name') or game_slug
|
|
if category_slug != 'misc':
|
|
playlist_title += f' - {category.get("name") or category_slug}'
|
|
entries = [{
|
|
**self.url_result(f'https://sovietscloset.com/video/{stream["id"]}', ie=SovietsClosetIE.ie_key()),
|
|
**self.video_meta(
|
|
video_id=stream['id'], game_name=game['name'], category_name=category.get('name'),
|
|
episode_number=i + 1, stream_date=stream.get('date')),
|
|
} for i, stream in enumerate(category['streams'])]
|
|
|
|
return self.playlist_result(entries, playlist_id, playlist_title)
|