mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-19 17:30:41 +00:00
168 lines
3.8 KiB
Python
168 lines
3.8 KiB
Python
"""
|
|
This library contains all the functions related to albums.
|
|
"""
|
|
import random
|
|
import urllib
|
|
from pprint import pprint
|
|
from typing import List
|
|
|
|
from app import api
|
|
from app import functions
|
|
from app import instances
|
|
from app import models
|
|
from app.lib import trackslib
|
|
from progress.bar import Bar
|
|
|
|
|
|
def get_all_albums() -> List[models.Album]:
|
|
"""
|
|
Returns a list of album objects for all albums in the database.
|
|
"""
|
|
print("Getting all albums...")
|
|
|
|
albums: List[models.Album] = []
|
|
db_albums = instances.album_instance.get_all_albums()
|
|
|
|
_bar = Bar("Creating albums", max=len(db_albums))
|
|
for album in db_albums:
|
|
aa = models.Album(album)
|
|
albums.append(aa)
|
|
_bar.next()
|
|
|
|
_bar.finish()
|
|
|
|
return albums
|
|
|
|
|
|
def create_everything() -> List[models.Track]:
|
|
"""
|
|
Creates album objects for all albums and returns
|
|
a list of track objects
|
|
"""
|
|
albums: list[models.Album] = get_all_albums()
|
|
|
|
api.ALBUMS = albums
|
|
api.ALBUMS.sort(key=lambda x: x.title)
|
|
|
|
tracks = trackslib.create_all_tracks()
|
|
|
|
api.TRACKS.clear()
|
|
api.TRACKS.extend(tracks)
|
|
api.TRACKS.sort(key=lambda x: x.title)
|
|
|
|
|
|
def find_album(albumtitle: str, artist: str) -> models.Album:
|
|
"""
|
|
Finds an album by album title and artist.
|
|
"""
|
|
left = 0
|
|
right = len(api.ALBUMS) - 1
|
|
iter = 0
|
|
|
|
while left <= right:
|
|
iter += 1
|
|
mid = (left + right) // 2
|
|
|
|
if api.ALBUMS[mid].title == albumtitle and api.ALBUMS[
|
|
mid].artist == artist:
|
|
return mid
|
|
|
|
if api.ALBUMS[mid].title < albumtitle:
|
|
left = mid + 1
|
|
else:
|
|
right = mid - 1
|
|
|
|
return None
|
|
|
|
|
|
def get_album_duration(album: list) -> int:
|
|
"""
|
|
Gets the duration of an album.
|
|
"""
|
|
|
|
album_duration = 0
|
|
|
|
for track in album:
|
|
album_duration += track["length"]
|
|
|
|
return album_duration
|
|
|
|
|
|
def use_defaults() -> str:
|
|
"""
|
|
Returns a path to a random image in the defaults directory.
|
|
"""
|
|
path = "defaults/" + str(random.randint(0, 20)) + ".webp"
|
|
return path
|
|
|
|
|
|
def get_album_image(album: list) -> str:
|
|
"""
|
|
Gets the image of an album.
|
|
"""
|
|
|
|
for track in album:
|
|
img_p = (track["album"] + track["albumartist"] + ".webp").replace(
|
|
"/", "::")
|
|
img = functions.extract_thumb(track["filepath"], webp_path=img_p)
|
|
|
|
if img is not None:
|
|
return img
|
|
|
|
return use_defaults()
|
|
|
|
|
|
def get_album_tracks(album: str, artist: str) -> List:
|
|
tracks = []
|
|
|
|
for track in api.DB_TRACKS:
|
|
try:
|
|
if track["album"] == album and track["albumartist"] == artist:
|
|
tracks.append(track)
|
|
except TypeError:
|
|
pprint(track, indent=4)
|
|
print(album, artist)
|
|
|
|
return tracks
|
|
|
|
|
|
def create_album(track) -> models.Album:
|
|
"""
|
|
Generates and returns an album object from a track object.
|
|
"""
|
|
album = {
|
|
"album": track["album"],
|
|
"artist": track["albumartist"],
|
|
}
|
|
|
|
album_tracks = get_album_tracks(album["album"], album["artist"])
|
|
|
|
album["count"] = len(album_tracks)
|
|
album["duration"] = get_album_duration(album_tracks)
|
|
album["date"] = album_tracks[0]["date"]
|
|
|
|
album["artistimage"] = urllib.parse.quote_plus(
|
|
album_tracks[0]["albumartist"] + ".webp")
|
|
|
|
album["image"] = get_album_image(album_tracks)
|
|
|
|
return models.Album(album)
|
|
|
|
|
|
def search_albums_by_name(query: str) -> List[models.Album]:
|
|
"""
|
|
Searches albums by album name.
|
|
"""
|
|
title_albums: List[models.Album] = []
|
|
artist_albums: List[models.Album] = []
|
|
|
|
for album in api.ALBUMS:
|
|
if query.lower() in album.title.lower():
|
|
title_albums.append(album)
|
|
|
|
for album in api.ALBUMS:
|
|
if query.lower() in album.artist.lower():
|
|
artist_albums.append(album)
|
|
|
|
return [*title_albums, *artist_albums]
|