mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-10 04:57:45 +00:00
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
"""
|
|
This library contains all the functions related to tracks.
|
|
"""
|
|
|
|
import os
|
|
from typing import List
|
|
from app import models, instances
|
|
from app.lib import albumslib
|
|
from app.helpers import remove_duplicates
|
|
from app import api
|
|
from progress.bar import Bar
|
|
|
|
|
|
def create_all_tracks() -> List[models.Track]:
|
|
"""
|
|
Gets all songs under the ~/ directory.
|
|
"""
|
|
tracks: list[models.Track] = []
|
|
|
|
_bar = Bar("Creating tracks", max=len(api.PRE_TRACKS))
|
|
|
|
for track in api.PRE_TRACKS:
|
|
try:
|
|
os.chmod(track["filepath"], 0o755)
|
|
except FileNotFoundError:
|
|
instances.songs_instance.remove_song_by_filepath(track["filepath"])
|
|
|
|
album = albumslib.find_album(track["album"], track["albumartist"])
|
|
|
|
track["image"] = album.image
|
|
|
|
tracks.append(models.Track(track))
|
|
_bar.next()
|
|
|
|
_bar.finish()
|
|
|
|
api.TRACKS.clear()
|
|
api.TRACKS.extend(tracks)
|
|
|
|
|
|
def get_album_tracks(albumname, artist):
|
|
"""Returns api tracks matching an album"""
|
|
_tracks: List[models.Track] = []
|
|
|
|
for track in api.TRACKS:
|
|
if track.album == albumname and track.albumartist == artist:
|
|
_tracks.append(track)
|
|
|
|
return remove_duplicates(_tracks)
|
|
|
|
|
|
def get_track_by_id(trackid: str) -> models.Track:
|
|
"""Returns api track matching an id"""
|
|
for track in api.TRACKS:
|
|
if track.trackid == trackid:
|
|
return track
|