add stage control to thermostat

Author: gtfierro

python/xbos/devices/thermostat.py

@@ -42,6 +42,7 @@ def _handle(msg):
         print "Got Thermostat at {0} last alive {1}".format(uri, alive['val'])
 
         self.client.subscribe("{0}/signal/info".format(uri), _handle)
+        self.client.subscribe("{0}/stages/info".format(uri), _handle)
 
     @property
     def cooling_setpoint(self, timeout=30):
@@ -88,9 +89,9 @@ def temperature(self, timeout=30):
         return read_self_timeout(self, 'temperature', timeout)
 
 
-    def write(self, state):
+    def write(self, state, uri='state'):
         po = PayloadObject((2,1,1,0), None, msgpack.packb(state))
-        self.client.publish('{0}/slot/state'.format(self._uri),payload_objects=(po,))
+        self.client.publish('{0}/slot/{1}'.format(self._uri, uri),payload_objects=(po,))
 
     def set_heating_setpoint(self, value):
         self.write({'heating_setpoint': value})
@@ -108,8 +109,8 @@ def set_fan_mode(self, value):
         self.write({'fan_mode': value})
 
     def set_enabled_heat_stages(self, value):
-        self.write({'enabled_heat_stages': value})
+        self.write({'enabled_heat_stages': value}, 'stages')
 
     def set_enabled_cool_stages(self, value):
-        self.write({'enabled_cool_stages': value})
+        self.write({'enabled_cool_stages': value}, 'stages')