funkwlmpv/src/fw_radios.py

114 lines
3.6 KiB
Python

from src.fw_api import s, get_radios, post_radio_session, get_track_radio, concatinate_endpoint
from src.utils import download_track
from src.mpv_control import player
from pyfzf.pyfzf import FzfPrompt
from loguru import logger
import threading
import time
fzf = FzfPrompt()
def list_radios():
radios = get_radios()
count = radios.get('count')
results = radios.get('results')
view = []
for i in results:
index = results.index(i)
id_radio = i.get('id')
name = i.get('name')
view.append(f'{index}.{name}')
if s.headers.get('Authorization'): # Radios avalaible only for auth user
view.append('Favourites')
view.append('Less listened')
view.append('Random')
view.append('Recently Added')
selected = fzf.prompt(view, f'--header \'Found {count} radios\'')[0].split('.', 1)
if 'Favourites' in selected:
radio_load(id_radio, 'favorites')
elif 'Random' in selected:
radio_load(id_radio, 'random')
elif 'Recently Added' in selected:
radio_load(id_radio, 'recently-added')
elif 'Less listened' in selected:
radio_load(id_radio, 'less-listened')
else:
id_selected = selected[0]
id_radio = results[int(id_selected)].get('id')
radio_load(id_radio)
def radio_generator(radio_session_id):
count_t = 0
while radio_session_id != '':
time.sleep(1)
if not radio_event_gen.wait(0):
break
count_t += 1
if count_t >= 60:
count_t = 0
radio_get_track(radio_session_id)
logger.info('Radio generator stopped')
radio_event_gen = threading.Event()
def radio_load(id_radio=None, type_radio='custom'):
requested_radio = {
'custom_radio': id_radio,
'radio_type': type_radio,
'related_object_id': None
}
radio_session_id = post_radio_session(requested_radio).get('id')
for i in range(0, 5):
radio_get_track(radio_session_id)
player.playlist_pos = 0
radio_event_gen.set()
radio_task = threading.Thread(target=radio_generator, args=(radio_session_id,), daemon=True)
radio_task.start()
while True:
try:
select = fzf.prompt(('Next', 'Prev', 'Pause', 'Download', 'Exit'), f'--header=\'Radio playing...\'')[0]
if select == 'Next':
radio_get_track(radio_session_id)
player.playlist_next()
elif select == 'Prev':
player.playlist_prev()
elif select == 'Pause':
if player.pause:
player.pause = False
else:
player.pause = True
elif select == 'Download':
print('Downloading...')
name_downloaded = download_track(player.filename)
print(f'Downloaded: {name_downloaded}')
elif select == 'Exit':
try:
radio_event_gen.clear()
except:
logger.exception('Error stopping Thread radio generator')
pass
player.playlist_clear()
player.stop()
break
except KeyboardInterrupt:
break
def radio_get_track(radio_session_id):
radio_context = get_track_radio({'session': radio_session_id})
if not radio_context:
return
if radio_context == "Radio doesn't have more candidates":
return
else:
track = radio_context.get('track')
listen_url = track.get('listen_url')
player.playlist_append(concatinate_endpoint(listen_url))