Add YS8009 support to Yolink (home-assistant#148538)

Author: matrixd2

homeassistant/components/yolink/manifest.json

@@ -6,5 +6,5 @@
   "dependencies": ["auth", "application_credentials"],
   "documentation": "https://www.home-assistant.io/integrations/yolink",
   "iot_class": "cloud_push",
-  "requirements": ["yolink-api==0.5.5"]
+  "requirements": ["yolink-api==0.5.7"]
 }

homeassistant/components/yolink/sensor.py

@@ -21,6 +21,7 @@
     ATTR_DEVICE_POWER_FAILURE_ALARM,
     ATTR_DEVICE_SIREN,
     ATTR_DEVICE_SMART_REMOTER,
+    ATTR_DEVICE_SOIL_TH_SENSOR,
     ATTR_DEVICE_SWITCH,
     ATTR_DEVICE_TH_SENSOR,
     ATTR_DEVICE_THERMOSTAT,
@@ -42,6 +43,7 @@
     PERCENTAGE,
     SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
     EntityCategory,
+    UnitOfConductivity,
     UnitOfEnergy,
     UnitOfLength,
     UnitOfPower,
@@ -103,6 +105,7 @@ class YoLinkSensorEntityDescription(SensorEntityDescription):
     ATTR_DEVICE_MANIPULATOR,
     ATTR_DEVICE_CO_SMOKE_SENSOR,
     ATTR_GARAGE_DOOR_CONTROLLER,
+    ATTR_DEVICE_SOIL_TH_SENSOR,
 ]
 
 BATTERY_POWER_SENSOR = [
@@ -122,6 +125,7 @@ class YoLinkSensorEntityDescription(SensorEntityDescription):
     ATTR_DEVICE_WATER_DEPTH_SENSOR,
     ATTR_DEVICE_WATER_METER_CONTROLLER,
     ATTR_DEVICE_MULTI_WATER_METER_CONTROLLER,
+    ATTR_DEVICE_SOIL_TH_SENSOR,
 ]
 
 MCU_DEV_TEMPERATURE_SENSOR = [
@@ -182,7 +186,7 @@ def cvt_volume(val: int | None) -> str | None:
         native_unit_of_measurement=PERCENTAGE,
         state_class=SensorStateClass.MEASUREMENT,
         exists_fn=lambda device: (
-            device.device_type in [ATTR_DEVICE_TH_SENSOR]
+            device.device_type in [ATTR_DEVICE_TH_SENSOR, ATTR_DEVICE_SOIL_TH_SENSOR]
             and device.device_model_name not in NONE_HUMIDITY_SENSOR_MODELS
         ),
     ),
@@ -191,7 +195,8 @@ def cvt_volume(val: int | None) -> str | None:
         device_class=SensorDeviceClass.TEMPERATURE,
         native_unit_of_measurement=UnitOfTemperature.CELSIUS,
         state_class=SensorStateClass.MEASUREMENT,
-        exists_fn=lambda device: device.device_type in [ATTR_DEVICE_TH_SENSOR],
+        exists_fn=lambda device: device.device_type
+        in [ATTR_DEVICE_TH_SENSOR, ATTR_DEVICE_SOIL_TH_SENSOR],
     ),
     # mcu temperature
     YoLinkSensorEntityDescription(
@@ -206,7 +211,7 @@ def cvt_volume(val: int | None) -> str | None:
         key="loraInfo",
         device_class=SensorDeviceClass.SIGNAL_STRENGTH,
         native_unit_of_measurement=SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
-        value=lambda value: value["signal"] if value is not None else None,
+        value=lambda value: value.get("signal") if value is not None else None,
         state_class=SensorStateClass.MEASUREMENT,
         entity_category=EntityCategory.DIAGNOSTIC,
         entity_registry_enabled_default=False,
@@ -302,6 +307,14 @@ def cvt_volume(val: int | None) -> str | None:
         exists_fn=lambda device: device.device_model_name in POWER_SUPPORT_MODELS,
         value=lambda value: value / 100 if value is not None else None,
     ),
+    YoLinkSensorEntityDescription(
+        key="conductivity",
+        device_class=SensorDeviceClass.CONDUCTIVITY,
+        native_unit_of_measurement=UnitOfConductivity.MICROSIEMENS_PER_CM,
+        state_class=SensorStateClass.MEASUREMENT,
+        exists_fn=lambda device: device.device_type in [ATTR_DEVICE_SOIL_TH_SENSOR],
+        should_update_entity=lambda value: value is not None,
+    ),
 )