diff --git a/homeassistant/components/fjaraskupan/binary_sensor.py b/homeassistant/components/fjaraskupan/binary_sensor.py index 9af93eaf9c0..9f24a3d39d2 100644 --- a/homeassistant/components/fjaraskupan/binary_sensor.py +++ b/homeassistant/components/fjaraskupan/binary_sensor.py @@ -7,7 +7,7 @@ from dataclasses import dataclass from fjaraskupan import Device, State from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_PROBLEM, + BinarySensorDeviceClass, BinarySensorEntity, BinarySensorEntityDescription, ) @@ -34,13 +34,13 @@ SENSORS = ( EntityDescription( key="grease-filter", name="Grease Filter", - device_class=DEVICE_CLASS_PROBLEM, + device_class=BinarySensorDeviceClass.PROBLEM, is_on=lambda state: state.grease_filter_full, ), EntityDescription( key="carbon-filter", name="Carbon Filter", - device_class=DEVICE_CLASS_PROBLEM, + device_class=BinarySensorDeviceClass.PROBLEM, is_on=lambda state: state.carbon_filter_full, ), ) diff --git a/homeassistant/components/fjaraskupan/number.py b/homeassistant/components/fjaraskupan/number.py index 66f719abd6f..eecb0b3b8e1 100644 --- a/homeassistant/components/fjaraskupan/number.py +++ b/homeassistant/components/fjaraskupan/number.py @@ -5,9 +5,9 @@ from fjaraskupan import Device, State from homeassistant.components.number import NumberEntity from homeassistant.config_entries import ConfigEntry -from homeassistant.const import ENTITY_CATEGORY_CONFIG, TIME_MINUTES +from homeassistant.const import TIME_MINUTES from homeassistant.core import HomeAssistant -from homeassistant.helpers.entity import DeviceInfo, Entity +from homeassistant.helpers.entity import DeviceInfo, Entity, EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.update_coordinator import ( CoordinatorEntity, @@ -40,7 +40,7 @@ class PeriodicVentingTime(CoordinatorEntity[State], NumberEntity): _attr_max_value: float = 59 _attr_min_value: float = 0 _attr_step: float = 1 - _attr_entity_category = ENTITY_CATEGORY_CONFIG + _attr_entity_category = EntityCategory.CONFIG _attr_unit_of_measurement = TIME_MINUTES def __init__( diff --git a/homeassistant/components/fjaraskupan/sensor.py b/homeassistant/components/fjaraskupan/sensor.py index 1821008a1d7..8c19b3e3cec 100644 --- a/homeassistant/components/fjaraskupan/sensor.py +++ b/homeassistant/components/fjaraskupan/sensor.py @@ -4,17 +4,14 @@ from __future__ import annotations from fjaraskupan import Device, State from homeassistant.components.sensor import ( - DEVICE_CLASS_SIGNAL_STRENGTH, - STATE_CLASS_MEASUREMENT, + SensorDeviceClass, SensorEntity, + SensorStateClass, ) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import ( - ENTITY_CATEGORY_DIAGNOSTIC, - SIGNAL_STRENGTH_DECIBELS_MILLIWATT, -) +from homeassistant.const import SIGNAL_STRENGTH_DECIBELS_MILLIWATT from homeassistant.core import HomeAssistant -from homeassistant.helpers.entity import DeviceInfo, Entity +from homeassistant.helpers.entity import DeviceInfo, Entity, EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import StateType from homeassistant.helpers.update_coordinator import ( @@ -56,11 +53,11 @@ class RssiSensor(CoordinatorEntity[State], SensorEntity): self._attr_unique_id = f"{device.address}-signal-strength" self._attr_device_info = device_info self._attr_name = f"{device_info['name']} Signal Strength" - self._attr_device_class = DEVICE_CLASS_SIGNAL_STRENGTH - self._attr_state_class = STATE_CLASS_MEASUREMENT + self._attr_device_class = SensorDeviceClass.SIGNAL_STRENGTH + self._attr_state_class = SensorStateClass.MEASUREMENT self._attr_native_unit_of_measurement = SIGNAL_STRENGTH_DECIBELS_MILLIWATT self._attr_entity_registry_enabled_default = False - self._attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC + self._attr_entity_category = EntityCategory.DIAGNOSTIC @property def native_value(self) -> StateType: