mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-06 19:25:34 +00:00
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
import dataclasses
|
|
from dataclasses import dataclass
|
|
|
|
from app import settings
|
|
from .artist import ArtistMinimal
|
|
from app.utils.hashing import create_hash
|
|
from app.utils.parsers import split_artists, remove_prod, parse_feat_from_title
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class Track:
|
|
"""
|
|
Track class
|
|
"""
|
|
|
|
album: str
|
|
albumartist: str | list[ArtistMinimal]
|
|
albumhash: str
|
|
artist: str | list[ArtistMinimal]
|
|
bitrate: int
|
|
copyright: str
|
|
date: str
|
|
disc: int
|
|
duration: int
|
|
filepath: str
|
|
folder: str
|
|
genre: str | list[str]
|
|
title: str
|
|
track: int
|
|
trackhash: str
|
|
|
|
filetype: str = ""
|
|
image: str = ""
|
|
artist_hashes: list[str] = dataclasses.field(default_factory=list)
|
|
is_favorite: bool = False
|
|
og_title: str = ""
|
|
|
|
def __post_init__(self):
|
|
self.og_title = self.title
|
|
if self.artist is not None:
|
|
artists = split_artists(self.artist)
|
|
new_title = self.title
|
|
|
|
if settings.EXTRACT_FEAT:
|
|
featured, new_title = parse_feat_from_title(self.title)
|
|
original_lower = "-".join([a.lower() for a in artists])
|
|
artists.extend([a for a in featured if a.lower() not in original_lower])
|
|
|
|
if settings.REMOVE_PROD:
|
|
new_title = remove_prod(new_title)
|
|
|
|
# if track is a single
|
|
if self.og_title == self.album:
|
|
self.album = new_title
|
|
|
|
self.title = new_title
|
|
|
|
self.artist_hashes = [create_hash(a, decode=True) for a in artists]
|
|
self.artist = [ArtistMinimal(a) for a in artists]
|
|
|
|
albumartists = split_artists(self.albumartist)
|
|
self.albumartist = [ArtistMinimal(a) for a in albumartists]
|
|
|
|
self.filetype = self.filepath.rsplit(".", maxsplit=1)[-1]
|
|
self.image = self.albumhash + ".webp"
|
|
|
|
if self.genre is not None:
|
|
self.genre = str(self.genre).replace("/", ",").replace(";", ",")
|
|
self.genre = str(self.genre).lower().split(",")
|
|
self.genre = [g.strip() for g in self.genre]
|