Add support remote_car from cloud #735

Author: AlexxIT

custom_components/yandex_station/__init__.py

@@ -50,6 +50,7 @@
 ]
 # for import section
 PLATFORMS = [
+    "binary_sensor",
     "button",
     "calendar",
     "camera",

custom_components/yandex_station/binary_sensor.py

@@ -0,0 +1,39 @@
+from homeassistant.components.binary_sensor import (
+    BinarySensorDeviceClass,
+    BinarySensorEntity,
+)
+
+from .core.entity import YandexCustomEntity
+from .hass import hass_utils
+
+INCLUDE_CAPABILITIES = ("devices.capabilities.lock",)
+
+ENTITY_DESCRIPTIONS = {
+    "lock": BinarySensorDeviceClass.LOCK,
+}
+
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    entities = []
+
+    for quasar, device, config in hass_utils.incluce_devices(hass, entry):
+        for instance in device["capabilities"]:
+            if instance["type"] in INCLUDE_CAPABILITIES:
+                entities.append(YandexBinarySensor(quasar, device, instance))
+
+    async_add_entities(entities)
+
+
+# noinspection PyAbstractClass
+class YandexBinarySensor(BinarySensorEntity, YandexCustomEntity):
+
+    def internal_init(self, capabilities: dict, properties: dict):
+        # {'access_methods': None, 'instance': 'lock', 'retrievable': True, 'values': ['closed', 'open']}
+        if desc := ENTITY_DESCRIPTIONS.get(self.instance):
+            self._attr_device_class = desc
+
+    def internal_update(self, capabilities: dict, properties: dict):
+        if value := capabilities.get(self.instance):
+            if self.instance == "lock":
+                # On means open (unlocked), Off means closed (locked)
+                self._attr_is_on = value == "open"

custom_components/yandex_station/climate.py

@@ -28,6 +28,12 @@ async def async_setup_entry(hass, entry, async_add_entities):
         for quasar, device, config in hass_utils.incluce_devices(hass, entry)
         if device["type"] in INCLUDE_TYPES
     )
+    # this should be fixed someday
+    async_add_entities(
+        YandexRemoteCarSeat(quasar, device, config)
+        for quasar, device, config in hass_utils.incluce_devices(hass, entry)
+        if device["type"] == "devices.types.remote_car.seat"
+    )
 
 
 # HA: auto, cool, dry, fan_only, heat; heat_cool, off
@@ -203,3 +209,35 @@ async def internal_set_hvac_mode(self, value: str) -> bool:
                     raise e
 
         return False
+
+
+class YandexRemoteCarSeat(ClimateEntity, YandexEntity):
+    _attr_temperature_unit = UnitOfTemperature.CELSIUS
+    _attr_supported_features = (
+        ClimateEntityFeature.TURN_ON
+        | ClimateEntityFeature.TURN_OFF
+        | ClimateEntityFeature.PRESET_MODE
+    )
+    _attr_hvac_modes = [HVACMode.OFF, HVACMode.HEAT]
+
+    def internal_init(self, capabilities: dict, properties: dict):
+        # {'instance': 'heating_mode', 'looped': False, 'name': 'режим обогрева', 'random_access': True, 'range': {'max': 3, 'min': 1, 'precision': 1}, 'retrievable': True, 'unit': ''}
+        if mode := capabilities.get("heating_mode"):
+            modes = range(mode["range"]["min"], mode["range"]["max"] + 1)
+            self._attr_preset_modes = [str(i) for i in modes]
+
+    def internal_update(self, capabilities: dict, properties: dict):
+        if (value := capabilities.get("on")) is not None:
+            self._attr_hvac_mode = HVACMode.HEAT if value else HVACMode.OFF
+
+        if value := capabilities.get("heating_mode"):
+            self._attr_preset_mode = str(value)
+
+    async def async_set_hvac_mode(self, hvac_mode: HVACMode):
+        if hvac_mode == HVACMode.OFF:
+            await self.device_action("on", False)
+        elif hvac_mode == HVACMode.HEAT:
+            await self.device_action("on", True)
+
+    async def async_set_preset_mode(self, preset_mode: str):
+        await self.device_action("heating_mode", int(preset_mode))

custom_components/yandex_station/core/entity.py

@@ -12,6 +12,8 @@
 def extract_instance(item: dict) -> str:
     if item["type"] == "devices.capabilities.on_off":
         return "on"
+    if item["type"] == "devices.capabilities.lock":
+        return "lock"
     return item["parameters"].get("instance")
 
 
