funkwlmpv/src/fw_tags.py

37 lines
1.2 KiB
Python

from src.fw_api import get_tags, get_tracks
from pyfzf.pyfzf import FzfPrompt
from loguru import logger
fzf = FzfPrompt()
@logger.catch
def list_tags(pg=None, search=None, error_given=''):
tags = get_tags(q=search, pg=pg)
tags_next = tags.get('next')
tags_prev = tags.get('previous')
tags_results = tags.get('results')
view = ['Search']
if tags_next:
view.append('Next page')
if tags_prev:
view.append('Prev page')
for i in tags_results:
index = tags_results.index(i)
tag_name = i.get('name')
view.append(f'{index}.{tag_name}')
select = fzf.prompt(view, f"--header=\'{error_given}\'")[0].split('.', 1)[0]
if select == 'Next page':
return list_tags(pg=tags_next)
elif select == 'Prev page':
return list_tags(pg=tags_prev)
elif select == 'Search':
print('Search by tag:')
return list_tags(search=input())
else:
selected_tag = tags_results[int(select)].get('name')
if get_tracks(tag=selected_tag, page_size=1)['count'] == 0:
return list_tags(pg=pg, search=search, error_given='This tag no contain tracks, select another')
return selected_tag