diff --git a/homeassistant/components/freedompro/__init__.py b/homeassistant/components/freedompro/__init__.py index 6a6253f2e06..327b6314a34 100644 --- a/homeassistant/components/freedompro/__init__.py +++ b/homeassistant/components/freedompro/__init__.py @@ -1,11 +1,14 @@ """Support for freedompro.""" +from __future__ import annotations + from datetime import timedelta import logging +from typing import Final from pyfreedompro import get_list, get_states from homeassistant.config_entries import ConfigEntry -from homeassistant.const import CONF_API_KEY +from homeassistant.const import CONF_API_KEY, Platform from homeassistant.core import HomeAssistant from homeassistant.helpers import aiohttp_client from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed @@ -14,15 +17,15 @@ from .const import DOMAIN _LOGGER = logging.getLogger(__name__) -PLATFORMS = [ - "binary_sensor", - "climate", - "cover", - "fan", - "light", - "lock", - "sensor", - "switch", +PLATFORMS: Final[list[Platform]] = [ + Platform.BINARY_SENSOR, + Platform.CLIMATE, + Platform.COVER, + Platform.FAN, + Platform.LIGHT, + Platform.LOCK, + Platform.SENSOR, + Platform.SWITCH, ] diff --git a/homeassistant/components/freedompro/binary_sensor.py b/homeassistant/components/freedompro/binary_sensor.py index ac70824be4c..b69f942e532 100644 --- a/homeassistant/components/freedompro/binary_sensor.py +++ b/homeassistant/components/freedompro/binary_sensor.py @@ -1,9 +1,6 @@ """Support for Freedompro binary_sensor.""" from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_MOTION, - DEVICE_CLASS_OCCUPANCY, - DEVICE_CLASS_OPENING, - DEVICE_CLASS_SMOKE, + BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.core import callback @@ -13,10 +10,10 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import DOMAIN DEVICE_CLASS_MAP = { - "smokeSensor": DEVICE_CLASS_SMOKE, - "occupancySensor": DEVICE_CLASS_OCCUPANCY, - "motionSensor": DEVICE_CLASS_MOTION, - "contactSensor": DEVICE_CLASS_OPENING, + "smokeSensor": BinarySensorDeviceClass.SMOKE, + "occupancySensor": BinarySensorDeviceClass.OCCUPANCY, + "motionSensor": BinarySensorDeviceClass.MOTION, + "contactSensor": BinarySensorDeviceClass.OPENING, } DEVICE_KEY_MAP = { diff --git a/homeassistant/components/freedompro/cover.py b/homeassistant/components/freedompro/cover.py index fd6c747da46..2131069eb0e 100644 --- a/homeassistant/components/freedompro/cover.py +++ b/homeassistant/components/freedompro/cover.py @@ -5,14 +5,10 @@ from pyfreedompro import put_state from homeassistant.components.cover import ( ATTR_POSITION, - DEVICE_CLASS_BLIND, - DEVICE_CLASS_DOOR, - DEVICE_CLASS_GARAGE, - DEVICE_CLASS_GATE, - DEVICE_CLASS_WINDOW, SUPPORT_CLOSE, SUPPORT_OPEN, SUPPORT_SET_POSITION, + CoverDeviceClass, CoverEntity, ) from homeassistant.const import CONF_API_KEY @@ -24,11 +20,11 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import DOMAIN DEVICE_CLASS_MAP = { - "windowCovering": DEVICE_CLASS_BLIND, - "gate": DEVICE_CLASS_GATE, - "garageDoor": DEVICE_CLASS_GARAGE, - "door": DEVICE_CLASS_DOOR, - "window": DEVICE_CLASS_WINDOW, + "windowCovering": CoverDeviceClass.BLIND, + "gate": CoverDeviceClass.GATE, + "garageDoor": CoverDeviceClass.GARAGE, + "door": CoverDeviceClass.DOOR, + "window": CoverDeviceClass.WINDOW, } SUPPORTED_SENSORS = {"windowCovering", "gate", "garageDoor", "door", "window"} diff --git a/homeassistant/components/freedompro/sensor.py b/homeassistant/components/freedompro/sensor.py index 74b54474dbd..04fe7ecddeb 100644 --- a/homeassistant/components/freedompro/sensor.py +++ b/homeassistant/components/freedompro/sensor.py @@ -1,10 +1,8 @@ """Support for Freedompro sensor.""" from homeassistant.components.sensor import ( - DEVICE_CLASS_HUMIDITY, - DEVICE_CLASS_ILLUMINANCE, - DEVICE_CLASS_TEMPERATURE, - STATE_CLASS_MEASUREMENT, + SensorDeviceClass, SensorEntity, + SensorStateClass, ) from homeassistant.const import LIGHT_LUX, PERCENTAGE, TEMP_CELSIUS from homeassistant.core import callback @@ -14,13 +12,13 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import DOMAIN DEVICE_CLASS_MAP = { - "temperatureSensor": DEVICE_CLASS_TEMPERATURE, - "humiditySensor": DEVICE_CLASS_HUMIDITY, - "lightSensor": DEVICE_CLASS_ILLUMINANCE, + "temperatureSensor": SensorDeviceClass.TEMPERATURE, + "humiditySensor": SensorDeviceClass.HUMIDITY, + "lightSensor": SensorDeviceClass.ILLUMINANCE, } STATE_CLASS_MAP = { - "temperatureSensor": STATE_CLASS_MEASUREMENT, - "humiditySensor": STATE_CLASS_MEASUREMENT, + "temperatureSensor": SensorStateClass.MEASUREMENT, + "humiditySensor": SensorStateClass.MEASUREMENT, "lightSensor": None, } UNIT_MAP = {