Adjust device class in xiaomi_miio (#83647)

This commit is contained in:
epenet 2022-12-10 11:10:13 +01:00 committed by GitHub
parent c947152c48
commit 3970da0ad3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -33,14 +33,12 @@ from homeassistant.const import (
CONF_TOKEN, CONF_TOKEN,
LIGHT_LUX, LIGHT_LUX,
PERCENTAGE, PERCENTAGE,
POWER_WATT,
PRESSURE_HPA,
REVOLUTIONS_PER_MINUTE, REVOLUTIONS_PER_MINUTE,
TEMP_CELSIUS, UnitOfPower,
TIME_DAYS, UnitOfPressure,
TIME_HOURS, UnitOfTemperature,
TIME_SECONDS, UnitOfTime,
VOLUME_CUBIC_METERS, UnitOfVolume,
) )
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity import EntityCategory from homeassistant.helpers.entity import EntityCategory
@ -162,7 +160,7 @@ SENSOR_TYPES = {
ATTR_TEMPERATURE: XiaomiMiioSensorDescription( ATTR_TEMPERATURE: XiaomiMiioSensorDescription(
key=ATTR_TEMPERATURE, key=ATTR_TEMPERATURE,
name="Temperature", name="Temperature",
native_unit_of_measurement=TEMP_CELSIUS, native_unit_of_measurement=UnitOfTemperature.CELSIUS,
device_class=SensorDeviceClass.TEMPERATURE, device_class=SensorDeviceClass.TEMPERATURE,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
@ -176,14 +174,14 @@ SENSOR_TYPES = {
ATTR_PRESSURE: XiaomiMiioSensorDescription( ATTR_PRESSURE: XiaomiMiioSensorDescription(
key=ATTR_PRESSURE, key=ATTR_PRESSURE,
name="Pressure", name="Pressure",
native_unit_of_measurement=PRESSURE_HPA, native_unit_of_measurement=UnitOfPressure.HPA,
device_class=SensorDeviceClass.PRESSURE, device_class=SensorDeviceClass.ATMOSPHERIC_PRESSURE,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
ATTR_LOAD_POWER: XiaomiMiioSensorDescription( ATTR_LOAD_POWER: XiaomiMiioSensorDescription(
key=ATTR_LOAD_POWER, key=ATTR_LOAD_POWER,
name="Load power", name="Load power",
native_unit_of_measurement=POWER_WATT, native_unit_of_measurement=UnitOfPower.WATT,
device_class=SensorDeviceClass.POWER, device_class=SensorDeviceClass.POWER,
), ),
ATTR_WATER_LEVEL: XiaomiMiioSensorDescription( ATTR_WATER_LEVEL: XiaomiMiioSensorDescription(
@ -237,7 +235,7 @@ SENSOR_TYPES = {
ATTR_USE_TIME: XiaomiMiioSensorDescription( ATTR_USE_TIME: XiaomiMiioSensorDescription(
key=ATTR_USE_TIME, key=ATTR_USE_TIME,
name="Use time", name="Use time",
native_unit_of_measurement=TIME_SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
icon="mdi:progress-clock", icon="mdi:progress-clock",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.TOTAL_INCREASING, state_class=SensorStateClass.TOTAL_INCREASING,
@ -304,7 +302,7 @@ SENSOR_TYPES = {
ATTR_FILTER_USE: XiaomiMiioSensorDescription( ATTR_FILTER_USE: XiaomiMiioSensorDescription(
key=ATTR_FILTER_HOURS_USED, key=ATTR_FILTER_HOURS_USED,
name="Filter use", name="Filter use",
native_unit_of_measurement=TIME_HOURS, native_unit_of_measurement=UnitOfTime.HOURS,
icon="mdi:clock-outline", icon="mdi:clock-outline",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
@ -313,7 +311,7 @@ SENSOR_TYPES = {
ATTR_FILTER_LEFT_TIME: XiaomiMiioSensorDescription( ATTR_FILTER_LEFT_TIME: XiaomiMiioSensorDescription(
key=ATTR_FILTER_LEFT_TIME, key=ATTR_FILTER_LEFT_TIME,
name="Filter time left", name="Filter time left",
native_unit_of_measurement=TIME_DAYS, native_unit_of_measurement=UnitOfTime.DAYS,
icon="mdi:clock-outline", icon="mdi:clock-outline",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
@ -331,7 +329,7 @@ SENSOR_TYPES = {
ATTR_DUST_FILTER_LIFE_REMAINING_DAYS: XiaomiMiioSensorDescription( ATTR_DUST_FILTER_LIFE_REMAINING_DAYS: XiaomiMiioSensorDescription(
key=ATTR_DUST_FILTER_LIFE_REMAINING_DAYS, key=ATTR_DUST_FILTER_LIFE_REMAINING_DAYS,
name="Dust filter life remaining days", name="Dust filter life remaining days",
native_unit_of_measurement=TIME_DAYS, native_unit_of_measurement=UnitOfTime.DAYS,
icon="mdi:clock-outline", icon="mdi:clock-outline",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
@ -349,7 +347,7 @@ SENSOR_TYPES = {
ATTR_UPPER_FILTER_LIFE_REMAINING_DAYS: XiaomiMiioSensorDescription( ATTR_UPPER_FILTER_LIFE_REMAINING_DAYS: XiaomiMiioSensorDescription(
key=ATTR_UPPER_FILTER_LIFE_REMAINING_DAYS, key=ATTR_UPPER_FILTER_LIFE_REMAINING_DAYS,
name="Upper filter life remaining days", name="Upper filter life remaining days",
native_unit_of_measurement=TIME_DAYS, native_unit_of_measurement=UnitOfTime.DAYS,
icon="mdi:clock-outline", icon="mdi:clock-outline",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
@ -365,8 +363,8 @@ SENSOR_TYPES = {
ATTR_PURIFY_VOLUME: XiaomiMiioSensorDescription( ATTR_PURIFY_VOLUME: XiaomiMiioSensorDescription(
key=ATTR_PURIFY_VOLUME, key=ATTR_PURIFY_VOLUME,
name="Purify volume", name="Purify volume",
native_unit_of_measurement=VOLUME_CUBIC_METERS, native_unit_of_measurement=UnitOfVolume.CUBIC_METERS,
device_class=SensorDeviceClass.GAS, device_class=SensorDeviceClass.VOLUME,
state_class=SensorStateClass.TOTAL_INCREASING, state_class=SensorStateClass.TOTAL_INCREASING,
entity_registry_enabled_default=False, entity_registry_enabled_default=False,
entity_category=EntityCategory.DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
@ -608,7 +606,7 @@ VACUUM_SENSORS = {
entity_category=EntityCategory.DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
f"last_clean_{ATTR_LAST_CLEAN_TIME}": XiaomiMiioSensorDescription( f"last_clean_{ATTR_LAST_CLEAN_TIME}": XiaomiMiioSensorDescription(
native_unit_of_measurement=TIME_SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
icon="mdi:timer-sand", icon="mdi:timer-sand",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
key=ATTR_LAST_CLEAN_TIME, key=ATTR_LAST_CLEAN_TIME,
@ -625,7 +623,7 @@ VACUUM_SENSORS = {
entity_category=EntityCategory.DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
f"current_{ATTR_STATUS_CLEAN_TIME}": XiaomiMiioSensorDescription( f"current_{ATTR_STATUS_CLEAN_TIME}": XiaomiMiioSensorDescription(
native_unit_of_measurement=TIME_SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
icon="mdi:timer-sand", icon="mdi:timer-sand",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
key=ATTR_STATUS_CLEAN_TIME, key=ATTR_STATUS_CLEAN_TIME,
@ -642,7 +640,7 @@ VACUUM_SENSORS = {
name="Current clean area", name="Current clean area",
), ),
f"clean_history_{ATTR_CLEAN_HISTORY_TOTAL_DURATION}": XiaomiMiioSensorDescription( f"clean_history_{ATTR_CLEAN_HISTORY_TOTAL_DURATION}": XiaomiMiioSensorDescription(
native_unit_of_measurement=TIME_SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
icon="mdi:timer-sand", icon="mdi:timer-sand",
key=ATTR_CLEAN_HISTORY_TOTAL_DURATION, key=ATTR_CLEAN_HISTORY_TOTAL_DURATION,
@ -681,7 +679,7 @@ VACUUM_SENSORS = {
entity_category=EntityCategory.DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
f"consumable_{ATTR_CONSUMABLE_STATUS_MAIN_BRUSH_LEFT}": XiaomiMiioSensorDescription( f"consumable_{ATTR_CONSUMABLE_STATUS_MAIN_BRUSH_LEFT}": XiaomiMiioSensorDescription(
native_unit_of_measurement=TIME_SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
icon="mdi:brush", icon="mdi:brush",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
key=ATTR_CONSUMABLE_STATUS_MAIN_BRUSH_LEFT, key=ATTR_CONSUMABLE_STATUS_MAIN_BRUSH_LEFT,
@ -690,7 +688,7 @@ VACUUM_SENSORS = {
entity_category=EntityCategory.DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
f"consumable_{ATTR_CONSUMABLE_STATUS_SIDE_BRUSH_LEFT}": XiaomiMiioSensorDescription( f"consumable_{ATTR_CONSUMABLE_STATUS_SIDE_BRUSH_LEFT}": XiaomiMiioSensorDescription(
native_unit_of_measurement=TIME_SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
icon="mdi:brush", icon="mdi:brush",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
key=ATTR_CONSUMABLE_STATUS_SIDE_BRUSH_LEFT, key=ATTR_CONSUMABLE_STATUS_SIDE_BRUSH_LEFT,
@ -699,7 +697,7 @@ VACUUM_SENSORS = {
entity_category=EntityCategory.DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
f"consumable_{ATTR_CONSUMABLE_STATUS_FILTER_LEFT}": XiaomiMiioSensorDescription( f"consumable_{ATTR_CONSUMABLE_STATUS_FILTER_LEFT}": XiaomiMiioSensorDescription(
native_unit_of_measurement=TIME_SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
icon="mdi:air-filter", icon="mdi:air-filter",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
key=ATTR_CONSUMABLE_STATUS_FILTER_LEFT, key=ATTR_CONSUMABLE_STATUS_FILTER_LEFT,
@ -708,7 +706,7 @@ VACUUM_SENSORS = {
entity_category=EntityCategory.DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
f"consumable_{ATTR_CONSUMABLE_STATUS_SENSOR_DIRTY_LEFT}": XiaomiMiioSensorDescription( f"consumable_{ATTR_CONSUMABLE_STATUS_SENSOR_DIRTY_LEFT}": XiaomiMiioSensorDescription(
native_unit_of_measurement=TIME_SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
icon="mdi:eye-outline", icon="mdi:eye-outline",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
key=ATTR_CONSUMABLE_STATUS_SENSOR_DIRTY_LEFT, key=ATTR_CONSUMABLE_STATUS_SENSOR_DIRTY_LEFT,