You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
3.0 KiB
75 lines
3.0 KiB
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from src.api import get_cover
|
|
from src.models.song_data import Datum
|
|
from src.utils import ttml_convent_to_lrc
|
|
|
|
|
|
class SongMetadata(BaseModel):
|
|
title: Optional[str] = None
|
|
artist: Optional[str] = None
|
|
album_artist: Optional[str] = None
|
|
album: Optional[str] = None
|
|
composer: Optional[str] = None
|
|
genre: Optional[str] = None
|
|
created: Optional[str] = None
|
|
track: Optional[str] = None
|
|
tracknum: Optional[int] = None
|
|
disk: Optional[int] = None
|
|
lyrics: Optional[str] = None
|
|
cover: bytes = None
|
|
cover_url: Optional[str] = None
|
|
copyright: Optional[str] = None
|
|
record_company: Optional[str] = None
|
|
upc: Optional[str] = None
|
|
isrc: Optional[str] = None
|
|
playlistIndex: Optional[int] = None
|
|
|
|
def to_itags_params(self, embed_metadata: list[str]):
|
|
tags = []
|
|
for key, value in self.model_dump().items():
|
|
if not value:
|
|
continue
|
|
if key in embed_metadata and value:
|
|
if "playlist" in key:
|
|
continue
|
|
if key == "cover":
|
|
continue
|
|
if key == "lyrics":
|
|
lrc = ttml_convent_to_lrc(value)
|
|
tags.append(f"{key}={lrc}")
|
|
continue
|
|
if key.lower() in ('upc', 'isrc'):
|
|
tags.append(f"WM/{key.lower()}={value}")
|
|
continue
|
|
if key == 'composer':
|
|
tags.append(f"writer={value}")
|
|
continue
|
|
tags.append(f"{key}={value}")
|
|
return ":".join(tags)
|
|
|
|
@classmethod
|
|
def parse_from_song_data(cls, song_data: Datum):
|
|
return cls(title=song_data.attributes.name, artist=song_data.attributes.artistName,
|
|
album_artist=song_data.relationships.albums.data[0].attributes.artistName,
|
|
album=song_data.attributes.albumName, composer=song_data.attributes.composerName,
|
|
genre=song_data.attributes.genreNames[0], created=song_data.attributes.releaseDate,
|
|
track=song_data.attributes.name, tracknum=song_data.attributes.trackNumber,
|
|
disk=song_data.attributes.discNumber, lyrics="", cover_url=song_data.attributes.artwork.url,
|
|
copyright=song_data.relationships.albums.data[0].attributes.copyright,
|
|
record_company=song_data.relationships.albums.data[0].attributes.recordLabel,
|
|
upc=song_data.relationships.albums.data[0].attributes.upc,
|
|
isrc=song_data.attributes.isrc,
|
|
rtng=1 if song_data.attributes.contentRating and song_data.attributes.contentRating == 'explicit' else 0
|
|
)
|
|
|
|
def set_lyrics(self, lyrics: str):
|
|
self.lyrics = lyrics
|
|
|
|
async def get_cover(self, cover_format: str, cover_size: str):
|
|
self.cover = await get_cover(self.cover_url, cover_format, cover_size)
|
|
|
|
def set_playlist_index(self, index: int):
|
|
self.playlistIndex = index
|
|
|