Use attributes

This commit is contained in:
epenet 2022-06-29 06:27:13 +00:00
parent 8c10d0a4b1
commit ff4851015d

View file

@ -125,6 +125,9 @@ async def async_setup_platform(
class RflinkCover(RflinkCommand, CoverEntity, RestoreEntity): class RflinkCover(RflinkCommand, CoverEntity, RestoreEntity):
"""Rflink entity which can switch on/stop/off (eg: cover).""" """Rflink entity which can switch on/stop/off (eg: cover)."""
_attr_assumed_state = True
_attr_should_poll = False
async def async_added_to_hass(self) -> None: async def async_added_to_hass(self) -> None:
"""Restore RFLink cover state (OPEN/CLOSE).""" """Restore RFLink cover state (OPEN/CLOSE)."""
await super().async_added_to_hass() await super().async_added_to_hass()
@ -141,21 +144,11 @@ class RflinkCover(RflinkCommand, CoverEntity, RestoreEntity):
elif command in ["off", "alloff", "down"]: elif command in ["off", "alloff", "down"]:
self._state = False self._state = False
@property
def should_poll(self) -> bool:
"""No polling available in RFlink cover."""
return False
@property @property
def is_closed(self) -> bool | None: def is_closed(self) -> bool | None:
"""Return if the cover is closed.""" """Return if the cover is closed."""
return not self._state return not self._state
@property
def assumed_state(self) -> bool:
"""Return True because covers can be stopped midway."""
return True
async def async_close_cover(self, **kwargs: Any) -> None: async def async_close_cover(self, **kwargs: Any) -> None:
"""Turn the device close.""" """Turn the device close."""
await self._async_handle_command("close_cover") await self._async_handle_command("close_cover")