New version : Oin Thermostat

oin_thermostat/select.py

@@ -0,0 +1,90 @@
+import logging
+import math
+
+from sense_hat import ACTION_HELD, ACTION_RELEASED, SenseHat
+
+logger = logging.getLogger(__name__)
+
+
+class Selector:
+    def __init__(self, send_data):
+        self.stick = SenseHat().stick
+        self.temperature = None
+        self.mode = None
+        self.switch = None
+        self.preset_modes = []
+        self.send_data = send_data
+        self.switch_held = False
+        self.default_data = {"temperature": None, "mode": None, "switch": None}
+
+        self.stick.direction_middle = self.toggle
+
+        self.stick.direction_up = self.increase_temperature
+        self.stick.direction_down = self.decrease_temperature
+
+        self.stick.direction_right = self.next_mode
+        self.stick.direction_left = self.prev_mode
+
+    @property
+    def ha_options(self):
+        return {
+            "temperature": {
+                "p": "sensor",
+                "value_template": "{{ value_json.temperature }}",
+                "unique_id": "oin_temp",
+                "device_class": "temperature",
+                "entity_category": "diagnostic",
+                "icon": "mdi:thermostat",
+                "unit_of_measurement": "°C",
+            },
+            "mode": {
+                "p": "sensor",
+                "name": "Mode",
+                "value_template": "{{ value_json.mode }}",
+                "unique_id": "oin_mode",
+                "entity_category": "diagnostic",
+                "icon": "mdi:thermostat-auto",
+            },
+            "switch": {
+                "p": "sensor",
+                "name": "Switch",
+                "value_template": "{{ value_json.switch }}",
+                "unique_id": "oin_switch",
+                "entity_category": "diagnostic",
+                "icon": "mdi:thermostat-auto",
+            },
+        }
+
+    def increase_temperature(self, event):
+        if event.action != ACTION_RELEASED and self.temperature is not None:
+            self.callback(temperature=math.floor(self.temperature * 2) / 2 + 0.5)
+
+    def decrease_temperature(self, event):
+        if event.action != ACTION_RELEASED and self.temperature is not None:
+            self.callback(temperature=math.ceil(self.temperature * 2) / 2 - 0.5)
+
+    def next_mode(self, event):
+        if event.action != ACTION_RELEASED and self.mode is not None:
+            self.callback(
+                mode=self.preset_modes[
+                    (self.preset_modes.index(self.mode) + 1) % len(self.preset_modes)
+                ]
+            )
+
+    def prev_mode(self, event):
+        if event.action != ACTION_RELEASED and self.mode is not None:
+            self.callback(
+                mode=self.preset_modes[
+                    (self.preset_modes.index(self.mode) - 1) % len(self.preset_modes)
+                ]
+            )
+
+    def toggle(self, event):
+        if not self.switch_held and event.action == ACTION_HELD:
+            self.switch_held = True
+            self.callback(switch="off" if self.switch else "heat")
+        elif self.switch_held and event.action == ACTION_RELEASED:
+            self.switch_held = False
+
+    def callback(self, **kwargs):
+        self.send_data(self.default_data | kwargs)