diff --git a/menu_screen.py b/menu_screen.py index b0b5188..40753ee 100644 --- a/menu_screen.py +++ b/menu_screen.py @@ -6,6 +6,7 @@ class CursedMenu(object): '''A class which abstracts the horrors of building a curses-based menu system''' def __init__(self, this_plant): '''Initialization''' + self.initialized = False self.screen = curses.initscr() curses.noecho() curses.cbreak() @@ -34,9 +35,9 @@ class CursedMenu(object): self.title = title self.subtitle = subtitle self.selected = 0 + self.initialized = True self.draw_menu() - def set_options(self, options): '''Validates that the last option is "Exit"''' if options[-1] is not 'Exit': @@ -90,11 +91,12 @@ class CursedMenu(object): def update_plant_live(self): # Updates plant data on menu screen, live! # Will eventually use this to display ascii art... - while self.exit is not True: - self.plant_string = self.plant.parse_plant() - self.plant_ticks = str(self.plant.ticks) - self.draw() - time.sleep(1) + if self.initialized: + while self.exit is not True: + self.plant_string = self.plant.parse_plant() + self.plant_ticks = str(self.plant.ticks) + self.draw() + time.sleep(1) def get_user_input(self): '''Gets the user's input and acts appropriately'''