@@ -112,7 +114,7 @@ async def device_actions(self, **kwargs):
 class YandexCustomEntity(YandexEntity):
     def __init__(self, quasar: YandexQuasar, device: dict, config: dict):
         self.instance = extract_instance(config)
-        super().__init__(quasar, device)
+        super().__init__(quasar, device, config)
         if name := config["parameters"].get("name"):
             self._attr_name += " " + name
         self._attr_unique_id += " " + self.instance

custom_components/yandex_station/core/yandex_quasar.py

@@ -39,6 +39,8 @@
     "camera_pan": "devices.capabilities.range",
     "camera_tilt": "devices.capabilities.range",
     "get_stream": "devices.capabilities.video_stream",
+    # "devices.types.remote_car.seat"
+    "heating_mode": "devices.capabilities.range",
     # don't work
     "hsv": "devices.capabilities.color_setting",
     "rgb": "devices.capabilities.color_setting",

custom_components/yandex_station/hass/hass_utils.py

@@ -102,6 +102,6 @@ async def load_fake_devies(hass: HomeAssistant, quasar: YandexQuasar):
 
     def job():
         with open(path, "rb") as f:
-            quasar.devices += json.load(f)
+            quasar.devices = json.load(f)
 
     await hass.async_add_executor_job(job)

custom_components/yandex_station/lock.py

@@ -0,0 +1,34 @@
+import logging
+
+from homeassistant.components.lock import LockEntity
+
+from .core.entity import YandexEntity
+from .hass import hass_utils
+
+_LOGGER = logging.getLogger(__name__)
+
+INCLUDE_CAPABILITIES = ("devices.capabilities.lock",)
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    entities = []
+
+    for quasar, device, config in hass_utils.incluce_devices(hass, entry):
+        for instance in device["capabilities"]:
+            if instance["type"] not in INCLUDE_CAPABILITIES:
+                continue
+            entities.append(YandexLock(quasar, device, instance))
+
+    async_add_entities(entities)
+
+
+# noinspection PyAbstractClass
+class YandexLock(LockEntity, YandexEntity):
+    def internal_update(self, capabilities: dict, properties: dict):
+        if value := capabilities.get("lock"):
+            self._attr_is_locked = value == "closed"
+
+    async def async_unlock(self):
+        await self.device_action("lock", "open")
+
+    async def async_lock(self):
+        await self.device_action("lock", "closed")

custom_components/yandex_station/sensor.py

@@ -13,6 +13,7 @@
     UnitOfElectricCurrent,
     UnitOfElectricPotential,
     UnitOfEnergy,
+    UnitOfLength,
     UnitOfPower,
     UnitOfPressure,
     UnitOfTemperature,
@@ -26,6 +27,7 @@
 
 # https://yandex.ru/dev/dialogs/smart-home/doc/concepts/device-type-sensor.html
 INCLUDE_TYPES = (
+    "devices.types.remote_car",
     "devices.types.sensor",
     "devices.types.sensor.button",
     "devices.types.sensor.climate",
@@ -79,6 +81,9 @@
     "electricity_meter": {"class": SENSOR.ENERGY, "units": UnitOfEnergy.KILO_WATT_HOUR},
     "gas_meter": {"class": SENSOR.GAS, "units": UnitOfVolume.CUBIC_METERS},
     "water_meter": {"class": SENSOR.WATER, "units": UnitOfVolume.CUBIC_METERS},
+    # there is no better option than a battery for fuel_level
+    "fuel_level": {"class": SENSOR.BATTERY, "units": PERCENTAGE},
+    "petrol_mileage": {"class": SENSOR.DISTANCE, "units": UnitOfLength.KILOMETERS},
 }
 
 
@@ -110,6 +115,11 @@ def internal_init(self, capabilities: dict, properties: dict):
             if "units" in desc:
                 self._attr_native_unit_of_measurement = desc["units"]
                 self._attr_state_class = SensorStateClass.MEASUREMENT
+        try:
+            if self.config["parameters"]["range"]["precision"] == 1:
+                self._attr_suggested_display_precision = 0
+        except KeyError:
+            pass
 
     def internal_update(self, capabilities: dict, properties: dict):
         if self.instance in properties:

custom_components/yandex_station/switch.py

@@ -9,6 +9,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 INCLUDE_TYPES = (
+    "devices.types.remote_car",
     "devices.types.switch",
     "devices.types.socket",
     "devices.types.ventilation",