From 29e3d06a427c37321d5940889f9abfe81d6f10ec Mon Sep 17 00:00:00 2001 From: mbo18 Date: Mon, 23 Jan 2023 13:36:21 +0100 Subject: [PATCH] Add unit and device class to ZHA RSSI sensor (#85390) * Add unit device class to ZHA RSSI sensor * Remove unit and device class for LQI * mypy * isort * mypy2 * Update sensor.py --- homeassistant/components/zha/sensor.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/homeassistant/components/zha/sensor.py b/homeassistant/components/zha/sensor.py index 4e1c8a54a9f..75fd9f4d188 100644 --- a/homeassistant/components/zha/sensor.py +++ b/homeassistant/components/zha/sensor.py @@ -21,6 +21,7 @@ from homeassistant.const import ( CONCENTRATION_PARTS_PER_MILLION, LIGHT_LUX, PERCENTAGE, + SIGNAL_STRENGTH_DECIBELS_MILLIWATT, Platform, UnitOfApparentPower, UnitOfElectricCurrent, @@ -834,6 +835,8 @@ class RSSISensor(Sensor, id_suffix="rssi"): """RSSI sensor for a device.""" _attr_state_class: SensorStateClass = SensorStateClass.MEASUREMENT + _attr_device_class: SensorDeviceClass | None = SensorDeviceClass.SIGNAL_STRENGTH + _attr_native_unit_of_measurement: str | None = SIGNAL_STRENGTH_DECIBELS_MILLIWATT _attr_entity_category = EntityCategory.DIAGNOSTIC _attr_entity_registry_enabled_default = False _attr_should_poll = True # BaseZhaEntity defaults to False @@ -868,6 +871,8 @@ class LQISensor(RSSISensor, id_suffix="lqi"): """LQI sensor for a device.""" _attr_name: str = "LQI" + _attr_device_class = None + _attr_native_unit_of_measurement = None @MULTI_MATCH(