From ff4851015d5e15b1b1304554228ca274d586977d Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 29 Jun 2022 06:27:13 +0000 Subject: [PATCH] Use attributes --- homeassistant/components/rflink/cover.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/homeassistant/components/rflink/cover.py b/homeassistant/components/rflink/cover.py index d228c73ba2d..e16b684f088 100644 --- a/homeassistant/components/rflink/cover.py +++ b/homeassistant/components/rflink/cover.py @@ -125,6 +125,9 @@ async def async_setup_platform( class RflinkCover(RflinkCommand, CoverEntity, RestoreEntity): """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: """Restore RFLink cover state (OPEN/CLOSE).""" await super().async_added_to_hass() @@ -141,21 +144,11 @@ class RflinkCover(RflinkCommand, CoverEntity, RestoreEntity): elif command in ["off", "alloff", "down"]: self._state = False - @property - def should_poll(self) -> bool: - """No polling available in RFlink cover.""" - return False - @property def is_closed(self) -> bool | None: """Return if the cover is closed.""" 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: """Turn the device close.""" await self._async_handle_command("close_cover")