Use new SensorDeviceClass enum in gios (#61609)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-12-13 02:15:37 +01:00 committed by GitHub
parent ca12f257ca
commit 9784523dfb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -4,17 +4,8 @@ from __future__ import annotations
from datetime import timedelta from datetime import timedelta
from typing import Final from typing import Final
from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT from homeassistant.components.sensor import SensorDeviceClass, SensorStateClass
from homeassistant.const import ( from homeassistant.const import CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
DEVICE_CLASS_AQI,
DEVICE_CLASS_CO,
DEVICE_CLASS_NITROGEN_DIOXIDE,
DEVICE_CLASS_OZONE,
DEVICE_CLASS_PM10,
DEVICE_CLASS_PM25,
DEVICE_CLASS_SULPHUR_DIOXIDE,
)
from .model import GiosSensorEntityDescription from .model import GiosSensorEntityDescription
@ -47,7 +38,7 @@ SENSOR_TYPES: Final[tuple[GiosSensorEntityDescription, ...]] = (
GiosSensorEntityDescription( GiosSensorEntityDescription(
key=ATTR_AQI, key=ATTR_AQI,
name="AQI", name="AQI",
device_class=DEVICE_CLASS_AQI, device_class=SensorDeviceClass.AQI,
value=None, value=None,
), ),
GiosSensorEntityDescription( GiosSensorEntityDescription(
@ -55,48 +46,48 @@ SENSOR_TYPES: Final[tuple[GiosSensorEntityDescription, ...]] = (
name="C6H6", name="C6H6",
icon="mdi:molecule", icon="mdi:molecule",
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
GiosSensorEntityDescription( GiosSensorEntityDescription(
key=ATTR_CO, key=ATTR_CO,
name="CO", name="CO",
device_class=DEVICE_CLASS_CO, device_class=SensorDeviceClass.CO,
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
GiosSensorEntityDescription( GiosSensorEntityDescription(
key=ATTR_NO2, key=ATTR_NO2,
name="NO2", name="NO2",
device_class=DEVICE_CLASS_NITROGEN_DIOXIDE, device_class=SensorDeviceClass.NITROGEN_DIOXIDE,
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
GiosSensorEntityDescription( GiosSensorEntityDescription(
key=ATTR_O3, key=ATTR_O3,
name="O3", name="O3",
device_class=DEVICE_CLASS_OZONE, device_class=SensorDeviceClass.OZONE,
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
GiosSensorEntityDescription( GiosSensorEntityDescription(
key=ATTR_PM10, key=ATTR_PM10,
name="PM10", name="PM10",
device_class=DEVICE_CLASS_PM10, device_class=SensorDeviceClass.PM10,
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
GiosSensorEntityDescription( GiosSensorEntityDescription(
key=ATTR_PM25, key=ATTR_PM25,
name="PM2.5", name="PM2.5",
device_class=DEVICE_CLASS_PM25, device_class=SensorDeviceClass.PM25,
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
GiosSensorEntityDescription( GiosSensorEntityDescription(
key=ATTR_SO2, key=ATTR_SO2,
name="SO2", name="SO2",
device_class=DEVICE_CLASS_SULPHUR_DIOXIDE, device_class=SensorDeviceClass.SULPHUR_DIOXIDE,
native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
), ),
) )