botany/menu_screen.py

139 lines
4.8 KiB
Python
Raw Normal View History

2017-03-08 08:18:45 +00:00
import curses, os, traceback, threading, time, datetime
2017-03-08 02:35:04 +00:00
class CursedMenu(object):
2017-03-08 08:18:45 +00:00
#TODO: create a side panel with log of events..?
2017-03-08 02:35:04 +00:00
'''A class which abstracts the horrors of building a curses-based menu system'''
def __init__(self, this_plant):
'''Initialization'''
self.screen = curses.initscr()
curses.noecho()
curses.cbreak()
curses.start_color()
curses.curs_set(0)
self.screen.keypad(1)
self.plant = this_plant
self.exit = False
# Highlighted and Normal line definitions
curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_WHITE)
self.highlighted = curses.color_pair(1)
self.normal = curses.A_NORMAL
2017-03-08 08:18:45 +00:00
screen_thread = threading.Thread(target=self.update_plant_live, args=())
screen_thread.daemon = True
screen_thread.start()
# TODO: tweaking this to try to get rid of garble bug
self.screen.clear()
2017-03-08 02:35:04 +00:00
def show(self, options, title="Title", subtitle="Subtitle"):
'''Draws a menu with the given parameters'''
self.set_options(options)
self.title = title
self.subtitle = subtitle
self.selected = 0
self.draw_menu()
def set_options(self, options):
'''Validates that the last option is "Exit"'''
if options[-1] is not 'Exit':
options.append('Exit')
self.options = options
def draw_menu(self):
'''Actually draws the menu and handles branching'''
request = ""
try:
while request is not "Exit":
self.draw()
request = self.get_user_input()
self.handle_request(request)
self.__exit__()
# Also calls __exit__, but adds traceback after
except Exception as exception:
self.__exit__()
traceback.print_exc()
def draw(self):
'''Draw the menu and lines'''
2017-03-08 08:18:45 +00:00
self.screen.refresh()
2017-03-08 02:35:04 +00:00
self.screen.border(0)
self.screen.addstr(2,2, self.title, curses.A_STANDOUT) # Title for this menu
self.screen.addstr(4,2, self.subtitle, curses.A_BOLD) #Subtitle for this menu
# Display all the menu items, showing the 'pos' item highlighted
for index in range(len(self.options)):
textstyle = self.normal
if index == self.selected:
textstyle = self.highlighted
self.screen.addstr(5+index,4, "%d - %s" % (index+1, self.options[index]), textstyle)
2017-03-08 08:18:45 +00:00
try:
self.screen.refresh()
except Exception as exception:
# Makes sure data is saved in event of a crash due to window resizing
self.__exit__()
traceback.print_exc()
2017-03-08 02:35:04 +00:00
def update_plant_live(self):
2017-03-08 08:18:45 +00:00
# TODO: fix thread synchronization issue.. text garbling
# Updates plant data on menu screen, live!
# Will eventually use this to display ascii art...
2017-03-08 02:35:04 +00:00
while self.exit is not True:
plant_string = self.plant.parse_plant()
plant_ticks = str(self.plant.ticks)
self.screen.addstr(10,2, plant_string, curses.A_NORMAL)
self.screen.addstr(11,2, plant_ticks, curses.A_NORMAL)
2017-03-08 08:18:45 +00:00
if self.plant.watered_date == datetime.datetime.now().date():
2017-03-08 08:27:57 +00:00
self.screen.addstr(6,13, " - plant watered today :)", curses.A_NORMAL)
2017-03-08 08:18:45 +00:00
2017-03-08 02:35:04 +00:00
self.screen.refresh()
time.sleep(1)
def get_user_input(self):
'''Gets the user's input and acts appropriately'''
user_in = self.screen.getch() # Gets user input
'''Enter and Exit Keys are special cases'''
if user_in == 10:
return self.options[self.selected]
if user_in == 27:
return self.options[-1]
# This is a number; check to see if we can set it
if user_in >= ord('1') and user_in <= ord(str(min(9,len(self.options)+1))):
self.selected = user_in - ord('0') - 1 # convert keypress back to a number, then subtract 1 to get index
return
# Increment or Decrement
if user_in == curses.KEY_DOWN: # down arrow
self.selected += 1
if user_in == curses.KEY_UP: # up arrow
self.selected -=1
self.selected = self.selected % len(self.options)
return
def handle_request(self, request):
'''This is where you do things with the request'''
if request is None: return
if request == 1:
self.screen.addstr(8,15,str(self.plant.ticks), curses.A_STANDOUT) # Title for this menu
self.screen.refresh()
2017-03-08 08:18:45 +00:00
if request == "water":
self.plant.water()
2017-03-08 02:35:04 +00:00
def __exit__(self):
self.exit = True
curses.curs_set(2)
curses.endwin()
os.system('clear')
'''demo'''
# cm = CursedMenu()
2017-03-08 08:18:45 +00:00
# cm.show([1,"water",3], title=' botany ', subtitle='Options')