Adjust type hints in rflink cover

This commit is contained in:
epenet
2022-06-24 07:50:49 +00:00
committed by Franck Nijhof
parent a267045a31
commit c4e17bfd98
2 changed files with 7 additions and 5 deletions

View File

@@ -315,7 +315,6 @@ class RflinkDevice(Entity):
"""
platform = None
_state = None
_available = True
def __init__(
@@ -345,6 +344,7 @@ class RflinkDevice(Entity):
self._nogroup_aliases = nogroup_aliases
self._should_fire_event = fire_event
self._signal_repetitions = signal_repetitions
self._state = None
@callback
def handle_event_callback(self, event):

View File

@@ -125,7 +125,9 @@ async def async_setup_platform(
class RflinkCover(RflinkCommand, CoverEntity, RestoreEntity):
"""Rflink entity which can switch on/stop/off (eg: cover)."""
async def async_added_to_hass(self):
_state: bool | None
async def async_added_to_hass(self) -> None:
"""Restore RFLink cover state (OPEN/CLOSE)."""
await super().async_added_to_hass()
if (old_state := await self.async_get_last_state()) is not None:
@@ -142,17 +144,17 @@ class RflinkCover(RflinkCommand, CoverEntity, RestoreEntity):
self._state = False
@property
def should_poll(self):
def should_poll(self) -> bool:
"""No polling available in RFlink cover."""
return False
@property
def is_closed(self):
def is_closed(self) -> bool | None:
"""Return if the cover is closed."""
return not self._state
@property
def assumed_state(self):
def assumed_state(self) -> bool:
"""Return True because covers can be stopped midway."""
return True