diff --git a/homeassistant/components/ezviz/binary_sensor.py b/homeassistant/components/ezviz/binary_sensor.py index e7d8be80509..942ceeecdb2 100644 --- a/homeassistant/components/ezviz/binary_sensor.py +++ b/homeassistant/components/ezviz/binary_sensor.py @@ -2,8 +2,7 @@ from __future__ import annotations from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_MOTION, - DEVICE_CLASS_UPDATE, + BinarySensorDeviceClass, BinarySensorEntity, BinarySensorEntityDescription, ) @@ -20,7 +19,7 @@ PARALLEL_UPDATES = 1 BINARY_SENSOR_TYPES: dict[str, BinarySensorEntityDescription] = { "Motion_Trigger": BinarySensorEntityDescription( key="Motion_Trigger", - device_class=DEVICE_CLASS_MOTION, + device_class=BinarySensorDeviceClass.MOTION, ), "alarm_schedules_enabled": BinarySensorEntityDescription( key="alarm_schedules_enabled" @@ -28,7 +27,7 @@ BINARY_SENSOR_TYPES: dict[str, BinarySensorEntityDescription] = { "encrypted": BinarySensorEntityDescription(key="encrypted"), "upgrade_available": BinarySensorEntityDescription( key="upgrade_available", - device_class=DEVICE_CLASS_UPDATE, + device_class=BinarySensorDeviceClass.UPDATE, ), } diff --git a/homeassistant/components/ezviz/sensor.py b/homeassistant/components/ezviz/sensor.py index 5197982a2c5..a7334a3d18b 100644 --- a/homeassistant/components/ezviz/sensor.py +++ b/homeassistant/components/ezviz/sensor.py @@ -1,10 +1,13 @@ """Support for Ezviz sensors.""" from __future__ import annotations -from homeassistant.components.binary_sensor import DEVICE_CLASS_MOTION -from homeassistant.components.sensor import SensorEntity, SensorEntityDescription +from homeassistant.components.sensor import ( + SensorDeviceClass, + SensorEntity, + SensorEntityDescription, +) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import DEVICE_CLASS_BATTERY, PERCENTAGE +from homeassistant.const import PERCENTAGE from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -19,7 +22,7 @@ SENSOR_TYPES: dict[str, SensorEntityDescription] = { "battery_level": SensorEntityDescription( key="battery_level", native_unit_of_measurement=PERCENTAGE, - device_class=DEVICE_CLASS_BATTERY, + device_class=SensorDeviceClass.BATTERY, ), "alarm_sound_mod": SensorEntityDescription(key="alarm_sound_mod"), "detection_sensibility": SensorEntityDescription(key="detection_sensibility"), @@ -32,10 +35,7 @@ SENSOR_TYPES: dict[str, SensorEntityDescription] = { "supported_channels": SensorEntityDescription(key="supported_channels"), "local_ip": SensorEntityDescription(key="local_ip"), "wan_ip": SensorEntityDescription(key="wan_ip"), - "PIR_Status": SensorEntityDescription( - key="PIR_Status", - device_class=DEVICE_CLASS_MOTION, - ), + "PIR_Status": SensorEntityDescription(key="PIR_Status"), "last_alarm_type_code": SensorEntityDescription(key="last_alarm_type_code"), "last_alarm_type_name": SensorEntityDescription(key="last_alarm_type_name"), } diff --git a/homeassistant/components/ezviz/switch.py b/homeassistant/components/ezviz/switch.py index 0324d508f7f..ea8f1e83f70 100644 --- a/homeassistant/components/ezviz/switch.py +++ b/homeassistant/components/ezviz/switch.py @@ -6,7 +6,7 @@ from typing import Any from pyezviz.constants import DeviceSwitchType from pyezviz.exceptions import HTTPError, PyEzvizError -from homeassistant.components.switch import DEVICE_CLASS_SWITCH, SwitchEntity +from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -40,7 +40,7 @@ class EzvizSwitch(EzvizEntity, SwitchEntity): """Representation of a Ezviz sensor.""" coordinator: EzvizDataUpdateCoordinator - ATTR_DEVICE_CLASS = DEVICE_CLASS_SWITCH + _attr_device_class = SwitchDeviceClass.SWITCH def __init__( self, coordinator: EzvizDataUpdateCoordinator, serial: str, switch: str