funkwlmpv/src/mpv_control.py

73 lines
2.5 KiB
Python
Raw Normal View History

2022-11-08 19:23:19 +00:00
import src.fw_api
2022-11-08 18:02:07 +00:00
from src.utils import download_track
from src.settings import get_config
2022-11-01 10:15:28 +00:00
from loguru import logger
2022-11-08 18:02:07 +00:00
from pyfzf.pyfzf import FzfPrompt
import mpv
2022-11-08 18:02:07 +00:00
fzf = FzfPrompt()
player = mpv.MPV()
player.ytdl = False # Prevent attempts load track with yt-dlp
player.prefetch_playlist = get_config('prefetch_playlist') # Fast loading next track, but high network traffic
2022-11-13 02:16:46 +00:00
show_like_button = get_config('show_like_button')
2022-11-08 18:02:07 +00:00
class player_fw_storage:
storage = {}
2022-11-08 18:02:07 +00:00
def set_http_header(headers=[]):
player.http_header_fields = headers
2022-11-09 09:49:33 +00:00
@logger.catch
def player_menu(header='', storage={}):
player_fw_storage.storage = storage
2022-11-08 18:02:07 +00:00
while True:
try:
2022-11-13 02:16:46 +00:00
player_items_menu = ['Next', 'Prev', 'Pause',
'Download', 'Info']
if show_like_button:
player_items_menu.append('Like')
player_items_menu.extend(['Hide artist', 'Exit'])
select = fzf.prompt(player_items_menu, f"--header=\'{header}\'")[0]
2022-11-08 18:02:07 +00:00
if select == 'Next':
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 == 'Info':
track = storage.get(player.filename)
for i in ('title', 'fid', 'license', 'album', 'artist'):
if i in ('album', 'artist'):
name_aa = track.get(i).get('name')
if not name_aa:
name_aa = track.get(i).get('title')
print(i + ': '+ name_aa)
key = track.get(i)
if key and isinstance(key, str):
print(i + ': ' + key)
input()
2022-11-13 02:16:46 +00:00
elif select == 'Like':
src.fw_api.favorite_track(storage.get(player.filename).get('id'))
2022-11-08 19:23:19 +00:00
elif select == 'Hide artist':
track = storage.get(player.filename)
src.fw_api.hide_content({'target': {'id': track.get('artist').get('id'), 'type': 'artist'}})
2022-11-08 18:02:07 +00:00
elif select == 'Exit':
player.playlist_clear()
player.stop()
break
except KeyboardInterrupt:
break