62 lines
1.4 KiB
Python
62 lines
1.4 KiB
Python
from pyfzf.pyfzf import FzfPrompt
|
|
|
|
FZF_PROMPT = None
|
|
|
|
|
|
def fzf( # pylint: disable=too-many-arguments
|
|
choices=None,
|
|
prompt=None,
|
|
fzf_options='',
|
|
delimiter='\n',
|
|
return_type=str,
|
|
force_list=False,
|
|
):
|
|
global FZF_PROMPT # pylint: disable=global-statement
|
|
|
|
if choices is None:
|
|
choices = []
|
|
|
|
if not isinstance(return_type, type):
|
|
raise ValueError(f'return_type must be a valid python type; "{return_type}" is not a type')
|
|
|
|
if FZF_PROMPT is None:
|
|
FZF_PROMPT = FzfPrompt()
|
|
|
|
options = ' '.join({
|
|
'-i',
|
|
'--layout=reverse',
|
|
'--ansi',
|
|
'--height=30%',
|
|
f'--prompt "{prompt} : "' if prompt is not None else '',
|
|
fzf_options,
|
|
})
|
|
|
|
selections = [
|
|
return_type(selection)
|
|
for selection in FZF_PROMPT.prompt(choices, options, delimiter)
|
|
]
|
|
|
|
if not force_list:
|
|
if len(selections) == 0:
|
|
return None
|
|
|
|
if len(selections) == 1:
|
|
return selections[0]
|
|
|
|
return selections
|
|
|
|
|
|
def fzf_tail(*args, **kwargs):
|
|
return _fzf_print(*args, **kwargs)[-1]
|
|
|
|
def fzf_head(*args, **kwargs):
|
|
return _fzf_print(*args, **kwargs)[0]
|
|
|
|
def _fzf_print(*args, fzf_options='', **kwargs):
|
|
return fzf(
|
|
*args,
|
|
**kwargs,
|
|
fzf_options = f'--print-query {fzf_options}',
|
|
force_list = True,
|
|
)
|