tabakrolletjie/gamestate.py

   1 """ The game state. """
   2
   3
   4 class GameState(object):
   5     def __init__(self):
   6         self._state = {
   7             "station": None,
   8             "turnips": [],
   9             "seeds": None,
  10         }
  11         self.harvested = 0
  12         self.eaten = 0
  13         self.days = 0
  14         self.resistances = {}
  15
  16     @property
  17     def station(self):
  18         return self._state["station"]
  19
  20     @property
  21     def turnips(self):
  22         return self._state["turnips"]
  23
  24     @turnips.setter
  25     def turnips(self, turnip_list):
  26         self._state["turnips"] = turnip_list
  27
  28     @property
  29     def seeds(self):
  30         if self._state["seeds"] is None:
  31             if (self._state["station"] and
  32                     "seeds" in self._state["station"]["config"]):
  33                 self._state["seeds"] = (
  34                     self._state["station"]["config"]["seeds"])
  35             else:
  36                 self._state["seeds"] = 0
  37         return self._state["seeds"]
  38
  39     @seeds.setter
  40     def seeds(self, value):
  41         self._state['seeds'] = value
  42
  43     def set_station(self, station):
  44         self._state["station"] = station
  45         self._state["turnips"] = []
  46         self._state["seeds"] = None
  47         self.resistances = {}
  48         self.harvested = 0
  49         self.eaten = 0
  50         self.days = 0
  51
  52     def get_spawn_positions(self):
  53         return self._state["station"]["config"]["spawn positions"]
  54
  55     @property
  56     def turnip_target(self):
  57         return self._state["station"]["config"]["turnip target"]
  58
  59     def update_lights(self, lights):
  60         self._state["station"]["lights"] = lights.serialize_lights()
  61
  62     def serialize(self):
  63         result = {
  64             "station": self._state["station"],
  65             "turnips": self._state["turnips"],
  66             "seeds": self._state['seeds'],
  67             "harvested": self.harvested,
  68             "eaten": self.eaten,
  69             "days": self.days,
  70             "resistances": self.resistances,
  71         }
  72         return result
  73
  74     def load_from_json(self, json_data):
  75         self._state["station"] = dict(**json_data["station"])
  76         self._state["turnips"] = json_data["turnips"]
  77         self._state["seeds"] = json_data["seeds"]
  78         self.harvested = json_data["harvested"]
  79         self.eaten = json_data["harvested"]
  80         self.days = json_data["days"]
  81         self.resistances = dict(**json_data["resistances"])