Use new enums in motion_blinds (#61931)

This commit is contained in:
epenet 2021-12-16 22:47:23 +01:00 committed by GitHub
parent b3105dc218
commit bb538a9782
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 32 deletions

View File

@ -8,12 +8,7 @@ import voluptuous as vol
from homeassistant.components.cover import ( from homeassistant.components.cover import (
ATTR_POSITION, ATTR_POSITION,
ATTR_TILT_POSITION, ATTR_TILT_POSITION,
DEVICE_CLASS_AWNING, CoverDeviceClass,
DEVICE_CLASS_BLIND,
DEVICE_CLASS_CURTAIN,
DEVICE_CLASS_GATE,
DEVICE_CLASS_SHADE,
DEVICE_CLASS_SHUTTER,
CoverEntity, CoverEntity,
) )
from homeassistant.helpers import config_validation as cv, entity_platform from homeassistant.helpers import config_validation as cv, entity_platform
@ -35,29 +30,29 @@ _LOGGER = logging.getLogger(__name__)
POSITION_DEVICE_MAP = { POSITION_DEVICE_MAP = {
BlindType.RollerBlind: DEVICE_CLASS_SHADE, BlindType.RollerBlind: CoverDeviceClass.SHADE,
BlindType.RomanBlind: DEVICE_CLASS_SHADE, BlindType.RomanBlind: CoverDeviceClass.SHADE,
BlindType.HoneycombBlind: DEVICE_CLASS_SHADE, BlindType.HoneycombBlind: CoverDeviceClass.SHADE,
BlindType.DimmingBlind: DEVICE_CLASS_SHADE, BlindType.DimmingBlind: CoverDeviceClass.SHADE,
BlindType.DayNightBlind: DEVICE_CLASS_SHADE, BlindType.DayNightBlind: CoverDeviceClass.SHADE,
BlindType.RollerShutter: DEVICE_CLASS_SHUTTER, BlindType.RollerShutter: CoverDeviceClass.SHUTTER,
BlindType.Switch: DEVICE_CLASS_SHUTTER, BlindType.Switch: CoverDeviceClass.SHUTTER,
BlindType.RollerGate: DEVICE_CLASS_GATE, BlindType.RollerGate: CoverDeviceClass.GATE,
BlindType.Awning: DEVICE_CLASS_AWNING, BlindType.Awning: CoverDeviceClass.AWNING,
BlindType.Curtain: DEVICE_CLASS_CURTAIN, BlindType.Curtain: CoverDeviceClass.CURTAIN,
BlindType.CurtainLeft: DEVICE_CLASS_CURTAIN, BlindType.CurtainLeft: CoverDeviceClass.CURTAIN,
BlindType.CurtainRight: DEVICE_CLASS_CURTAIN, BlindType.CurtainRight: CoverDeviceClass.CURTAIN,
} }
TILT_DEVICE_MAP = { TILT_DEVICE_MAP = {
BlindType.VenetianBlind: DEVICE_CLASS_BLIND, BlindType.VenetianBlind: CoverDeviceClass.BLIND,
BlindType.ShangriLaBlind: DEVICE_CLASS_BLIND, BlindType.ShangriLaBlind: CoverDeviceClass.BLIND,
BlindType.DoubleRoller: DEVICE_CLASS_SHADE, BlindType.DoubleRoller: CoverDeviceClass.SHADE,
BlindType.VerticalBlind: DEVICE_CLASS_BLIND, BlindType.VerticalBlind: CoverDeviceClass.BLIND,
} }
TDBU_DEVICE_MAP = { TDBU_DEVICE_MAP = {
BlindType.TopDownBottomUp: DEVICE_CLASS_SHADE, BlindType.TopDownBottomUp: CoverDeviceClass.SHADE,
} }

View File

@ -1,13 +1,8 @@
"""Support for Motion Blinds sensors.""" """Support for Motion Blinds sensors."""
from motionblinds import BlindType from motionblinds import BlindType
from homeassistant.components.sensor import SensorEntity from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import ( from homeassistant.const import PERCENTAGE, SIGNAL_STRENGTH_DECIBELS_MILLIWATT
DEVICE_CLASS_BATTERY,
DEVICE_CLASS_SIGNAL_STRENGTH,
PERCENTAGE,
SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
)
from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity from homeassistant.helpers.update_coordinator import CoordinatorEntity
@ -43,7 +38,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
class MotionBatterySensor(CoordinatorEntity, SensorEntity): class MotionBatterySensor(CoordinatorEntity, SensorEntity):
"""Representation of a Motion Battery Sensor.""" """Representation of a Motion Battery Sensor."""
_attr_device_class = DEVICE_CLASS_BATTERY _attr_device_class = SensorDeviceClass.BATTERY
_attr_native_unit_of_measurement = PERCENTAGE _attr_native_unit_of_measurement = PERCENTAGE
def __init__(self, coordinator, blind): def __init__(self, coordinator, blind):
@ -119,7 +114,7 @@ class MotionTDBUBatterySensor(MotionBatterySensor):
class MotionSignalStrengthSensor(CoordinatorEntity, SensorEntity): class MotionSignalStrengthSensor(CoordinatorEntity, SensorEntity):
"""Representation of a Motion Signal Strength Sensor.""" """Representation of a Motion Signal Strength Sensor."""
_attr_device_class = DEVICE_CLASS_SIGNAL_STRENGTH _attr_device_class = SensorDeviceClass.SIGNAL_STRENGTH
_attr_entity_registry_enabled_default = False _attr_entity_registry_enabled_default = False
_attr_native_unit_of_measurement = SIGNAL_STRENGTH_DECIBELS_MILLIWATT _attr_native_unit_of_measurement = SIGNAL_STRENGTH_DECIBELS_MILLIWATT