funkwlmpv/src/fw_tags.py

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