116 lines
3.6 KiB
Python
116 lines
3.6 KiB
Python
"""BleBox cover entity."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from blebox_uniapi.box import Box
|
|
import blebox_uniapi.cover
|
|
|
|
from homeassistant.components.cover import (
|
|
ATTR_POSITION,
|
|
CoverDeviceClass,
|
|
CoverEntity,
|
|
CoverEntityFeature,
|
|
)
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import STATE_CLOSED, STATE_CLOSING, STATE_OPEN, STATE_OPENING
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from . import BleBoxEntity
|
|
from .const import DOMAIN, PRODUCT
|
|
|
|
BLEBOX_TO_COVER_DEVICE_CLASSES = {
|
|
"gate": CoverDeviceClass.GATE,
|
|
"gatebox": CoverDeviceClass.DOOR,
|
|
"shutter": CoverDeviceClass.SHUTTER,
|
|
}
|
|
|
|
|
|
BLEBOX_TO_HASS_COVER_STATES = {
|
|
None: None,
|
|
0: STATE_CLOSING, # moving down
|
|
1: STATE_OPENING, # moving up
|
|
2: STATE_OPEN, # manually stopped
|
|
3: STATE_CLOSED, # lower limit
|
|
4: STATE_OPEN, # upper limit / open
|
|
# gateController
|
|
5: STATE_OPEN, # overload
|
|
6: STATE_OPEN, # motor failure
|
|
# 7 is not used
|
|
8: STATE_OPEN, # safety stop
|
|
}
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Set up a BleBox entry."""
|
|
product: Box = hass.data[DOMAIN][config_entry.entry_id][PRODUCT]
|
|
entities = [
|
|
BleBoxCoverEntity(feature) for feature in product.features.get("covers", [])
|
|
]
|
|
async_add_entities(entities, True)
|
|
|
|
|
|
class BleBoxCoverEntity(BleBoxEntity[blebox_uniapi.cover.Cover], CoverEntity):
|
|
"""Representation of a BleBox cover feature."""
|
|
|
|
def __init__(self, feature: blebox_uniapi.cover.Cover) -> None:
|
|
"""Initialize a BleBox cover feature."""
|
|
super().__init__(feature)
|
|
self._attr_device_class = BLEBOX_TO_COVER_DEVICE_CLASSES[feature.device_class]
|
|
position = CoverEntityFeature.SET_POSITION if feature.is_slider else 0
|
|
stop = CoverEntityFeature.STOP if feature.has_stop else 0
|
|
self._attr_supported_features = (
|
|
position | stop | CoverEntityFeature.OPEN | CoverEntityFeature.CLOSE
|
|
)
|
|
|
|
@property
|
|
def current_cover_position(self) -> int | None:
|
|
"""Return the current cover position."""
|
|
position = self._feature.current
|
|
if position == -1: # possible for shutterBox
|
|
return None
|
|
|
|
return None if position is None else 100 - position
|
|
|
|
@property
|
|
def is_opening(self) -> bool | None:
|
|
"""Return whether cover is opening."""
|
|
return self._is_state(STATE_OPENING)
|
|
|
|
@property
|
|
def is_closing(self) -> bool | None:
|
|
"""Return whether cover is closing."""
|
|
return self._is_state(STATE_CLOSING)
|
|
|
|
@property
|
|
def is_closed(self) -> bool | None:
|
|
"""Return whether cover is closed."""
|
|
return self._is_state(STATE_CLOSED)
|
|
|
|
async def async_open_cover(self, **kwargs: Any) -> None:
|
|
"""Open the cover position."""
|
|
await self._feature.async_open()
|
|
|
|
async def async_close_cover(self, **kwargs: Any) -> None:
|
|
"""Close the cover position."""
|
|
await self._feature.async_close()
|
|
|
|
async def async_set_cover_position(self, **kwargs: Any) -> None:
|
|
"""Set the cover position."""
|
|
|
|
position = kwargs[ATTR_POSITION]
|
|
await self._feature.async_set_position(100 - position)
|
|
|
|
async def async_stop_cover(self, **kwargs: Any) -> None:
|
|
"""Stop the cover."""
|
|
await self._feature.async_stop()
|
|
|
|
def _is_state(self, state_name) -> bool | None:
|
|
value = BLEBOX_TO_HASS_COVER_STATES[self._feature.state]
|
|
return None if value is None else value == state_name
|