mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-06 03:05:35 +00:00
336360d509
+ rewrite load all albums + artist logic with itertools.groupby + add a function to convert seconds to string
75 lines
1.8 KiB
Python
75 lines
1.8 KiB
Python
"""
|
|
Contains methods relating to albums.
|
|
"""
|
|
|
|
from dataclasses import asdict
|
|
from typing import Any
|
|
from itertools import groupby
|
|
|
|
|
|
from app.models.track import Track
|
|
from app.store.albums import AlbumStore
|
|
from app.store.tracks import TrackStore
|
|
|
|
|
|
def create_albums():
|
|
"""
|
|
Creates albums from the tracks in the store.
|
|
"""
|
|
|
|
# group all tracks by albumhash
|
|
tracks = TrackStore.tracks
|
|
tracks = sorted(tracks, key=lambda t: t.albumhash)
|
|
grouped = groupby(tracks, lambda t: t.albumhash)
|
|
|
|
# create albums from the groups
|
|
albums: list[Track] = []
|
|
for albumhash, tracks in grouped:
|
|
count = len(list(tracks))
|
|
duration = sum(t.duration for t in tracks)
|
|
created_date = min(t.created_date for t in tracks)
|
|
|
|
album = AlbumStore.create_album(list(tracks)[0])
|
|
album.set_count(count)
|
|
album.set_duration(duration)
|
|
album.set_created_date(created_date)
|
|
|
|
albums.append(album)
|
|
|
|
return albums
|
|
|
|
|
|
def validate_albums():
|
|
"""
|
|
Removes albums that have no tracks.
|
|
|
|
Probably albums that were added from incompletely written files.
|
|
"""
|
|
|
|
album_hashes = {t.albumhash for t in TrackStore.tracks}
|
|
albums = AlbumStore.albums
|
|
|
|
for album in albums:
|
|
if album.albumhash not in album_hashes:
|
|
AlbumStore.remove_album(album)
|
|
|
|
|
|
def remove_duplicate_on_merge_versions(tracks: list[Track]) -> list[Track]:
|
|
"""
|
|
Removes duplicate tracks when merging versions of the same album.
|
|
"""
|
|
|
|
pass
|
|
|
|
|
|
def sort_by_track_no(tracks: list[Track]) -> list[dict[str, Any]]:
|
|
tracks = [asdict(t) for t in tracks]
|
|
|
|
for t in tracks:
|
|
track = str(t["track"]).zfill(3)
|
|
t["_pos"] = int(f"{t['disc']}{track}")
|
|
|
|
tracks = sorted(tracks, key=lambda t: t["_pos"])
|
|
|
|
return tracks
|