mirror of
https://github.com/WorldObservationLog/AppleMusicDecrypt.git
synced 2026-01-15 14:22:54 -03:00
99 lines
3.9 KiB
Python
99 lines
3.9 KiB
Python
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
|
|
|
|
NOT_INCLUDED_FIELD = ["cover", "playlistIndex", "bit_depth", "sample_rate",
|
|
"sample_rate_kHz", "song_id", "album_id", "album_created"]
|
|
|
|
|
|
class SongMetadata(BaseModel):
|
|
song_id: Optional[str] = None
|
|
title: Optional[str] = None
|
|
artist: Optional[str] = None
|
|
album_id: Optional[str] = None
|
|
album_artist: Optional[str] = None
|
|
album: Optional[str] = None
|
|
album_created: 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
|
|
bit_depth: Optional[int] = None
|
|
sample_rate: Optional[int] = None
|
|
sample_rate_kHz: Optional[str] = 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 key in NOT_INCLUDED_FIELD:
|
|
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,
|
|
album_created=song_data.relationships.albums.data[0].attributes.releaseDate,
|
|
rtng=cls._rating(song_data.attributes.contentRating),
|
|
song_id=song_data.id, album_id=song_data.relationships.albums.data[0].id
|
|
)
|
|
|
|
@staticmethod
|
|
def _rating(content_rating: Optional[str]) -> int:
|
|
if not content_rating:
|
|
return 0
|
|
if content_rating == "explicit":
|
|
return 1
|
|
if content_rating == "clean":
|
|
return 2
|
|
|
|
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
|
|
|
|
def set_bit_depth_and_sample_rate(self, bit_depth: int, sample_rate: int):
|
|
self.bit_depth = bit_depth
|
|
self.sample_rate = sample_rate
|
|
self.sample_rate_kHz = str(sample_rate / 1000)
|