Use new enums in freedompro (#61445)

* Use new enums in freedompro
* Also update PLATFORMS

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-12-13 12:57:16 +01:00 committed by GitHub
parent ff015d4ea4
commit c8f2d4a82b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 31 additions and 37 deletions

View file

@ -1,11 +1,14 @@
"""Support for freedompro.""" """Support for freedompro."""
from __future__ import annotations
from datetime import timedelta from datetime import timedelta
import logging import logging
from typing import Final
from pyfreedompro import get_list, get_states from pyfreedompro import get_list, get_states
from homeassistant.config_entries import ConfigEntry 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.core import HomeAssistant
from homeassistant.helpers import aiohttp_client from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@ -14,15 +17,15 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
PLATFORMS = [ PLATFORMS: Final[list[Platform]] = [
"binary_sensor", Platform.BINARY_SENSOR,
"climate", Platform.CLIMATE,
"cover", Platform.COVER,
"fan", Platform.FAN,
"light", Platform.LIGHT,
"lock", Platform.LOCK,
"sensor", Platform.SENSOR,
"switch", Platform.SWITCH,
] ]

View file

@ -1,9 +1,6 @@
"""Support for Freedompro binary_sensor.""" """Support for Freedompro binary_sensor."""
from homeassistant.components.binary_sensor import ( from homeassistant.components.binary_sensor import (
DEVICE_CLASS_MOTION, BinarySensorDeviceClass,
DEVICE_CLASS_OCCUPANCY,
DEVICE_CLASS_OPENING,
DEVICE_CLASS_SMOKE,
BinarySensorEntity, BinarySensorEntity,
) )
from homeassistant.core import callback from homeassistant.core import callback
@ -13,10 +10,10 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN from .const import DOMAIN
DEVICE_CLASS_MAP = { DEVICE_CLASS_MAP = {
"smokeSensor": DEVICE_CLASS_SMOKE, "smokeSensor": BinarySensorDeviceClass.SMOKE,
"occupancySensor": DEVICE_CLASS_OCCUPANCY, "occupancySensor": BinarySensorDeviceClass.OCCUPANCY,
"motionSensor": DEVICE_CLASS_MOTION, "motionSensor": BinarySensorDeviceClass.MOTION,
"contactSensor": DEVICE_CLASS_OPENING, "contactSensor": BinarySensorDeviceClass.OPENING,
} }
DEVICE_KEY_MAP = { DEVICE_KEY_MAP = {

View file

@ -5,14 +5,10 @@ from pyfreedompro import put_state
from homeassistant.components.cover import ( from homeassistant.components.cover import (
ATTR_POSITION, ATTR_POSITION,
DEVICE_CLASS_BLIND,
DEVICE_CLASS_DOOR,
DEVICE_CLASS_GARAGE,
DEVICE_CLASS_GATE,
DEVICE_CLASS_WINDOW,
SUPPORT_CLOSE, SUPPORT_CLOSE,
SUPPORT_OPEN, SUPPORT_OPEN,
SUPPORT_SET_POSITION, SUPPORT_SET_POSITION,
CoverDeviceClass,
CoverEntity, CoverEntity,
) )
from homeassistant.const import CONF_API_KEY from homeassistant.const import CONF_API_KEY
@ -24,11 +20,11 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN from .const import DOMAIN
DEVICE_CLASS_MAP = { DEVICE_CLASS_MAP = {
"windowCovering": DEVICE_CLASS_BLIND, "windowCovering": CoverDeviceClass.BLIND,
"gate": DEVICE_CLASS_GATE, "gate": CoverDeviceClass.GATE,
"garageDoor": DEVICE_CLASS_GARAGE, "garageDoor": CoverDeviceClass.GARAGE,
"door": DEVICE_CLASS_DOOR, "door": CoverDeviceClass.DOOR,
"window": DEVICE_CLASS_WINDOW, "window": CoverDeviceClass.WINDOW,
} }
SUPPORTED_SENSORS = {"windowCovering", "gate", "garageDoor", "door", "window"} SUPPORTED_SENSORS = {"windowCovering", "gate", "garageDoor", "door", "window"}

View file

@ -1,10 +1,8 @@
"""Support for Freedompro sensor.""" """Support for Freedompro sensor."""
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
DEVICE_CLASS_HUMIDITY, SensorDeviceClass,
DEVICE_CLASS_ILLUMINANCE,
DEVICE_CLASS_TEMPERATURE,
STATE_CLASS_MEASUREMENT,
SensorEntity, SensorEntity,
SensorStateClass,
) )
from homeassistant.const import LIGHT_LUX, PERCENTAGE, TEMP_CELSIUS from homeassistant.const import LIGHT_LUX, PERCENTAGE, TEMP_CELSIUS
from homeassistant.core import callback from homeassistant.core import callback
@ -14,13 +12,13 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import DOMAIN from .const import DOMAIN
DEVICE_CLASS_MAP = { DEVICE_CLASS_MAP = {
"temperatureSensor": DEVICE_CLASS_TEMPERATURE, "temperatureSensor": SensorDeviceClass.TEMPERATURE,
"humiditySensor": DEVICE_CLASS_HUMIDITY, "humiditySensor": SensorDeviceClass.HUMIDITY,
"lightSensor": DEVICE_CLASS_ILLUMINANCE, "lightSensor": SensorDeviceClass.ILLUMINANCE,
} }
STATE_CLASS_MAP = { STATE_CLASS_MAP = {
"temperatureSensor": STATE_CLASS_MEASUREMENT, "temperatureSensor": SensorStateClass.MEASUREMENT,
"humiditySensor": STATE_CLASS_MEASUREMENT, "humiditySensor": SensorStateClass.MEASUREMENT,
"lightSensor": None, "lightSensor": None,
} }
UNIT_MAP = { UNIT_MAP = {