jeena/home-assistant
1
0
Fork 0
Code Issues Pull requests Projects Packages Wiki Activity Actions

replace sleeps with track_point_in_time

Browse source
This commit is contained in:
sfam 2015-10-13 00:56:24 +00:00
parent e57b3ae847
commit 1b7ce2146c
4 changed files with 69 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

77
homeassistant/components/alarm_control_panel/manual.py
View file

@ -33,8 +33,10 @@ Default is 120 seconds.
"""
import logging
import time
import datetime
import homeassistant.components.alarm_control_panel as alarm
from homeassistant.helpers.event import track_point_in_time
import homeassistant.util.dt as dt_util
from homeassistant.const import (
STATE_ALARM_DISARMED, STATE_ALARM_ARMED_HOME, STATE_ALARM_ARMED_AWAY,
@ -62,6 +64,7 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
# pylint: disable=too-many-arguments, too-many-instance-attributes
# pylint: disable=abstract-method
class ManualAlarm(alarm.AlarmControlPanel):
""" represents an alarm status within home assistant. """
@ -70,8 +73,9 @@ class ManualAlarm(alarm.AlarmControlPanel):
self._hass = hass
self._name = name
self._code = code
self._pending_time = pending_time
self._trigger_time = trigger_time
self._pending_time = datetime.timedelta(seconds=pending_time)
self._trigger_time = datetime.timedelta(seconds=trigger_time)
self._state_ts = None
self._pending_to = None
@property
@ -94,49 +98,70 @@ class ManualAlarm(alarm.AlarmControlPanel):
""" One or more characters """
return None if self._code is None else '.+'
def change_alarm_state(self, begin, end, delay=0):
def update_state(self, state, pending_to):
""" changes between state with delay """
self._state = begin
self._pending_to = end
self._state = state
self._state_ts = dt_util.utcnow()
self._pending_to = pending_to
self.update_ha_state()
time.sleep(delay)
if self._pending_to == end and begin != end:
self._state = end
self._pending_to = None
self.update_ha_state()
def alarm_disarm(self, code=None):
""" Send disarm command. """
if code == str(self._code) or self.code_format is None:
self.change_alarm_state(
STATE_ALARM_DISARMED, STATE_ALARM_DISARMED)
self.update_state(STATE_ALARM_DISARMED, None)
else:
_LOGGER.warning("Wrong code entered while disarming!")
def alarm_arm_home(self, code=None):
""" Send arm home command. """
if code == str(self._code) or self.code_format is None:
self.change_alarm_state(
STATE_ALARM_PENDING, STATE_ALARM_ARMED_HOME,
self._pending_time)
self.update_state(STATE_ALARM_PENDING, STATE_ALARM_ARMED_HOME)
def delayed_alarm_arm_home(event_time):
""" callback for delayed action """
if self._pending_to == STATE_ALARM_ARMED_HOME and \
dt_util.utcnow() - self._state_ts >= self._pending_time:
self.update_state(STATE_ALARM_ARMED_HOME, None)
track_point_in_time(
self._hass, delayed_alarm_arm_home,
dt_util.utcnow() + self._pending_time)
else:
_LOGGER.warning("Wrong code entered while arming home!")
def alarm_arm_away(self, code=None):
""" Send arm away command. """
if code == str(self._code) or self.code_format is None:
self.change_alarm_state(
STATE_ALARM_PENDING, STATE_ALARM_ARMED_AWAY,
self._pending_time)
self.update_state(STATE_ALARM_PENDING, STATE_ALARM_ARMED_AWAY)
def delayed_alarm_arm_away(event_time):
""" callback for delayed action """
if self._pending_to == STATE_ALARM_ARMED_AWAY and \
dt_util.utcnow() - self._state_ts >= self._pending_time:
self.update_state(STATE_ALARM_ARMED_AWAY, None)
track_point_in_time(
self._hass, delayed_alarm_arm_away,
dt_util.utcnow() + self._pending_time)
else:
_LOGGER.warning("Wrong code entered while arming away!")
def alarm_trigger(self, code=None):
""" Send alarm trigger command. No code needed. """
self.change_alarm_state(
STATE_ALARM_PENDING, STATE_ALARM_TRIGGERED,
self._pending_time)
if self._state == STATE_ALARM_TRIGGERED:
self.change_alarm_state(
STATE_ALARM_TRIGGERED, STATE_ALARM_DISARMED,
self._trigger_time)
self.update_state(STATE_ALARM_PENDING, STATE_ALARM_TRIGGERED)
def delayed_alarm_trigger(event_time):
""" callback for delayed action """
if self._pending_to == STATE_ALARM_TRIGGERED and \
dt_util.utcnow() - self._state_ts >= self._pending_time:
self.update_state(STATE_ALARM_TRIGGERED, STATE_ALARM_DISARMED)
def delayed_alarm_disarm(event_time):
""" callback for delayed action """
if self._pending_to == STATE_ALARM_DISARMED and \
dt_util.utcnow() - self._state_ts >= self._trigger_time:
self.update_state(STATE_ALARM_DISARMED, None)
track_point_in_time(
self._hass, delayed_alarm_disarm,
dt_util.utcnow() + self._trigger_time)
track_point_in_time(
self._hass, delayed_alarm_trigger,
dt_util.utcnow() + self._pending_time)