From ae1f3ae30c0847de1d252f84657f03b6d767fd9b Mon Sep 17 00:00:00 2001 From: Hazurl Date: Wed, 24 Jan 2018 12:03:29 +0100 Subject: [PATCH] Add getter for title/desciption/is_end in State --- src/state.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/state.py b/src/state.py index 81438ca..cad0224 100644 --- a/src/state.py +++ b/src/state.py @@ -5,6 +5,9 @@ class State(object): def __init__(self, title, description, is_end): self.assets = {} + self.title = title + self.description = description + self.end = is_end def set_music(self, music): @@ -36,12 +39,12 @@ def add_option(self, label, next_state, required, acquired): def get_title(self): """Return the title""" - pass + return self.title def get_description(self): """Return the description""" - pass + return self.description def get_assets(self): @@ -53,7 +56,7 @@ def get_assets(self): def is_end(self): """Return true if it's an end state, false otherwise""" - pass + return self.end def get_option_count(self):