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.
58 lines
2.1 KiB
58 lines
2.1 KiB
5 months ago
|
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: str
|
||
|
|
artist: str
|
||
|
|
album_artist: str
|
||
|
|
album: str
|
||
|
|
composer: str
|
||
|
|
genre: str
|
||
|
|
created: str
|
||
|
|
track: str
|
||
|
|
tracknum: int
|
||
|
|
disk: int
|
||
|
|
lyrics: str
|
||
|
|
cover: bytes = None
|
||
|
|
cover_url: str
|
||
|
|
copyright: str
|
||
|
|
record_company: str
|
||
|
|
upc: str
|
||
|
|
isrc: str
|
||
|
|
|
||
|
|
def to_itags_params(self, embed_metadata: list[str], cover_format: str):
|
||
|
|
tags = []
|
||
|
|
for key, value in self.model_dump().items():
|
||
|
|
if key in embed_metadata and value:
|
||
|
|
if key == "cover":
|
||
|
|
continue
|
||
|
|
if key == "lyrics":
|
||
|
|
lrc = ttml_convent_to_lrc(value)
|
||
|
|
tags.append(f"{key}={lrc}")
|
||
|
|
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
|
||
|
|
)
|
||
|
|
|
||
|
|
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)
|