Skip to content
This repository has been archived by the owner on Jun 17, 2024. It is now read-only.

Commit

Permalink
picker almost done
Browse files Browse the repository at this point in the history
  • Loading branch information
Joshua Hegedus committed Jan 22, 2023
1 parent 8e13096 commit 48932d0
Showing 1 changed file with 106 additions and 27 deletions.
133 changes: 106 additions & 27 deletions mod-downloader/picker.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,57 +4,136 @@
from getch import getch, pause
import curses
from mod import ModManager, ModPack, Mod
from abc import ABC, abstractmethod

KEYS_ENTER = (curses.KEY_ENTER, b"\n", b"\r")
KEYS_UP = (curses.KEY_UP,b'H', b'k')
KEYS_DOWN = (b'P', b'j')
KEYS_SELECT = (curses.KEY_RIGHT, b' ')
KEYS_ESC = b'\x1b'

screen: "curses._CursesWindow"

class Menu:
def __init__(self, title: str, options: list[str], indicator='->', selected=0):
# file_path = 'terminal_output.txt'
# sys.stdout = open(file_path, "w")

def __init__(self, title: str, options: list, callback, indicator: str, selected: int, shown_content: int):
self.title = title
self.indicator = indicator
self.indicator_space = ' ' * len(indicator)
self.options = options
self.selected = selected
self.page = 0
self.shown_content = shown_content
self.selected_stlye = colored.attr("underlined")
self.callback = callback

def __str__(self):
return self.title

def action_check(self) -> bytes:
move = getch()
if move in KEYS_DOWN and self.selected + 1 < len(self.options):
self.selected += 1
if move in KEYS_UP and self.selected - 1 >= 0:
self.selected -= 1
if self.selected >= self.page * self.shown_content + self.shown_content:
self.page += 1
elif self.selected <= (self.page * self.shown_content) - 1:
self.page -= 1
return move

@abstractmethod
def show(self, parent=''):
pass


class SingleMenu(Menu):
def __init__(self, title: str, options: list, callback, indicator='->', selected=0, shown_content=15):
super().__init__(title, options, callback, indicator, selected, shown_content)

def show(self):
selected_stlye = colored.attr("underlined")
def show(self, parent=''):
move = None
page = 0
max_page = (len(self.options) / 10) * 10
if max_page < len(self.options):
max_page += 1
print(max_page)
while move not in KEYS_ENTER:
os.system('cls')
print(f'Page: {page}, Selected: {self.selected}')
for index in range(page * 10, 10 + page * 10):
print(f'{parent}{self.title}')
for index in range(self.page * self.shown_content, self.shown_content + self.page * self.shown_content):
if index < len(self.options):
option = self.options[index]
if index == self.selected:
print(stylize(f'{self.indicator} {option}', selected_stlye))
print(stylize(f'{self.indicator} {option}', self.selected_stlye))
else:
print(f'{self.indicator_space} {option}')
move = getch()
if move in KEYS_DOWN and self.selected + 1 < len(self.options):
self.selected += 1
if move in KEYS_UP and self.selected - 1 >= 0:
self.selected -= 1
if self.selected >= page * 10 + 10:
page += 1
elif self.selected <= (page * 10) - 1:
page -= 1
return self.options[self.selected]
move = self.action_check()
if move in KEYS_ESC:
return None
self.callback(self.options[self.selected])


class MultiMenu(Menu):
def __init__(self, title: str, options: list, callback, indicator='->', selected=0, shown_content=15, select_open='[', select_close=']', select_mark='*', unselect_mark=' '):
super().__init__(title, options, callback, indicator, selected, shown_content)

self.unselected = f'{select_open}{unselect_mark}{select_close}'
self.selected_in = f'{select_open}{select_mark}{select_close}'

def show(self, parent=''):
move = None
selection = []
while move not in KEYS_ENTER:
os.system('cls')
print(f'{parent}{self.title}')
for index in range(self.page * self.shown_content, self.shown_content + self.page * self.shown_content):
if index < len(self.options):
option = self.options[index]
if index == self.selected:
if option in selection:
print(stylize(f'{self.indicator} {self.selected_in} {option}', self.selected_stlye))
else:
print(stylize(f'{self.indicator} {self.unselected} {option}', self.selected_stlye))
else:
if option in selection:
print(f'{self.indicator_space} {self.selected_in} {option}')
else:
print(f'{self.indicator_space} {self.unselected} {option}')
move = self.action_check()
if move in KEYS_SELECT:
if self.options[self.selected] in selection:
selection.remove(self.options[self.selected])
else:
selection.append(self.options[self.selected])
if move in KEYS_ESC:
return None
self.callback(selection)


class MenuWrapper(Menu):
def __init__(self, title: str, options: list[Menu], indicator='->', selected=0, shown_content=15):
super().__init__(title, options, None, indicator, selected, shown_content)

def show(self, parent=''):
move = KEYS_ENTER[1] # type: bytes
while move not in KEYS_ESC:
os.system('cls')
print(f'{parent}{self.title}')
for index in range(self.page * self.shown_content, self.shown_content + self.page * self.shown_content):
if index < len(self.options):
option = self.options[index]
if index == self.selected:
print(stylize(f'{self.indicator} {option}', self.selected_stlye))
else:
print(f'{self.indicator_space} {option}')
move = self.action_check()
if move in KEYS_ENTER:
self.options[self.selected].show(f'{self.title} -> ')

mods = ModManager(True)

menu = Menu("Test Menu", mods.mod_list)
pack = menu.show() # type: Mod
print(f'Selected item: {pack}')
def print_results(results):
print(results)
exit(0)

menu1 = SingleMenu("Menu 1", ["Opt 1", "Opt 2", "Opt 3"], print_results)
menu2 = SingleMenu("Menu 2", ["2Opt 1", "2Opt 2", "2Opt 3"], print_results)
menu3 = MultiMenu("Mod Selection", mods.mod_list, print_results, shown_content=20)

main_menu = MenuWrapper("Main Menu", [menu1, menu2, menu3])
main_menu.show()

0 comments on commit 48932d0

Please sign in to comment.