From 777fec62a5d2499db49dd4fd012539673b578ce5 Mon Sep 17 00:00:00 2001
From: Erik Montnemery <erik@montnemery.com>
Date: Tue, 13 Jul 2021 19:35:55 +0200
Subject: [PATCH] Set device class for climacell temperature sensors (#52965)

---
 homeassistant/components/climacell/const.py  | 4 ++++
 homeassistant/components/climacell/sensor.py | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/homeassistant/components/climacell/const.py b/homeassistant/components/climacell/const.py
index 057cef5e993..7f93b9a6507 100644
--- a/homeassistant/components/climacell/const.py
+++ b/homeassistant/components/climacell/const.py
@@ -26,6 +26,7 @@ from homeassistant.components.weather import (
     ATTR_CONDITION_WINDY,
 )
 from homeassistant.const import (
+    ATTR_DEVICE_CLASS,
     ATTR_NAME,
     CONCENTRATION_MICROGRAMS_PER_CUBIC_FOOT,
     CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
@@ -34,6 +35,7 @@ from homeassistant.const import (
     CONF_UNIT_OF_MEASUREMENT,
     CONF_UNIT_SYSTEM_IMPERIAL,
     CONF_UNIT_SYSTEM_METRIC,
+    DEVICE_CLASS_TEMPERATURE,
     IRRADIATION_BTUS_PER_HOUR_SQUARE_FOOT,
     IRRADIATION_WATTS_PER_SQUARE_METER,
     LENGTH_KILOMETERS,
@@ -161,6 +163,7 @@ CC_SENSOR_TYPES = [
             val, TEMP_FAHRENHEIT, TEMP_CELSIUS
         ),
         ATTR_IS_METRIC_CHECK: True,
+        ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
     },
     {
         ATTR_FIELD: CC_ATTR_DEW_POINT,
@@ -171,6 +174,7 @@ CC_SENSOR_TYPES = [
             val, TEMP_FAHRENHEIT, TEMP_CELSIUS
         ),
         ATTR_IS_METRIC_CHECK: True,
+        ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
     },
     {
         ATTR_FIELD: CC_ATTR_PRESSURE_SURFACE_LEVEL,
diff --git a/homeassistant/components/climacell/sensor.py b/homeassistant/components/climacell/sensor.py
index 2c620cc65a1..e5d9a2cdb75 100644
--- a/homeassistant/components/climacell/sensor.py
+++ b/homeassistant/components/climacell/sensor.py
@@ -12,6 +12,7 @@ from homeassistant.components.sensor import SensorEntity
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
     ATTR_ATTRIBUTION,
+    ATTR_DEVICE_CLASS,
     ATTR_NAME,
     CONF_API_VERSION,
     CONF_NAME,
@@ -73,6 +74,7 @@ class BaseClimaCellSensorEntity(ClimaCellEntity, SensorEntity):
         """Initialize ClimaCell Sensor Entity."""
         super().__init__(config_entry, coordinator, api_version)
         self.sensor_type = sensor_type
+        self._attr_device_class = self.sensor_type.get(ATTR_DEVICE_CLASS)
 
     @property
     def entity_registry_enabled_default(self) -> bool: