From f1979f8b68035f98bd493fb7c34b98176e7bedde Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Fri, 10 Dec 2021 14:53:58 +0100
Subject: [PATCH] Use new enums in fjaraskupan (#61438)
Co-authored-by: epenet <epenet@users.noreply.github.com>
---
.../components/fjaraskupan/binary_sensor.py | 6 +++---
homeassistant/components/fjaraskupan/number.py | 6 +++---
homeassistant/components/fjaraskupan/sensor.py | 17 +++++++----------
3 files changed, 13 insertions(+), 16 deletions(-)
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: