From c1a09d2bac01f678b0210963bde2d2f41d39b3de Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 8 Dec 2021 21:13:59 +0100 Subject: [PATCH] Use new DeviceClass and StateClass enums in ambee (#61264) Co-authored-by: epenet --- homeassistant/components/ambee/const.py | 54 ++++++++++++------------- 1 file changed, 27 insertions(+), 27 deletions(-) diff --git a/homeassistant/components/ambee/const.py b/homeassistant/components/ambee/const.py index a63aa4b804d..8f8f2237654 100644 --- a/homeassistant/components/ambee/const.py +++ b/homeassistant/components/ambee/const.py @@ -6,15 +6,15 @@ import logging from typing import Final from homeassistant.components.sensor import ( - STATE_CLASS_MEASUREMENT, + SensorDeviceClass, SensorEntityDescription, + SensorStateClass, ) from homeassistant.const import ( CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, CONCENTRATION_PARTS_PER_BILLION, CONCENTRATION_PARTS_PER_CUBIC_METER, CONCENTRATION_PARTS_PER_MILLION, - DEVICE_CLASS_CO, ) DOMAIN: Final = "ambee" @@ -37,43 +37,43 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="particulate_matter_2_5", name="Particulate Matter < 2.5 μm", native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), SensorEntityDescription( key="particulate_matter_10", name="Particulate Matter < 10 μm", native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), SensorEntityDescription( key="sulphur_dioxide", name="Sulphur Dioxide (SO2)", native_unit_of_measurement=CONCENTRATION_PARTS_PER_BILLION, - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), SensorEntityDescription( key="nitrogen_dioxide", name="Nitrogen Dioxide (NO2)", native_unit_of_measurement=CONCENTRATION_PARTS_PER_BILLION, - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), SensorEntityDescription( key="ozone", name="Ozone", native_unit_of_measurement=CONCENTRATION_PARTS_PER_BILLION, - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), SensorEntityDescription( key="carbon_monoxide", name="Carbon Monoxide (CO)", - device_class=DEVICE_CLASS_CO, + device_class=SensorDeviceClass.CO, native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION, - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), SensorEntityDescription( key="air_quality_index", name="Air Quality Index (AQI)", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, ), ], SERVICE_POLLEN: [ @@ -81,21 +81,21 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="grass", name="Grass Pollen", icon="mdi:grass", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, ), SensorEntityDescription( key="tree", name="Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, ), SensorEntityDescription( key="weed", name="Weed Pollen", icon="mdi:sprout", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, ), SensorEntityDescription( @@ -120,7 +120,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="grass_poaceae", name="Poaceae Grass Pollen", icon="mdi:grass", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -128,7 +128,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_alder", name="Alder Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -136,7 +136,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_birch", name="Birch Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -144,7 +144,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_cypress", name="Cypress Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -152,7 +152,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_elm", name="Elm Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -160,7 +160,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_hazel", name="Hazel Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -168,7 +168,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_oak", name="Oak Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -176,7 +176,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_pine", name="Pine Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -184,7 +184,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_plane", name="Plane Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -192,7 +192,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="tree_poplar", name="Poplar Tree Pollen", icon="mdi:tree", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -200,7 +200,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="weed_chenopod", name="Chenopod Weed Pollen", icon="mdi:sprout", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -208,7 +208,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="weed_mugwort", name="Mugwort Weed Pollen", icon="mdi:sprout", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -216,7 +216,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="weed_nettle", name="Nettle Weed Pollen", icon="mdi:sprout", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ), @@ -224,7 +224,7 @@ SENSORS: dict[str, list[SensorEntityDescription]] = { key="weed_ragweed", name="Ragweed Weed Pollen", icon="mdi:sprout", - state_class=STATE_CLASS_MEASUREMENT, + state_class=SensorStateClass.MEASUREMENT, native_unit_of_measurement=CONCENTRATION_PARTS_PER_CUBIC_METER, entity_registry_enabled_default=False, ),