Use attr** in somfy-mylink (#62381)
Co-authored-by: epenet <epenet@users.noreply.github.com>pull/62417/head
epenet
GitHub
parent
afc42ff835
commit
07e1e174ac
|
|
@ -1,12 +1,7 @@
|
|||
"""Cover Platform for the Somfy MyLink component."""
|
||||
import logging
|
||||
|
||||
from homeassistant.components.cover import (
|
||||
DEVICE_CLASS_BLIND,
|
||||
DEVICE_CLASS_SHUTTER,
|
||||
DEVICE_CLASS_WINDOW,
|
||||
CoverEntity,
|
||||
)
|
||||
from homeassistant.components.cover import CoverDeviceClass, CoverEntity
|
||||
from homeassistant.const import STATE_CLOSED, STATE_OPEN
|
||||
from homeassistant.helpers.entity import DeviceInfo
|
||||
from homeassistant.helpers.restore_state import RestoreEntity
|
||||
|
|
@ -21,7 +16,10 @@ from .const import (
|
|||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
MYLINK_COVER_TYPE_TO_DEVICE_CLASS = {0: DEVICE_CLASS_BLIND, 1: DEVICE_CLASS_SHUTTER}
|
||||
MYLINK_COVER_TYPE_TO_DEVICE_CLASS = {
|
||||
0: CoverDeviceClass.BLIND,
|
||||
1: CoverDeviceClass.SHUTTER,
|
||||
}
|
||||
|
||||
|
||||
async def async_setup_entry(hass, config_entry, async_add_entities):
|
||||
|
|
@ -38,7 +36,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
|
|||
"target_id": cover["targetID"],
|
||||
"name": cover["name"],
|
||||
"device_class": MYLINK_COVER_TYPE_TO_DEVICE_CLASS.get(
|
||||
cover.get("type"), DEVICE_CLASS_WINDOW
|
||||
cover.get("type"), CoverDeviceClass.WINDOW
|
||||
),
|
||||
"reverse": reversed_target_ids.get(cover["targetID"], False),
|
||||
}
|
||||
|
|
@ -57,76 +55,33 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
|
|||
class SomfyShade(RestoreEntity, CoverEntity):
|
||||
"""Object for controlling a Somfy cover."""
|
||||
|
||||
_attr_should_poll = False
|
||||
_attr_assumed_state = True
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
somfy_mylink,
|
||||
target_id,
|
||||
name="SomfyShade",
|
||||
reverse=False,
|
||||
device_class=DEVICE_CLASS_WINDOW,
|
||||
device_class=CoverDeviceClass.WINDOW,
|
||||
):
|
||||
"""Initialize the cover."""
|
||||
self.somfy_mylink = somfy_mylink
|
||||
self._target_id = target_id
|
||||
self._name = name
|
||||
self._attr_unique_id = target_id
|
||||
self._attr_name = name
|
||||
self._reverse = reverse
|
||||
self._closed = None
|
||||
self._is_opening = None
|
||||
self._is_closing = None
|
||||
self._device_class = device_class
|
||||
|
||||
@property
|
||||
def should_poll(self):
|
||||
"""No polling since assumed state."""
|
||||
return False
|
||||
|
||||
@property
|
||||
def unique_id(self):
|
||||
"""Return the unique ID of this cover."""
|
||||
return self._target_id
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Return the name of the cover."""
|
||||
return self._name
|
||||
|
||||
@property
|
||||
def assumed_state(self):
|
||||
"""Let HA know the integration is assumed state."""
|
||||
return True
|
||||
|
||||
@property
|
||||
def device_class(self):
|
||||
"""Return the class of this device, from component DEVICE_CLASSES."""
|
||||
return self._device_class
|
||||
|
||||
@property
|
||||
def is_opening(self):
|
||||
"""Return if the cover is opening."""
|
||||
return self._is_opening
|
||||
|
||||
@property
|
||||
def is_closing(self):
|
||||
"""Return if the cover is closing."""
|
||||
return self._is_closing
|
||||
|
||||
@property
|
||||
def is_closed(self) -> bool:
|
||||
"""Return if the cover is closed."""
|
||||
return self._closed
|
||||
|
||||
@property
|
||||
def device_info(self) -> DeviceInfo:
|
||||
"""Return the device_info of the device."""
|
||||
return DeviceInfo(
|
||||
self._attr_device_class = device_class
|
||||
self._attr_device_info = DeviceInfo(
|
||||
identifiers={(DOMAIN, self._target_id)},
|
||||
manufacturer=MANUFACTURER,
|
||||
name=self._name,
|
||||
name=name,
|
||||
)
|
||||
|
||||
async def async_close_cover(self, **kwargs):
|
||||
"""Close the cover."""
|
||||
self._is_closing = True
|
||||
self._attr_is_closing = True
|
||||
self.async_write_ha_state()
|
||||
try:
|
||||
# Blocks until the close command is sent
|
||||
|
|
@ -134,14 +89,14 @@ class SomfyShade(RestoreEntity, CoverEntity):
|
|||
await self.somfy_mylink.move_down(self._target_id)
|
||||
else:
|
||||
await self.somfy_mylink.move_up(self._target_id)
|
||||
self._closed = True
|
||||
self._attr_is_closed = True
|
||||
finally:
|
||||
self._is_closing = None
|
||||
self._attr_is_closing = None
|
||||
self.async_write_ha_state()
|
||||
|
||||
async def async_open_cover(self, **kwargs):
|
||||
"""Open the cover."""
|
||||
self._is_opening = True
|
||||
self._attr_is_opening = True
|
||||
self.async_write_ha_state()
|
||||
try:
|
||||
# Blocks until the open command is sent
|
||||
|
|
@ -149,9 +104,9 @@ class SomfyShade(RestoreEntity, CoverEntity):
|
|||
await self.somfy_mylink.move_up(self._target_id)
|
||||
else:
|
||||
await self.somfy_mylink.move_down(self._target_id)
|
||||
self._closed = False
|
||||
self._attr_is_closed = False
|
||||
finally:
|
||||
self._is_opening = None
|
||||
self._attr_is_opening = None
|
||||
self.async_write_ha_state()
|
||||
|
||||
async def async_stop_cover(self, **kwargs):
|
||||
|
|
@ -168,4 +123,4 @@ class SomfyShade(RestoreEntity, CoverEntity):
|
|||
STATE_OPEN,
|
||||
STATE_CLOSED,
|
||||
):
|
||||
self._closed = last_state.state == STATE_CLOSED
|
||||
self._attr_is_closed = last_state.state == STATE_CLOSED
|
||||
|
|
|
|||
Loading…
Reference in New Issue