57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from src.fw_api import get_artists, get_tracks, get_albums, concatinate_endpoint
|
|
import src.fw_artists
|
|
from src.mpv_control import player, player_menu
|
|
from pyfzf.pyfzf import FzfPrompt
|
|
from loguru import logger
|
|
|
|
fzf = FzfPrompt()
|
|
|
|
@logger.catch
|
|
def list_albums(albums=None, pg=None, search=None, artist=None, include_channels=None):
|
|
albums_next = None
|
|
albums_prev = None
|
|
play_artist_albums = False
|
|
if not albums:
|
|
albums = get_albums(q=search, artist=artist, include_channels=include_channels, pg=pg)
|
|
albums_next = albums.get('next')
|
|
albums_prev = albums.get('previous')
|
|
albums_results = albums.get('results')
|
|
else:
|
|
play_artist_albums = True
|
|
albums_results = albums
|
|
view = ['Search']
|
|
if play_artist_albums:
|
|
view.append('Play all')
|
|
if albums_next:
|
|
view.append('Next page')
|
|
if albums_prev:
|
|
view.append('Prev page')
|
|
|
|
for i in albums_results:
|
|
index = albums_results.index(i)
|
|
album_name = i.get('title')
|
|
view.append(f'{index}.{album_name}')
|
|
select = fzf.prompt(view)[0].split('.', 1)[0]
|
|
if select == 'Next page':
|
|
list_albums(pg=albums_next)
|
|
elif select == 'Prev page':
|
|
list_albums(pg=albums_prev)
|
|
elif select == 'Search':
|
|
print('Search by albums: ')
|
|
list_albums(search=input())
|
|
elif select == 'Play all':
|
|
src.fw_artists.play_artist(albums_results[0].get('artist'))
|
|
else:
|
|
play_album(album_id=albums_results[int(select)].get('id'))
|
|
|
|
|
|
def play_album(album_id):
|
|
tracks = get_tracks(album=album_id, include_channels=True)
|
|
tracks_results = tracks.get('results')
|
|
storage = {}
|
|
for i in tracks_results:
|
|
listen_url = concatinate_endpoint(i.get('listen_url'))
|
|
storage[listen_url] = i
|
|
player.loadfile(listen_url, 'append-play')
|
|
player_menu("Album playing...", storage)
|