import itertools import re from .common import InfoExtractor from ..utils import ( determine_ext, int_or_none, join_nonempty, merge_dicts, str_or_none, traverse_obj, unified_strdate, url_or_none, ) class ExtremeMusicBaseIE(InfoExtractor): _API_URL = 'https://snapi.extrememusic.com' _REQUEST_HEADERS = None def _process_version_arg(self, arg): self._version_requested = arg def _set_request_headers(self, url, video_id, country=None): if not self._REQUEST_HEADERS: # the site serves different versions of the same playlist id due to geo-restriction, # so use user's own country code or geo_bypass_country code if not country: country = self._download_webpage('https://ipapi.co/country_code', video_id) env = self._download_json('https://www.extrememusic.com/env', video_id) self._REQUEST_HEADERS = { 'Accept': 'application/json', 'Origin': 'https://www.extrememusic.com', 'Referer': url, 'Sec-Fetch-Mode': 'cors', 'X-API-Auth': env['token'], 'X-Site-Id': 4, 'X-Viewer-Country': country.upper(), } return self._REQUEST_HEADERS def _get_album_data(self, album_id, video_id, fatal=True): self._process_version_arg(self._configuration_arg('ver') or self._configuration_arg('version')) album = self._download_json(f'{self._API_URL}/albums/{album_id}', video_id, fatal=fatal, note='Downloading album data', headers=self._REQUEST_HEADERS) if video_id == album_id: bio = self._download_json(f'{self._API_URL}/albums/{album_id}/bio', video_id, fatal=False, note='Downloading album data', headers=self._REQUEST_HEADERS) return merge_dicts(album, bio or {}) else: return album def _extract_track(self, album_data, track_id=None, version_id=None): album_data = album_data or {} if 'tracks' in album_data and 'track_sounds' in album_data: if not track_id and version_id: track_id = traverse_obj(album_data['track_sounds'], (lambda _, v: v['id'] == int(version_id), 'track_id', {int}), get_all=False) track = traverse_obj(album_data['tracks'], (lambda _, v: v['id'] == int(track_id), {dict}), get_all=False) info = {**traverse_obj(track, { 'track': ('title', {str}), 'track_number': ('sort_order', {lambda v: v + 1}, {int}), 'track_id': ('track_no', {str}), 'description': ('description', {lambda v: str_or_none(v) or None}), 'artists': ('artists', {lambda v: v or traverse_obj(album_data, ('album', 'artist'))}, {lambda v: (v if isinstance(v, list) else [v]) if v else None}), 'composers': ('composers', ..., 'name'), 'genres': (('genre', 'subgenre'), ..., 'label'), 'tag': ('keywords', ..., 'label'), 'album': ('album_title', {lambda v: str_or_none(v) or None}), }), **traverse_obj(album_data, { 'album_artists': ('album', 'artist', {lambda v: [v] if v else None}), 'upload_date': ('album', 'created', {unified_strdate}), })} entries, thumbnails = [], [] for image in traverse_obj(track, ('images', 'default')): thumbnails.append(traverse_obj(image, { 'url': ('url', {url_or_none}), 'width': ('width', {int_or_none}), 'height': ('height', {int_or_none}), })) for idx, sound_id in enumerate([version_id] if version_id else track['track_sound_ids']): if sound := traverse_obj(album_data['track_sounds'], (lambda _, v: v['id'] == int(sound_id) and v['track_id'] == int(track_id), {dict}), get_all=False): if (version_id or (not version_id and ((not self._version_requested and idx == 0) or 'all' in self._version_requested or sound['version_type'].lower() in self._version_requested))): formats = [] for audio_url in traverse_obj(sound, ('assets', 'audio', ('preview_url', 'preview_url_hls'))): if determine_ext(audio_url) == 'm3u8': m3u8_url = re.sub(r'\.m3u8\?.*', '/HLS/128_v4.m3u8', audio_url) for f in self._extract_m3u8_formats(m3u8_url, sound_id, 'mpeg', fatal=False): formats.append({ **f, 'vcodec': 'none', 'perference': -2, }) else: formats.append({ 'url': audio_url, 'vcodec': 'none', }) entries.append({ 'id': str(sound_id), 'title': join_nonempty('title', 'version_type', from_dict=sound, delim=' - '), 'alt_title': sound['version_type'], **info, 'thumbnails': thumbnails, 'duration': sound.get('duration'), 'formats': formats, 'webpage_url': f"https://www.extrememusic.com/albums/{track['album_id']}?item={track_id}&ver={sound_id}", }) if len(entries) > 1: return { 'id': track_id, **info, 'entries': entries, '_type': 'playlist', } elif len(entries) == 1: return entries[0] return [] class ExtremeMusicIE(ExtremeMusicBaseIE): _VALID_URL = r'https?://(?:www\.)?extrememusic\.com/albums/(?P\d+)\?(.*item=(?P\d+))?(.*ver=(?P\d+))?' _TESTS = [{ 'url': 'https://www.extrememusic.com/albums/15875?item=263381&ver=1265009&sharedTrack=dHJ1ZQ==', 'info_dict': { 'id': '1265009', 'ext': 'mp3', 'title': 'FOLLOW - Instrumental', 'alt_title': 'Instrumental', 'track': 'FOLLOW', 'track_number': 5, 'track_id': 'HPE316_05', 'artists': ['PRAERS'], 'composers': ['Joseph Andrew Banfi', 'Thomas Louis James White'], 'genres': ['POP', 'DREAM', 'INDIE'], 'tag': 'count:7', 'album': 'AVALON', 'album_artists': ['PRAERS'], 'upload_date': '20240729', 'thumbnail': 'https://d2oet5a29f64lj.cloudfront.net/img-data/w/2480/album/600/HPE316.jpg', 'duration': 246, }, }, { 'url': 'https://www.extrememusic.com/albums/15823?ver=1262087', 'info_dict': { 'id': '1262087', 'ext': 'mp3', 'title': 'MAGICAL HIGHWAY - VOCALS', 'alt_title': 'VOCALS', 'track': 'MAGICAL HIGHWAY', 'track_number': 2, 'track_id': 'ASM0002_02', 'description': 'Full version - a fun, happy and upbeat pop track with a medium - fast tempo - electronic, bouncy, bright', 'composers': ['ENB'], 'genres': ['POP', 'ELECTRO', 'JPOP'], 'tag': 'count:8', 'album': 'TOKYO POPPIN\'', 'upload_date': '20240709', 'thumbnail': 'https://d2oet5a29f64lj.cloudfront.net/img-data/w/2480/album/600/ASM0002.jpg', 'duration': 265, }, }, { 'url': 'https://www.extrememusic.com/albums/15064?item=254704', 'info_dict': { 'id': '1178851', 'ext': 'mp3', 'title': 'SWEET TOOTH - Full Version', 'alt_title': 'Full Version', 'track': 'SWEET TOOTH', 'track_number': 2, 'track_id': 'HPE263_02', 'artists': ['PILOT PAISLEY-ROSE'], 'composers': ['PILOT PAISLEY ROSE SARACENO', 'SAMUEL JAMES BRANDT'], 'genres': ['POP', 'ELECTRO', 'ROCK'], 'tag': 'count:7', 'album': 'ADDICTED', 'album_artists': ['PILOT PAISLEY-ROSE'], 'upload_date': '20230629', 'thumbnail': 'https://d2oet5a29f64lj.cloudfront.net/img-data/w/2480/album/600/HPE263.jpg', 'duration': 161, }, }, { 'url': 'https://www.extrememusic.com/albums/1315?item=24795', 'info_dict': { 'id': '61003', 'ext': 'mp3', 'title': 'JOY TO THE WORLD (INST) - Instrumental', 'alt_title': 'Instrumental', 'track': 'JOY TO THE WORLD (INST)', 'track_number': 6, 'track_id': 'XEL016_06', 'composers': ['TRADITIONAL'], 'genres': ['HOLIDAY', 'CHRISTMAS'], 'tag': 'count:5', 'album': 'CHRISTMAS SPARKLE', 'upload_date': '20041001', 'thumbnail': 'https://d2oet5a29f64lj.cloudfront.net/img-data/w/2480/album/600/XEL016.jpg', 'duration': 132, }, }] def _real_extract(self, url): album_id, track_id, version_id = self._match_valid_url(url).group('album', 'id', 'ver') self._set_request_headers(url, track_id or version_id, self.get_param('geo_bypass_country') or 'DE') album_data = self._get_album_data(album_id, track_id or version_id) if result := self._extract_track(album_data, track_id, version_id): return result else: self.raise_no_formats('No formats were found') class ExtremeMusicAIE(ExtremeMusicBaseIE): IE_NAME = 'ExtremeMusic:album' _VALID_URL = r'https?://(?:www\.)?extrememusic\.com/albums/(?P\d+)(?!.*(item|ver)=)' _TESTS = [{ 'url': 'https://www.extrememusic.com/albums/6778', 'info_dict': { 'id': '6778', 'album': 'Ethereal Voices', }, 'playlist_count': 11, }, { 'url': 'https://www.extrememusic.com/albums/15835', 'info_dict': { 'id': '15835', 'album': 'BIGGEST BANG', 'description': 'Minus Aura, a minimalist duo who create deep drama and emotion to put you under their spell.', 'artists': ['MINUS AURA'], 'genres': ['ELECTRONICA', 'POP', 'SYNTH'], 'tag': ['ELECTRONIC', 'STRUGGLE'], }, 'playlist_count': 4, }] def _real_extract(self, url): album_id = self._match_id(url) self._set_request_headers(url, album_id, self.get_param('geo_bypass_country') or 'DE') album_data = self._get_album_data(album_id, album_id) entries = [] for track_id in traverse_obj(album_data, ('tracks', ..., 'id')): if track := self._extract_track(album_data, track_id=track_id): if track.get('entries'): entries.extend(track['entries']) else: entries.append(track) if entries: subgenres = traverse_obj(album_data, ('album', 'subgenres', {str_or_none})) return merge_dicts(traverse_obj(album_data.get('album'), { 'id': ('id', {lambda v: str(v)}), 'album': ('title', {str_or_none}), 'description': ('description', {lambda v: str_or_none(v) or None}), 'artists': ('artist', {lambda v: [v] if v else None}), 'genres': ('genres', {str_or_none}, {lambda v: join_nonempty(v, subgenres, delim=', ')}, {lambda v: v.split(', ') if v else None}), 'tag': ('keywords', {lambda v: v.split(', ') if v else None}), }), { 'description': traverse_obj(album_data, ('bio', 'description', {lambda v: str_or_none(v) or None})), 'entries': entries, '_type': 'playlist', }) else: self.raise_no_formats('No formats were found') class ExtremeMusicPIE(ExtremeMusicBaseIE): IE_NAME = 'ExtremeMusic:playlist' _VALID_URL = r'https?://(?:www\.)?extrememusic\.com/playlists/(?P[^?]+)' _TESTS = [{ 'url': 'https://www.extrememusic.com/playlists/Kf3fAppAKK2UpAUUp7KK1pBDBMrC62c_Kf8UKAAppUUKppK2UAp92K7Appp8xMx', 'info_dict': { 'id': 'Kf3fAppAKK2UpAUUp7KK1pBDBMrC62c_Kf8UKAAppUUKppK2UAp92K7Appp8xMx', 'title': 'NICE', 'thumbnail': 'https://d2oet5a29f64lj.cloudfront.net/img-data/w/2480/featureditem/square/thumbnail_PLAYLIST_Nice-square-(formerly ChristmasTraditional).jpg', }, 'playlist_mincount': 29, 'expected_warnings': ['This playlist has geo-restricted items. Try using --xff to specify a different country code.'], }, { 'url': 'https://www.extrememusic.com/playlists/fUKKU5KAfK61pAAKp4U4KpKUxsRk2ki_fU117KpUUAAUKAUfpA6UAfAKK8Ul5ji', 'info_dict': { 'id': 'fUKKU5KAfK61pAAKp4U4KpKUxsRk2ki_fU117KpUUAAUKAUfpA6UAfAKK8Ul5ji', 'title': 'NEO CLASSICAL', 'thumbnail': 'https://d2oet5a29f64lj.cloudfront.net/img-data/w/2480/featureditem/square/NeoClassical.jpg', }, 'playlist_mincount': 50, }] def _real_extract(self, url): playlist_id = self._match_id(url) headers = self._set_request_headers(url, playlist_id, self.get_param('geo_bypass_country')) def playlist_query(playlist_id, offset, limit): # playlist api: https://snapi.extrememusic.com/playlists?id={playlist_id}&range={offset}%2C{limit}' return self._download_json( 'https://snapi.extrememusic.com/playlists', playlist_id, note=f'Downloading item {offset + 1}-{offset + limit}', query={ 'id': playlist_id, 'range': f'{offset},{limit}', }, headers=headers) thumbnails, entries = [], [] album_data, track_done, limit = {}, [], 50 for i in itertools.count(): playlist = playlist_query(playlist_id, i * limit, limit) if len(playlist['playlist_items']) == 0: break else: track_ids = traverse_obj(playlist, ('playlist_items', ..., 'track_id')) for track_id in list(dict.fromkeys(track_ids)): if track_id not in track_done: album_id = traverse_obj(playlist, ('tracks', lambda _, v: v['id'] == track_id, 'album_id', {int}), get_all=False) if album_id not in album_data: album_data[album_id] = self._get_album_data(album_id, track_id, fatal=False) playlist['album'] = traverse_obj(album_data, (album_id, 'album', {dict})) if track := self._extract_track(playlist, track_id=track_id): if track.get('entries'): entries.extend(track['entries']) else: entries.append(track) track_done.append(track_id) if len(entries) >= playlist['playlist']['playlist_items_count']: break if entries: if len(entries) < playlist['playlist']['playlist_items_count']: self.report_warning('This playlist has geo-restricted items. Try using --xff to specify a different country code.') for image in traverse_obj(playlist['playlist'], ('images', 'square')): thumbnails.append(traverse_obj(image, { 'url': ('url', {url_or_none}), 'width': ('width', {int_or_none}), 'height': ('height', {int_or_none}), })) return {k: v for k, v in { 'id': playlist['playlist']['id'], 'title': playlist['playlist']['title'], 'thumbnail': traverse_obj(thumbnails, (0, 'url', {url_or_none})), 'thumbnails': thumbnails, 'uploader': playlist['playlist']['owner_name'], 'entries': entries, '_type': 'playlist', }.items() if v} else: self.raise_no_formats('No formats were found')