99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
"""
|
|
homeassistant.components.automation.time
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Offers time listening automation rules.
|
|
"""
|
|
import logging
|
|
|
|
from homeassistant.util import convert
|
|
import homeassistant.util.dt as dt_util
|
|
from homeassistant.helpers.event import track_time_change
|
|
|
|
CONF_HOURS = "hours"
|
|
CONF_MINUTES = "minutes"
|
|
CONF_SECONDS = "seconds"
|
|
CONF_BEFORE = "before"
|
|
CONF_AFTER = "after"
|
|
CONF_WEEKDAY = "weekday"
|
|
|
|
WEEKDAYS = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat', 'sun']
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def trigger(hass, config, action):
|
|
""" Listen for state changes based on `config`. """
|
|
if CONF_AFTER in config:
|
|
after = dt_util.parse_time_str(config[CONF_AFTER])
|
|
if after is None:
|
|
_LOGGER.error(
|
|
'Received invalid after value: %s', config[CONF_AFTER])
|
|
return False
|
|
hours, minutes, seconds = after.hour, after.minute, after.second
|
|
elif CONF_HOURS in config or CONF_MINUTES in config \
|
|
or CONF_SECONDS in config:
|
|
hours = convert(config.get(CONF_HOURS), int)
|
|
minutes = convert(config.get(CONF_MINUTES), int)
|
|
seconds = convert(config.get(CONF_SECONDS), int)
|
|
else:
|
|
_LOGGER.error('One of %s, %s, %s OR %s needs to be specified',
|
|
CONF_HOURS, CONF_MINUTES, CONF_SECONDS, CONF_AFTER)
|
|
return False
|
|
|
|
def time_automation_listener(now):
|
|
""" Listens for time changes and calls action. """
|
|
action()
|
|
|
|
track_time_change(hass, time_automation_listener,
|
|
hour=hours, minute=minutes, second=seconds)
|
|
|
|
return True
|
|
|
|
|
|
def if_action(hass, config):
|
|
""" Wraps action method with time based condition. """
|
|
before = config.get(CONF_BEFORE)
|
|
after = config.get(CONF_AFTER)
|
|
weekday = config.get(CONF_WEEKDAY)
|
|
|
|
if before is None and after is None and weekday is None:
|
|
logging.getLogger(__name__).error(
|
|
"Missing if-condition configuration key %s, %s or %s",
|
|
CONF_BEFORE, CONF_AFTER, CONF_WEEKDAY)
|
|
return None
|
|
|
|
def time_if():
|
|
""" Validate time based if-condition """
|
|
now = dt_util.now()
|
|
if before is not None:
|
|
time = dt_util.parse_time_str(before)
|
|
if time is None:
|
|
return False
|
|
|
|
before_point = now.replace(hour=time.hour, minute=time.minute)
|
|
|
|
if now > before_point:
|
|
return False
|
|
|
|
if after is not None:
|
|
time = dt_util.parse_time_str(after)
|
|
if time is None:
|
|
return False
|
|
|
|
after_point = now.replace(hour=time.hour, minute=time.minute)
|
|
|
|
if now < after_point:
|
|
return False
|
|
|
|
if weekday is not None:
|
|
now_weekday = WEEKDAYS[now.weekday()]
|
|
|
|
if isinstance(weekday, str) and weekday != now_weekday or \
|
|
now_weekday not in weekday:
|
|
return False
|
|
|
|
return True
|
|
|
|
return time_if
|