evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Eliminate doorbird switch polling (#43215)

Browse source
This commit is contained in:
J. Nick Koston 2020-11-14 10:47:38 -10:00 • committed by GitHub
parent 8b63e22c99
commit 4e00a8a3d0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 26 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

34
homeassistant/components/doorbird/switch.py
View file

@ -2,6 +2,8 @@
import datetime import datetime
from homeassistant.components.switch import SwitchEntity from homeassistant.components.switch import SwitchEntity
from homeassistant.core import callback
from homeassistant.helpers.event import async_track_point_in_utc_time
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from .const import DOMAIN, DOOR_STATION, DOOR_STATION_INFO from .const import DOMAIN, DOOR_STATION, DOOR_STATION_INFO
@ -37,13 +39,13 @@ class DoorBirdSwitch(DoorBirdEntity, SwitchEntity):
self._doorstation = doorstation self._doorstation = doorstation
self._relay = relay self._relay = relay
self._state = False self._state = False
self._assume_off = datetime.datetime.min
if relay == IR_RELAY: if relay == IR_RELAY:
self._time = datetime.timedelta(minutes=5) self._time = datetime.timedelta(minutes=5)
else: else:
self._time = datetime.timedelta(seconds=5) self._time = datetime.timedelta(seconds=5)
self._unique_id = f"{self._mac_addr}_{self._relay}" self._unique_id = f"{self._mac_addr}_{self._relay}"
self._reset_sub = None
@property @property
def unique_id(self): def unique_id(self):
@ -63,27 +65,41 @@ class DoorBirdSwitch(DoorBirdEntity, SwitchEntity):
"""Return the icon to display.""" """Return the icon to display."""
return "mdi:lightbulb" if self._relay == IR_RELAY else "mdi:dip-switch" return "mdi:lightbulb" if self._relay == IR_RELAY else "mdi:dip-switch"
@property
def should_poll(self):
"""No need to poll."""
return False
@property @property
def is_on(self): def is_on(self):
"""Get the assumed state of the relay.""" """Get the assumed state of the relay."""
return self._state return self._state
def turn_on(self, **kwargs): async def async_turn_on(self, **kwargs):
"""Power the relay."""
if self._reset_sub is not None:
self._reset_sub()
self._reset_sub = None
self._reset_sub = async_track_point_in_utc_time(
self.hass, self._async_turn_off, dt_util.utcnow() + self._time
)
await self.hass.async_add_executor_job(self._turn_on)
self.async_write_ha_state()
def _turn_on(self):
"""Power the relay.""" """Power the relay."""
if self._relay == IR_RELAY: if self._relay == IR_RELAY:
self._state = self._doorstation.device.turn_light_on() self._state = self._doorstation.device.turn_light_on()
else: else:
self._state = self._doorstation.device.energize_relay(self._relay) self._state = self._doorstation.device.energize_relay(self._relay)
now = dt_util.utcnow() async def async_turn_off(self, **kwargs):
self._assume_off = now + self._time
def turn_off(self, **kwargs):
"""Turn off the relays is not needed. They are time-based.""" """Turn off the relays is not needed. They are time-based."""
raise NotImplementedError("DoorBird relays cannot be manually turned off.") raise NotImplementedError("DoorBird relays cannot be manually turned off.")
async def async_update(self): @callback
def _async_turn_off(self, *_):
"""Wait for the correct amount of assumed time to pass.""" """Wait for the correct amount of assumed time to pass."""
if self._state and self._assume_off <= dt_util.utcnow():
self._state = False self._state = False
self._assume_off = datetime.datetime.min self._reset_sub = None
self.async_write_ha_state()