X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fgamestate.py;h=9fe4fc85d3557ea0d991a48e664c3627b0912bf3;hb=48c809306d2aaf6ed2487aad4970138601d74a60;hp=6f7f5c45afe62ac586e7fae86e78876e8177e350;hpb=fc5270a33b3765154fb1e737414197e278eb1e52;p=tabakrolletjie.git

diff --git a/tabakrolletjie/gamestate.py b/tabakrolletjie/gamestate.py
index 6f7f5c4..9fe4fc8 100644
--- a/tabakrolletjie/gamestate.py
+++ b/tabakrolletjie/gamestate.py
@@ -1,4 +1,54 @@
 """ The game state. """
 
+
 class GameState(object):
-    pass
+    def __init__(self):
+        self._state = {
+            "station": None,
+            "turnips": [],
+            "seeds": None,
+        }
+        self.harvested = 0
+        self.eaten = 0
+        self.days = 0
+        self.resistances = {}
+
+    @property
+    def station(self):
+        return self._state["station"]
+
+    @property
+    def turnips(self):
+        return self._state["turnips"]
+
+    @turnips.setter
+    def turnips(self, turnip_list):
+        self._state["turnips"] = turnip_list
+
+    @property
+    def seeds(self):
+        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):
+        self._state['seeds'] = value
+
+    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"]