From 8cde2e805b0129222695fbab4e5c67da8cbae6e1 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Fri, 17 Dec 2021 12:44:29 +0100
Subject: [PATCH] Use new SensorDeviceClass enum in climacell (#61362)

Co-authored-by: epenet <epenet@users.noreply.github.com>
---
 homeassistant/components/climacell/const.py | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/homeassistant/components/climacell/const.py b/homeassistant/components/climacell/const.py
index 3b5bc360d7c..83152cc38f2 100644
--- a/homeassistant/components/climacell/const.py
+++ b/homeassistant/components/climacell/const.py
@@ -17,7 +17,7 @@ from pyclimacell.const import (
     WeatherCode,
 )
 
-from homeassistant.components.sensor import SensorEntityDescription
+from homeassistant.components.sensor import SensorDeviceClass, SensorEntityDescription
 from homeassistant.components.weather import (
     ATTR_CONDITION_CLEAR_NIGHT,
     ATTR_CONDITION_CLOUDY,
@@ -37,9 +37,6 @@ from homeassistant.const import (
     CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
     CONCENTRATION_PARTS_PER_BILLION,
     CONCENTRATION_PARTS_PER_MILLION,
-    DEVICE_CLASS_CO,
-    DEVICE_CLASS_PRESSURE,
-    DEVICE_CLASS_TEMPERATURE,
     IRRADIATION_BTUS_PER_HOUR_SQUARE_FOOT,
     IRRADIATION_WATTS_PER_SQUARE_METER,
     LENGTH_KILOMETERS,
@@ -180,7 +177,7 @@ CC_SENSOR_TYPES = (
         unit_metric=TEMP_CELSIUS,
         metric_conversion=lambda val: temp_convert(val, TEMP_FAHRENHEIT, TEMP_CELSIUS),
         is_metric_check=True,
-        device_class=DEVICE_CLASS_TEMPERATURE,
+        device_class=SensorDeviceClass.TEMPERATURE,
     ),
     ClimaCellSensorEntityDescription(
         key=CC_ATTR_DEW_POINT,
@@ -189,7 +186,7 @@ CC_SENSOR_TYPES = (
         unit_metric=TEMP_CELSIUS,
         metric_conversion=lambda val: temp_convert(val, TEMP_FAHRENHEIT, TEMP_CELSIUS),
         is_metric_check=True,
-        device_class=DEVICE_CLASS_TEMPERATURE,
+        device_class=SensorDeviceClass.TEMPERATURE,
     ),
     ClimaCellSensorEntityDescription(
         key=CC_ATTR_PRESSURE_SURFACE_LEVEL,
@@ -200,7 +197,7 @@ CC_SENSOR_TYPES = (
             val, PRESSURE_INHG, PRESSURE_HPA
         ),
         is_metric_check=True,
-        device_class=DEVICE_CLASS_PRESSURE,
+        device_class=SensorDeviceClass.PRESSURE,
     ),
     ClimaCellSensorEntityDescription(
         key=CC_ATTR_SOLAR_GHI,
@@ -283,7 +280,7 @@ CC_SENSOR_TYPES = (
         name="Carbon Monoxide",
         unit_imperial=CONCENTRATION_PARTS_PER_MILLION,
         unit_metric=CONCENTRATION_PARTS_PER_MILLION,
-        device_class=DEVICE_CLASS_CO,
+        device_class=SensorDeviceClass.CO,
     ),
     ClimaCellSensorEntityDescription(
         key=CC_ATTR_SULFUR_DIOXIDE,
@@ -436,7 +433,7 @@ CC_V3_SENSOR_TYPES = (
         name="Carbon Monoxide",
         unit_imperial=CONCENTRATION_PARTS_PER_MILLION,
         unit_metric=CONCENTRATION_PARTS_PER_MILLION,
-        device_class=DEVICE_CLASS_CO,
+        device_class=SensorDeviceClass.CO,
     ),
     ClimaCellSensorEntityDescription(
         key=CC_V3_ATTR_SULFUR_DIOXIDE,