Initial commit

1 files changed, 114 insertions, 0 deletions

diff --git a/menu.py b/menu.py
new file mode 100644
index 0000000..a04e1ae
--- /dev/null
+++ b/menu.py
@@ -0,0 +1,114 @@
+# check if the entry.py and time modules are being used. Remove import if not.
+from msvcrt import *
+from entry import *
+from time import *
+
+class Menu():
+	def __init__(self, prompt, entries, color_int=7, side_bar=None):
+		self.prompt = prompt
+		self.entries = entries
+		self.iter_entries = iter(entries)
+		self.color_int = color_int
+		self.next_item = None
+		self.side_bar = side_bar
+
+	@property
+	def text_color(self):
+		if self.color_int > 0 and self.color_int <=7:
+			return f'\033[9{self.color_int}m'
+
+	@property
+	def indicate_color(self):
+		if self.color_int > 0 and self.color_int <=7:
+			return f'\033[10{self.color_int}m\033[30m'
+		
+	def refresh(self):
+		if self.color_int == 7: color_int = None
+		else: color_int = self.color_int
+		self.text = '\n'
+		if color_int: self.text += f'\033[9{color_int}m'
+		self.text += self.prompt
+		self.text += '\n' * 2
+		for entry in self.entries:	
+			if hasattr(entry, 'get_options'):
+				entry.get_options()
+			self.text += '\t' + entry.indicate(color_int) 
+			if color_int: self.text += f'\033[9{color_int}m'
+			self.text += '\n'
+		self.text += '\n'
+	
+	def open(self, prev_menu):
+		self.clear()
+		self.prev_menu = prev_menu
+		self.next_item = None
+		self.entries.set_select()
+		self.refresh()
+		self.print_side_bar()
+		print(self.text)
+		self.keyboard_nav()
+
+	def clear(self):
+		print('\033[0m', '\033[?25l', '\033[2J', '\033[0;0f', sep='', end='')
+
+	def reprint(self):
+		self.clear()
+		self.print_side_bar()
+		print(self.text)
+	
+	def print_side_bar(self):		
+		if self.side_bar:
+			lines = self.side_bar.split('\n')
+			l = 2
+			for line in lines:
+				print(f'\033[{l};40f', f'\033[9{self.color_int}m', line, '\033[0m', sep='', end='')
+				l+=1
+			print('\033[0;0f', end='')
+
+
+	def nav_entries(self, next='NEXT'):
+		self.entries.nav(next)
+		self.refresh()
+		self.reprint()
+
+
+	def edit_param(self, param):
+		param.edit = True
+		self.refresh()
+		self.reprint()
+		param.keyboard_edit()
+		param.edit = False
+		self.refresh()
+		self.reprint()
+
+	def keyboard_nav(self):
+		while True:
+			cmd = getch()
+			if cmd == b'\x1b' or cmd == b'\x08':
+				self.next_item = self.prev_menu
+				break
+			elif cmd == b'\x00' or cmd == b'\xe0':
+				cmd = getch()
+				if cmd == b'M' or cmd == b'P':
+					self.nav_entries(next='NEXT')
+				elif cmd == b'K' or cmd == b'H':
+					self.nav_entries(next='PREV')
+			elif cmd == b'\t':
+					self.nav_entries(next='NEXT')
+			elif cmd == b'\r':
+				for entry in self.entries:
+					if entry.select and type(entry) is Param:
+						self.edit_param(entry)
+					elif entry.select:
+						self.next_item = entry.item
+				if self.next_item != None:
+					break
+
+
+
+
+
+
+
+		
+	
+	
\ No newline at end of file