Merge pull request #18 from Hazurl/state-options

Implement options related methods in State

Author: Jackojc

src/state.py

@@ -5,6 +5,7 @@ class State(object):
 
     def __init__(self, title, description, is_end):
         self.assets = {}
+        self.options = []
         self.title = title
         self.description = description
         self.end = is_end
@@ -34,7 +35,13 @@ def add_sprite(self, sprite, position_x, position_y, scale):
     def add_option(self, label, next_state, required, acquired):
         """Add an option, required and acquired are a dictonary with key 'items', 'history', 'stats'
         Each of them are a dictionary name to value"""
-        pass
+        option = {
+            'label' : label,
+            'next_state' : next_state,
+            'required' : required,
+            'acquired' : acquired
+        }
+        self.options.append(option)
 
 
     def get_title(self):
@@ -59,14 +66,9 @@ def is_end(self):
         return self.end
 
 
-    def get_option_count(self):
-        """Return the nuber of option in this state"""
-        pass
-
-
-    def get_option(self, index):
+    def get_options(self):
         """Return the option at 'index'"""
-        pass
+        return self.options.copy()
 
 
 def load_state_from_json(json_content):