Adjust type hints in rflink cover (#73946)

* Adjust type hints in rflink cover

* Move definition back to init

* Use attributes

* Revert "Use attributes"

This reverts commit ff4851015d.

* Use _attr_should_poll
pull/73834/head
epenet 2022-06-29 09:46:05 +02:00 committed by GitHub
parent 00810235c9
commit aca0fd3178
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 14 deletions

View File

@ -1,4 +1,6 @@
"""Support for Rflink devices."""
from __future__ import annotations
import asyncio
from collections import defaultdict
import logging
@ -315,8 +317,9 @@ class RflinkDevice(Entity):
"""
platform = None
_state = None
_state: bool | None = None
_available = True
_attr_should_poll = False
def __init__(
self,
@ -369,11 +372,6 @@ class RflinkDevice(Entity):
"""Platform specific event handler."""
raise NotImplementedError()
@property
def should_poll(self):
"""No polling needed."""
return False
@property
def name(self):
"""Return a name for the device."""

View File

@ -125,7 +125,7 @@ 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):
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 +142,12 @@ class RflinkCover(RflinkCommand, CoverEntity, RestoreEntity):
self._state = False
@property
def should_poll(self):
"""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