funkwlmpv/src/fw_artists.py

56 lines
1.8 KiB
Python
Raw Normal View History

2022-11-08 19:58:27 +02:00
from src.fw_api import get_artists, get_tracks, concatinate_endpoint
from src.fw_albums import list_albums
from src.mpv_control import player, player_menu
from pyfzf.pyfzf import FzfPrompt
from loguru import logger
fzf = FzfPrompt()
@logger.catch
2022-11-08 23:24:59 +02:00
def list_artists(pg=None, search=None):
artists = get_artists(q=search,pg=pg)
2022-11-08 19:58:27 +02:00
artists_next = artists.get('next')
artists_prev = artists.get('previous')
artists_results = artists.get('results')
2022-11-08 23:24:59 +02:00
view = ['Search']
2022-11-08 19:58:27 +02:00
if artists_next:
view.append('Next page')
if artists_prev:
view.append('Prev page')
for i in artists_results:
index = artists_results.index(i)
artist_name = i.get('name')
view.append(f'{index}.{artist_name}')
select = fzf.prompt(view)[0].split('.', 1)[0]
if select == 'Next page':
list_artists(pg=artists_next)
elif select == 'Prev page':
list_artists(pg=artists_prev)
2022-11-08 23:24:59 +02:00
elif select == 'Search':
print('Search by artist:')
list_artists(search=input())
2022-11-08 19:58:27 +02:00
else:
list_albums(albums=artists_results[int(select)].get('albums'))
def play_artist(artist_id):
2022-11-09 13:47:38 +02:00
tracks = get_tracks(artist=artist_id, include_channels=True, pg=None)
tracks_next = tracks.get('next')
tracks_count = tracks.get('count')
2022-11-08 19:58:27 +02:00
storage = {}
2022-11-09 13:47:38 +02:00
if tracks_count > 50:
print(f'Loading {tracks_count} tracks...')
while True:
tracks_results = tracks.get('results')
tracks_next = tracks.get('next')
for i in tracks_results:
listen_url = concatinate_endpoint(i.get('listen_url'))
storage[listen_url] = i
player.loadfile(listen_url, 'append-play')
if tracks_next:
tracks = get_tracks(artist=artist_id, include_channels=True, pg=tracks_next)
else:
break
2022-11-08 19:58:27 +02:00
player_menu("Artist playing...", storage)