62 lines
1.4 KiB
Python
62 lines
1.4 KiB
Python
import keyboard
|
|
import time
|
|
import os
|
|
|
|
|
|
def put_cursor(line, col):
|
|
return f'\033[{int(line)};{int(col)}f'
|
|
|
|
|
|
class Menu:
|
|
def __init__(self, start_line):
|
|
self.start_line = start_line
|
|
self.options = [
|
|
'test 1',
|
|
'test 2',
|
|
'test 3'
|
|
]
|
|
self.selected = 0
|
|
self.highlight = '\033[38;5;0m\033[48;5;2m'
|
|
self.regular = '\033[38;5;2m'
|
|
|
|
def next(self):
|
|
self.selected += 1
|
|
if self.selected > len(self.options) - 1:
|
|
self.selected = 0
|
|
|
|
def prev(self):
|
|
self.selected -= 1
|
|
if self.selected < 0:
|
|
self.selected = len(self.options) - 1
|
|
|
|
def draw(self):
|
|
term_size = os.get_terminal_size()
|
|
current_line = self.start_line
|
|
|
|
for i, option in enumerate(self.options):
|
|
print(f'{put_cursor(self.start_line + i, 0)}\033[2;0;0m', end='')
|
|
print(' ' * term_size.columns, end='\r')
|
|
if i == self.selected:
|
|
print(f'{self.highlight}{option}')
|
|
else:
|
|
print(f'{self.regular}{option}')
|
|
|
|
|
|
|
|
def main():
|
|
selected_item = 0
|
|
menu = Menu(10)
|
|
menu.draw()
|
|
while True:
|
|
pressed_key = keyboard.read_event()
|
|
if pressed_key.event_type == 'down':
|
|
if pressed_key.name == 'j':
|
|
menu.next()
|
|
elif pressed_key.name == 'k':
|
|
menu.prev()
|
|
menu.draw()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|