university-setup/scripts/rofi.py

38 lines
1.3 KiB
Python
Raw Permalink Normal View History

2019-09-15 20:42:11 +02:00
import subprocess
2021-09-28 19:32:19 +02:00
2019-09-15 20:42:11 +02:00
def rofi(prompt, options, rofi_args=[], fuzzy=True):
optionstr = '\n'.join(option.replace('\n', ' ') for option in options)
args = ['rofi', '-sort', '-no-levenshtein-sort']
if fuzzy:
args += ['-matching', 'fuzzy']
args += ['-dmenu', '-p', prompt, '-format', 's', '-i']
args += rofi_args
args = [str(arg) for arg in args]
result = subprocess.run(args, input=optionstr, stdout=subprocess.PIPE, universal_newlines=True)
returncode = result.returncode
stdout = result.stdout.strip()
selected = stdout.strip()
try:
index = [opt.strip() for opt in options].index(selected)
except ValueError:
index = -1
2021-10-01 19:08:36 +02:00
# We handle the return code from rofi here:
# 0 of course means successful, we pass this on
# 1 means that the user exited the prompt without specifying an option
# returns codes >=10 are custom return codes specified with '-kb-custom-<n> <keybind>' options
# that are passed to rofi. We subtract 9 from them to pass '<n>' to the caller
2019-09-15 20:42:11 +02:00
if returncode == 0:
key = 0
elif returncode == 1:
key = -1
elif returncode > 9:
key = returncode - 9
2021-10-01 19:08:36 +02:00
else: # This case should never be reached
key = -2
2019-09-15 20:42:11 +02:00
return key, index, selected