From 6bdc20ad68536579786d8e6365f7d778e9da92e8 Mon Sep 17 00:00:00 2001 From: Alberto Xamin Date: Tue, 31 Jan 2023 18:27:40 +0100 Subject: [PATCH 1/2] Migrate deprecated device classes --- .../midea_dehumidifier_lan/binary_sensor.py | 11 +++++------ custom_components/midea_dehumidifier_lan/sensor.py | 9 ++++----- 2 files changed, 9 insertions(+), 11 deletions(-) diff --git a/custom_components/midea_dehumidifier_lan/binary_sensor.py b/custom_components/midea_dehumidifier_lan/binary_sensor.py index 59b8aaa..ec277fb 100644 --- a/custom_components/midea_dehumidifier_lan/binary_sensor.py +++ b/custom_components/midea_dehumidifier_lan/binary_sensor.py @@ -1,8 +1,7 @@ """Adds binary sensors for appliances.""" from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_COLD, - DEVICE_CLASS_PROBLEM, + BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.config_entries import ConfigEntry @@ -63,7 +62,7 @@ class TankFullSensor(ApplianceEntity, BinarySensorEntity): dehumidifier from operating) """ - _attr_device_class = DEVICE_CLASS_PROBLEM + _attr_device_class = BinarySensorDeviceClass.PROBLEM _name_suffix = " Tank Full" def on_update(self) -> None: @@ -79,7 +78,7 @@ class TankRemovedSensor(ApplianceEntity, BinarySensorEntity): dehumidifier from operating) """ - _attr_device_class = DEVICE_CLASS_PROBLEM + _attr_device_class = BinarySensorDeviceClass.PROBLEM _name_suffix = " Tank Removed" _capability_attr = "pump" @@ -92,7 +91,7 @@ class FilterReplacementSensor(ApplianceEntity, BinarySensorEntity): Describes filter replacement binary sensors (indicated as problem) """ - _attr_device_class = DEVICE_CLASS_PROBLEM + _attr_device_class = BinarySensorDeviceClass.PROBLEM _attr_entity_registry_enabled_default = False _name_suffix = " Replace Filter" _capability_attr = "filter" @@ -111,7 +110,7 @@ class DefrostingSensor(ApplianceEntity, BinarySensorEntity): Describes defrosting mode binary sensors (indicated as cold) """ - _attr_device_class = DEVICE_CLASS_COLD + _attr_device_class = BinarySensorDeviceClass.COLD _attr_entity_registry_enabled_default = False _name_suffix = " Defrosting" diff --git a/custom_components/midea_dehumidifier_lan/sensor.py b/custom_components/midea_dehumidifier_lan/sensor.py index af7ea69..6b1390c 100644 --- a/custom_components/midea_dehumidifier_lan/sensor.py +++ b/custom_components/midea_dehumidifier_lan/sensor.py @@ -3,8 +3,7 @@ from homeassistant.components.sensor import SensorEntity, SensorStateClass from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( - DEVICE_CLASS_HUMIDITY, - DEVICE_CLASS_TEMPERATURE, + SensorDeviceClass, PERCENTAGE, TEMP_CELSIUS, ) @@ -48,7 +47,7 @@ async def async_setup_entry( class CurrentHumiditySensor(ApplianceEntity, SensorEntity): """Crrent environment humidity sensor""" - _attr_device_class = DEVICE_CLASS_HUMIDITY + _attr_device_class = SensorDeviceClass.HUMIDITY _attr_native_unit_of_measurement = PERCENTAGE _attr_state_class = SensorStateClass.MEASUREMENT _name_suffix = " Humidity" @@ -60,7 +59,7 @@ def on_update(self) -> None: class CurrentTemperatureSensor(ApplianceEntity, SensorEntity): """Current environment temperature sensor""" - _attr_device_class = DEVICE_CLASS_TEMPERATURE + _attr_device_class = SensorDeviceClass.TEMPERATURE _attr_native_unit_of_measurement = TEMP_CELSIUS _attr_state_class = SensorStateClass.MEASUREMENT _name_suffix = " Temperature" @@ -89,7 +88,7 @@ def on_update(self) -> None: class OutsideTemperatureSensor(ApplianceEntity, SensorEntity): """Current outside temperature sensor""" - _attr_device_class = DEVICE_CLASS_TEMPERATURE + _attr_device_class = SensorDeviceClass.TEMPERATURE _attr_native_unit_of_measurement = TEMP_CELSIUS _attr_state_class = SensorStateClass.MEASUREMENT _unique_id_prefx = UNIQUE_CLIMATE_PREFIX From 7a2d560f9008ffb7fa09d2829b04a1fde173684f Mon Sep 17 00:00:00 2001 From: Alberto Xamin Date: Fri, 3 Feb 2023 15:44:42 +0100 Subject: [PATCH 2/2] fix temperature sensors --- custom_components/midea_dehumidifier_lan/sensor.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/custom_components/midea_dehumidifier_lan/sensor.py b/custom_components/midea_dehumidifier_lan/sensor.py index 6b1390c..b6112af 100644 --- a/custom_components/midea_dehumidifier_lan/sensor.py +++ b/custom_components/midea_dehumidifier_lan/sensor.py @@ -1,9 +1,12 @@ """Adds sensors for each appliance.""" -from homeassistant.components.sensor import SensorEntity, SensorStateClass +from homeassistant.components.sensor import ( + SensorEntity, + SensorStateClass, + SensorDeviceClass +) from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( - SensorDeviceClass, PERCENTAGE, TEMP_CELSIUS, )