81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
from os import system
|
|
import keyboard
|
|
from colorama import Fore
|
|
class MenuManager:
|
|
'''
|
|
Vytváří navigovatelná textová menu
|
|
'''
|
|
def __init__(self,selections:list,additional:str):
|
|
self.selected = 0 # aktuální výběr
|
|
self.selections = selections # Dostupné možnosti
|
|
self.additional = additional # Text, který se zobrazí nad menu
|
|
keyboard.add_hotkey("up",self.up)
|
|
keyboard.add_hotkey("down",self.down)
|
|
keyboard.add_hotkey("enter",self.make_selection)
|
|
self.show_menu()
|
|
input()
|
|
|
|
def make_selection(self) -> None:
|
|
keyboard.remove_all_hotkeys()
|
|
|
|
def up(self):
|
|
if self.selected == 0:
|
|
self.selected = len(self.selections)-1
|
|
else:
|
|
self.selected -= 1
|
|
system("cls||clear")
|
|
self.show_menu()
|
|
|
|
def down(self):
|
|
if self.selected == len(self.selections)-1:
|
|
self.selected = 0
|
|
else:
|
|
self.selected += 1
|
|
system("cls||clear")
|
|
self.show_menu()
|
|
|
|
def show_menu(self):
|
|
system("cls||clear")
|
|
print(self.additional)
|
|
for selection in self.selections:
|
|
if(self.selected == self.selections.index(selection)):
|
|
print(f"{Fore.RED}->{Fore.RESET} {selection}")
|
|
else:
|
|
print(f" {selection}")
|
|
|
|
class HasItemDialogue(MenuManager):
|
|
'''
|
|
Odvozená třída pro kontrolu, zda-li má hráč předmět
|
|
'''
|
|
def __init__(self, selections: list, additional: str,inv:list,need_item:list):
|
|
system("cls||clear")
|
|
self.inventory = inv
|
|
self.need_items = need_item
|
|
super().__init__(selections, additional)
|
|
|
|
def show_menu(self):
|
|
print(self.additional)
|
|
|
|
for i,selection in enumerate(self.selections):
|
|
if(self.need_items[i] != None and all(element not in self.inventory for element in self.need_items[i])):
|
|
c = ""
|
|
for i,item in enumerate(self.need_items[i]):
|
|
if item not in self.inventory:
|
|
if i == len(self.need_items)-self.need_items.count(None)-1: # last item, don't add a comma
|
|
c+=f"{item} "
|
|
else:
|
|
c+=f"{item}, "
|
|
# user does not have the needed item
|
|
if(self.selected == i):
|
|
print(f"{Fore.RED}-> {Fore.CYAN}{selection}{Fore.RESET} (Need {c})")
|
|
else:
|
|
print(f" {Fore.CYAN}{selection}{Fore.RESET}")
|
|
else:
|
|
# we don't need to change color for an item user doesn't have
|
|
if(self.selected == i):
|
|
print(f"{Fore.RED}->{Fore.RESET} {selection}")
|
|
else:
|
|
print(f" {selection}")
|
|
def make_selection(self) -> int:
|
|
keyboard.remove_all_hotkeys()
|