Remove unused import.

[tabakrolletjie.git] / tabakrolletjie / gamestate.py

diff --git a/tabakrolletjie/gamestate.py b/tabakrolletjie/gamestate.py
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..714a0475b7b5507a170f8176276827feaa08bad7 100644 (file)
--- a/tabakrolletjie/gamestate.py
+++ b/tabakrolletjie/gamestate.py
@@ -0,0 +1,43 @@
+""" The game state. """
+
+
+class GameState(object):
+    def __init__(self):
+        self._state = {
+            "station": None,
+            "turnips": [],
+        }
+        self.harvested = 0
+        self.eaten = 0
+
+    @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 '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
+
+    @seeds.setter
+    def seeds(self, value):
+        self._state['seeds'] = value
+
+    def set_station(self, station):
+        self._state["station"] = station
+
+    def get_spawn_positions(self):
+        return self._state["station"]["config"]["spawn positions"]