Add battery pack manager.

diff --git a/tabakrolletjie/battery.py b/tabakrolletjie/battery.py
new file mode 100644 (file)
index 0000000..c543698
--- /dev/null
+++ b/tabakrolletjie/battery.py
@@ -0,0 +1,36 @@
+""" Keep 'em charged and they'll never let you down. """
+
+
+class BatteryManager(object):
+    """ Manages a battery. """
+
+    def __init__(self, gamestate):
+        self._state = gamestate.station["battery"]
+
+    @property
+    def current(self):
+        return self._state["current"]
+
+    @current.setter
+    def current(self, value):
+        self._state["current"] = min(max(0, value), self.max)
+
+    @property
+    def max(self):
+        return self._state["max"]
+
+    @max.setter
+    def max(self, value):
+        self._state["max"] = max(0, value)
+
+    @property
+    def recharge(self):
+        return self._state["recharge"]
+
+    @recharge.setter
+    def recharge(self, value):
+        self._state["recharge"] = max(0, value)
+
+    def apply_recharge(self):
+        self.current += self.recharge
+        print self.current