logic tweak

pull/24/head
Jake Funke 2019-02-28 00:41:15 +00:00
parent 300b9e8d00
commit 8fd35a5927
1 changed files with 3 additions and 7 deletions

View File

@ -79,16 +79,12 @@ class CursedMenu(object):
def update_options(self): def update_options(self):
# Makes sure you can get a new plant if it dies # Makes sure you can get a new plant if it dies
if self.plant.dead: if self.plant.dead or self.plant.stage == 5:
if "harvest" not in self.options: if "harvest" not in self.options:
self.options.insert(-1,"harvest") self.options.insert(-1,"harvest")
else: else:
if self.plant.stage == 5: if "harvest" in self.options:
if "harvest" not in self.options: self.options.remove("harvest")
self.options.insert(-1,"harvest")
else:
if "harvest" in self.options:
self.options.remove("harvest")
def set_options(self, options): def set_options(self, options):
# Validates that the last option is "exit" # Validates that the last option is "exit"