add coverSize

This commit is contained in:
zhaarey
2024-05-05 19:32:23 +08:00
parent 3f07e1ab7d
commit 9b89c48a07
5 changed files with 7 additions and 5 deletions

View File

@@ -84,9 +84,9 @@ async def get_meta(album_id: str, token: str, storefront: str, lang: str):
@retry(retry=retry_if_exception_type((httpx.TimeoutException, httpcore.ConnectError, SSLError, FileNotFoundError)),
stop=stop_after_attempt(5),
before_sleep=before_sleep_log(logger, logging.WARN))
async def get_cover(url: str, cover_format: str):
async def get_cover(url: str, cover_format: str, cover_size: str):
formatted_url = regex.sub('bb.jpg', f'bb.{cover_format}', url)
req = await client.get(formatted_url.replace("{w}x{h}", "10000x10000"),
req = await client.get(formatted_url.replace("{w}x{h}", cover_size),
headers={"User-Agent": user_agent_browser})
return req.content

View File

@@ -26,6 +26,7 @@ class Download(BaseModel):
saveLyrics: bool
saveCover: bool
coverFormat: str
coverSize: str
afterDownloaded: str
check: str

View File

@@ -58,5 +58,5 @@ class SongMetadata(BaseModel):
def set_lyrics(self, lyrics: str):
self.lyrics = lyrics
async def get_cover(self, cover_format: str):
self.cover = await get_cover(self.cover_url, cover_format)
async def get_cover(self, cover_format: str, cover_size: str):
self.cover = await get_cover(self.cover_url, cover_format, cover_size)

View File

@@ -26,7 +26,7 @@ async def rip_song(song: Song, auth_params: GlobalAuthParams, codec: str, config
if not force_save and check_song_exists(song_metadata, config.download, codec):
logger.info(f"Song: {song_metadata.artist} - {song_metadata.title} already exists")
return
await song_metadata.get_cover(config.download.coverFormat)
await song_metadata.get_cover(config.download.coverFormat, config.download.coverSize)
if song_data.attributes.hasTimeSyncedLyrics:
lyrics = await get_song_lyrics(song.id, song.storefront, auth_params.accountAccessToken,
auth_params.dsid, auth_params.accountToken, config.region.language)