mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2026-09-07 11:41:36 +03:00
112 lines
4.0 KiB
Python
112 lines
4.0 KiB
Python
import functools
|
|
|
|
from .common import InfoExtractor
|
|
from ..utils import InAdvancePagedList, int_or_none, qualities, str_or_none, traverse_obj
|
|
|
|
|
|
class ReverbNationIE(InfoExtractor):
|
|
IE_NAME = 'reverbnation:song'
|
|
_VALID_URL = r'^https?://(?:www\.)?reverbnation\.com/.*?/song/(?P<id>\d+).*?$'
|
|
_TESTS = [{
|
|
'url': 'http://www.reverbnation.com/alkilados/song/16965047-mona-lisa',
|
|
'md5': 'c0aaf339bcee189495fdf5a8c8ba8645',
|
|
'info_dict': {
|
|
'id': '16965047',
|
|
'ext': 'mp3',
|
|
'tbr': 192,
|
|
'duration': 217,
|
|
'title': 'MONA LISA',
|
|
'uploader': 'ALKILADOS',
|
|
'uploader_id': '216429',
|
|
'thumbnail': r're:^https?://.*\.jpg',
|
|
},
|
|
}]
|
|
|
|
def _real_extract(self, url):
|
|
song_id = self._match_id(url)
|
|
|
|
api_res = self._download_json(
|
|
f'https://api.reverbnation.com/song/{song_id}',
|
|
song_id,
|
|
note=f'Downloading information of song {song_id}',
|
|
)
|
|
|
|
THUMBNAILS = ('thumbnail', 'image')
|
|
quality = qualities(THUMBNAILS)
|
|
thumbnails = []
|
|
for thumb_key in THUMBNAILS:
|
|
if api_res.get(thumb_key):
|
|
thumbnails.append({
|
|
'url': api_res[thumb_key],
|
|
'preference': quality(thumb_key),
|
|
})
|
|
|
|
return {
|
|
'id': song_id,
|
|
'title': api_res['name'],
|
|
'url': api_res['url'],
|
|
'uploader': api_res.get('artist', {}).get('name'),
|
|
'uploader_id': str_or_none(api_res.get('artist', {}).get('id')),
|
|
'thumbnails': thumbnails,
|
|
'duration': api_res.get('duration'),
|
|
'tbr': api_res.get('bitrate'),
|
|
'ext': 'mp3',
|
|
'vcodec': 'none',
|
|
}
|
|
|
|
|
|
class ReverbNationArtistIE(InfoExtractor):
|
|
IE_NAME = 'reverbnation:artist'
|
|
_VALID_URL = r'^https?://(?:www\.)?reverbnation\.com/(?P<id>[\w-]+)(?:/songs)?$'
|
|
_TESTS = [{
|
|
'url': 'https://www.reverbnation.com/morganandersson',
|
|
'info_dict': {
|
|
'id': '1078497',
|
|
'title': 'morganandersson',
|
|
},
|
|
'playlist_mincount': 8,
|
|
}, {
|
|
'url': 'https://www.reverbnation.com/monogem/songs',
|
|
'info_dict': {
|
|
'id': '3716672',
|
|
'title': 'monogem',
|
|
},
|
|
'playlist_mincount': 10,
|
|
}]
|
|
_PAGE_SIZE = 25
|
|
|
|
def _yield_songs(self, json_data):
|
|
for song in json_data.get('results'):
|
|
data = {
|
|
'id': str_or_none(song.get('id')),
|
|
'title': song.get('name'),
|
|
'url': song.get('url'),
|
|
'uploader': song.get('artist', {}).get('name'),
|
|
'uploader_id': str_or_none(song.get('artist', {}).get('id')),
|
|
'thumbnail': song.get('thumbnail'),
|
|
'duration': int_or_none(song.get('duration')),
|
|
'tbr': int_or_none(song.get('bitrate')),
|
|
'ext': 'mp3',
|
|
'vcodec': 'none',
|
|
}
|
|
yield data
|
|
|
|
def _fetch_page(self, artist_id, page):
|
|
return self._download_json(f'https://www.reverbnation.com/api/artist/{artist_id}/songs?page={page}&per_page={self._PAGE_SIZE}', f'{artist_id}_{page}')
|
|
|
|
def _entries(self, token, first_page_data, page):
|
|
page_data = first_page_data if not page else self._fetch_page(token, page + 1)
|
|
yield from self._yield_songs(page_data)
|
|
|
|
def _real_extract(self, url):
|
|
display_id = self._match_id(url)
|
|
webpage = self._download_webpage(url, display_id)
|
|
artist_url = self._html_search_meta('twitter:player', webpage, 'player url')
|
|
artist_id = self._search_regex(r'artist_(?P<artist>\d+)', artist_url, 'artist id')
|
|
playlist_data = self._fetch_page(artist_id, 1)
|
|
total_pages = traverse_obj(playlist_data, ('pagination', 'page_count', {int}))
|
|
|
|
return self.playlist_result(InAdvancePagedList(
|
|
functools.partial(self._entries, artist_id, playlist_data),
|
|
total_pages, self._PAGE_SIZE), artist_id, display_id)
|