X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fgamestate.py;h=6f6725b8e0bda36ef5a031946306194d0496b0e2;hb=7fce31b4c16b25ba1892fadcc8ef48ef25f73d39;hp=714a0475b7b5507a170f8176276827feaa08bad7;hpb=769c2476fd3820be2daee7d4b87e5c8186e2da11;p=tabakrolletjie.git

diff --git a/tabakrolletjie/gamestate.py b/tabakrolletjie/gamestate.py
index 714a047..6f6725b 100644
--- a/tabakrolletjie/gamestate.py
+++ b/tabakrolletjie/gamestate.py
@@ -6,9 +6,12 @@ class GameState(object):
         self._state = {
             "station": None,
             "turnips": [],
+            "seeds": None,
         }
         self.harvested = 0
         self.eaten = 0
+        self.days = 0
+        self.resistances = {}
 
     @property
     def station(self):
@@ -24,13 +27,14 @@ class GameState(object):
 
     @property
     def seeds(self):
-        if 'seeds' in self._state:
-            return self._state['seeds']
-        elif (self._state["station"] and
-                'seeds' in self._state["station"]["config"]):
-            self._state['seeds'] = self._state["station"]["config"]["seeds"]
-            return self._state['seeds']
-        return 0
+        if self._state["seeds"] is None:
+            if (self._state["station"] and
+                    "seeds" in self._state["station"]["config"]):
+                self._state["seeds"] = (
+                    self._state["station"]["config"]["seeds"])
+            else:
+                self._state["seeds"] = 0
+        return self._state["seeds"]
 
     @seeds.setter
     def seeds(self, value):
@@ -38,6 +42,16 @@ class GameState(object):
 
     def set_station(self, station):
         self._state["station"] = station
+        self._state["turnips"] = []
+        self._state["seeds"] = None
+        self.resistances = {}
 
     def get_spawn_positions(self):
         return self._state["station"]["config"]["spawn positions"]
+
+    @property
+    def turnip_target(self):
+        return self._state["station"]["config"]["turnip target"]
+
+    def update_lights(self, lights):
+        self._state["station"]["lights"] = lights.serialize_lights()