mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-07 03:35:35 +00:00
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
import dataclasses
|
|
import json
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
|
|
from app import settings
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class Playlist:
|
|
"""Creates playlist objects"""
|
|
|
|
id: int
|
|
artisthashes: str | list[str]
|
|
banner_pos: int
|
|
has_gif: str | bool
|
|
image: str
|
|
last_updated: str
|
|
name: str
|
|
trackhashes: str | list[str]
|
|
|
|
thumb: str = ""
|
|
count: int = 0
|
|
duration: int = 0
|
|
has_image: bool = False
|
|
images: list[str] = dataclasses.field(default_factory=list)
|
|
|
|
def __post_init__(self):
|
|
self.trackhashes = json.loads(str(self.trackhashes))
|
|
# self.artisthashes = json.loads(str(self.artisthashes))
|
|
# commentted until we need it 👆
|
|
self.artisthashes = []
|
|
|
|
self.count = len(self.trackhashes)
|
|
self.has_gif = bool(int(self.has_gif))
|
|
self.has_image = (
|
|
Path(settings.Paths.get_playlist_img_path()) / str(self.image)
|
|
).exists()
|
|
|
|
if self.image is not None:
|
|
self.thumb = "thumb_" + self.image
|
|
else:
|
|
self.image = "None"
|
|
self.thumb = "None"
|
|
|
|
def set_duration(self, duration: int):
|
|
self.duration = duration
|
|
|
|
def clear_lists(self):
|
|
"""
|
|
Removes data from lists to make it lighter for sending
|
|
over the API.
|
|
"""
|
|
self.trackhashes = []
|
|
self.artisthashes = []
|