Set device class for climacell temperature sensors (#52965)

This commit is contained in:
Erik Montnemery 2021-07-13 19:35:55 +02:00 committed by GitHub
parent 23b0633ae2
commit 777fec62a5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 0 deletions

View File

@ -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,

View File

@ -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: