mirror of
https://github.com/home-assistant/core
synced 2024-09-09 12:51:22 +02:00
463 lines
15 KiB
Python
463 lines
15 KiB
Python
"""
|
|
homeassistant
|
|
~~~~~~~~~~~~~
|
|
|
|
Home Assistant is a Home Automation framework for observing the state
|
|
of objects and react to changes.
|
|
"""
|
|
|
|
import time
|
|
import logging
|
|
import threading
|
|
from collections import defaultdict, namedtuple
|
|
import datetime as dt
|
|
|
|
import homeassistant.util as util
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
MATCH_ALL = '*'
|
|
|
|
DOMAIN = "homeassistant"
|
|
|
|
STATE_ON = "on"
|
|
STATE_OFF = "off"
|
|
STATE_NOT_HOME = 'device_not_home'
|
|
STATE_HOME = 'device_home'
|
|
|
|
SERVICE_TURN_ON = "turn_on"
|
|
SERVICE_TURN_OFF = "turn_off"
|
|
SERVICE_HOMEASSISTANT_STOP = "stop"
|
|
|
|
EVENT_HOMEASSISTANT_START = "homeassistant.start"
|
|
EVENT_STATE_CHANGED = "state_changed"
|
|
EVENT_TIME_CHANGED = "time_changed"
|
|
|
|
TIMER_INTERVAL = 10 # seconds
|
|
|
|
# We want to be able to fire every time a minute starts (seconds=0).
|
|
# We want this so other modules can use that to make sure they fire
|
|
# every minute.
|
|
assert 60 % TIMER_INTERVAL == 0, "60 % TIMER_INTERVAL should be 0!"
|
|
|
|
|
|
def start_home_assistant(bus):
|
|
""" Start home assistant. """
|
|
request_shutdown = threading.Event()
|
|
|
|
bus.register_service(DOMAIN, SERVICE_HOMEASSISTANT_STOP,
|
|
lambda service: request_shutdown.set())
|
|
|
|
Timer(bus)
|
|
|
|
bus.fire_event(EVENT_HOMEASSISTANT_START)
|
|
|
|
while not request_shutdown.isSet():
|
|
try:
|
|
time.sleep(1)
|
|
|
|
except KeyboardInterrupt:
|
|
break
|
|
|
|
|
|
def _process_match_param(parameter):
|
|
""" Wraps parameter in a list if it is not one and returns it. """
|
|
if parameter is None:
|
|
return MATCH_ALL
|
|
elif isinstance(parameter, list):
|
|
return parameter
|
|
else:
|
|
return [parameter]
|
|
|
|
|
|
def _matcher(subject, pattern):
|
|
""" Returns True if subject matches the pattern.
|
|
|
|
Pattern is either a list of allowed subjects or a `MATCH_ALL`.
|
|
"""
|
|
return MATCH_ALL == pattern or subject in pattern
|
|
|
|
|
|
def split_state_category(category):
|
|
""" Splits a state category into domain, object_id. """
|
|
return category.split(".", 1)
|
|
|
|
|
|
def filter_categories(categories, domain_filter=None, strip_domain=False):
|
|
""" Filter a list of categories based on domain. Setting strip_domain
|
|
will only return the object_ids. """
|
|
return [
|
|
split_state_category(cat)[1] if strip_domain else cat
|
|
for cat in categories if
|
|
not domain_filter or cat.startswith(domain_filter)
|
|
]
|
|
|
|
|
|
def track_state_change(bus, category, action, from_state=None, to_state=None):
|
|
""" Helper method to track specific state changes. """
|
|
from_state = _process_match_param(from_state)
|
|
to_state = _process_match_param(to_state)
|
|
|
|
def listener(event):
|
|
""" State change listener that listens for specific state changes. """
|
|
if category == event.data['category'] and \
|
|
_matcher(event.data['old_state'].state, from_state) and \
|
|
_matcher(event.data['new_state'].state, to_state):
|
|
|
|
action(event.data['category'],
|
|
event.data['old_state'],
|
|
event.data['new_state'])
|
|
|
|
bus.listen_event(EVENT_STATE_CHANGED, listener)
|
|
|
|
|
|
# pylint: disable=too-many-arguments
|
|
def track_time_change(bus, action,
|
|
year=None, month=None, day=None,
|
|
hour=None, minute=None, second=None,
|
|
point_in_time=None, listen_once=False):
|
|
""" Adds a listener that will listen for a specified or matching time. """
|
|
year, month = _process_match_param(year), _process_match_param(month)
|
|
day = _process_match_param(day)
|
|
|
|
hour, minute = _process_match_param(hour), _process_match_param(minute)
|
|
second = _process_match_param(second)
|
|
|
|
def listener(event):
|
|
""" Listens for matching time_changed events. """
|
|
now = event.data['now']
|
|
|
|
if (point_in_time and now > point_in_time) or \
|
|
(not point_in_time and
|
|
_matcher(now.year, year) and
|
|
_matcher(now.month, month) and
|
|
_matcher(now.day, day) and
|
|
_matcher(now.hour, hour) and
|
|
_matcher(now.minute, minute) and
|
|
_matcher(now.second, second)):
|
|
|
|
# point_in_time are exact points in time
|
|
# so we always remove it after fire
|
|
if listen_once or point_in_time:
|
|
event.bus.remove_event_listener(EVENT_TIME_CHANGED, listener)
|
|
|
|
action(now)
|
|
|
|
bus.listen_event(EVENT_TIME_CHANGED, listener)
|
|
|
|
ServiceCall = namedtuple("ServiceCall", ["bus", "domain", "service", "data"])
|
|
Event = namedtuple("Event", ["bus", "event_type", "data"])
|
|
|
|
|
|
class Bus(object):
|
|
""" Class that allows different components to communicate via services
|
|
and events.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self._event_listeners = {}
|
|
self._services = {}
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
@property
|
|
def services(self):
|
|
""" Dict with per domain a list of available services. """
|
|
return {domain: self._services[domain].keys()
|
|
for domain in self._services}
|
|
|
|
@property
|
|
def event_listeners(self):
|
|
""" Dict with events that is being listened for and the number
|
|
of listeners.
|
|
"""
|
|
return {key: len(self._event_listeners[key])
|
|
for key in self._event_listeners}
|
|
|
|
def call_service(self, domain, service, service_data=None):
|
|
""" Calls a service. """
|
|
|
|
try:
|
|
self._services[domain][service]
|
|
except KeyError:
|
|
# Domain or Service does not exist
|
|
raise ServiceDoesNotExistException(
|
|
"Service does not exist: {}/{}".format(domain, service))
|
|
|
|
if not service_data:
|
|
service_data = {}
|
|
|
|
def run():
|
|
""" Executes a service. """
|
|
service_call = ServiceCall(self, domain, service, service_data)
|
|
|
|
try:
|
|
self._services[domain][service](service_call)
|
|
except Exception: # pylint: disable=broad-except
|
|
self.logger.exception("Bus:Exception in service {}/{}".format(
|
|
domain, service))
|
|
|
|
# We dont want the eventbus to be blocking - run in a thread.
|
|
threading.Thread(target=run).start()
|
|
|
|
def register_service(self, domain, service, service_callback):
|
|
""" Register a service. """
|
|
try:
|
|
self._services[domain][service] = service_callback
|
|
except KeyError:
|
|
# Domain does not exist yet
|
|
self._services[domain] = {service: service_callback}
|
|
|
|
def fire_event(self, event_type, event_data=None):
|
|
""" Fire an event. """
|
|
|
|
# Copy the list of the current listeners because some listeners
|
|
# choose to remove themselves as a listener while being executed
|
|
# which causes the iterator to be confused.
|
|
get = self._event_listeners.get
|
|
listeners = get(MATCH_ALL, []) + get(event_type, [])
|
|
|
|
if not listeners:
|
|
return
|
|
|
|
event_data = event_data or {}
|
|
|
|
self.logger.info("Bus:Event {}: {}".format(
|
|
event_type, event_data))
|
|
|
|
def run():
|
|
""" Fire listeners for event. """
|
|
event = Event(self, event_type, event_data)
|
|
|
|
for listener in listeners:
|
|
try:
|
|
listener(event)
|
|
|
|
except Exception: # pylint: disable=broad-except
|
|
self.logger.exception("Bus:Exception in event listener")
|
|
|
|
# We dont want the bus to be blocking - run in a thread.
|
|
threading.Thread(target=run).start()
|
|
|
|
def listen_event(self, event_type, listener):
|
|
""" Listen for all events or events of a specific type.
|
|
|
|
To listen to all events specify the constant ``MATCH_ALL``
|
|
as event_type.
|
|
"""
|
|
try:
|
|
self._event_listeners[event_type].append(listener)
|
|
except KeyError: # event_type did not exist
|
|
self._event_listeners[event_type] = [listener]
|
|
|
|
def listen_once_event(self, event_type, listener):
|
|
""" Listen once for event of a specific type.
|
|
|
|
To listen to all events specify the constant ``MATCH_ALL``
|
|
as event_type.
|
|
|
|
Note: at the moment it is impossible to remove a one time listener.
|
|
"""
|
|
|
|
def onetime_listener(event):
|
|
""" Removes listener from eventbus and then fires listener. """
|
|
self.remove_event_listener(event_type, onetime_listener)
|
|
|
|
listener(event)
|
|
|
|
self.listen_event(event_type, onetime_listener)
|
|
|
|
def remove_event_listener(self, event_type, listener):
|
|
""" Removes a listener of a specific event_type. """
|
|
try:
|
|
self._event_listeners[event_type].remove(listener)
|
|
|
|
if len(self._event_listeners[event_type]) == 0:
|
|
del self._event_listeners[event_type]
|
|
|
|
except (KeyError, ValueError):
|
|
pass
|
|
|
|
|
|
class State(object):
|
|
""" Object to represent a state within the state machine. """
|
|
|
|
def __init__(self, state, attributes=None, last_changed=None):
|
|
self.state = state
|
|
self.attributes = attributes or {}
|
|
last_changed = last_changed or dt.datetime.now()
|
|
|
|
# Strip microsecond from last_changed else we cannot guarantee
|
|
# state == State.from_json_dict(state.to_json_dict())
|
|
# This behavior occurs because to_json_dict strips microseconds
|
|
if last_changed.microsecond:
|
|
self.last_changed = last_changed - dt.timedelta(
|
|
microseconds=last_changed.microsecond)
|
|
else:
|
|
self.last_changed = last_changed
|
|
|
|
def to_json_dict(self, category=None):
|
|
""" Converts State to a dict to be used within JSON.
|
|
Ensures: state == State.from_json_dict(state.to_json_dict()) """
|
|
|
|
json_dict = {'state': self.state,
|
|
'attributes': self.attributes,
|
|
'last_changed': util.datetime_to_str(self.last_changed)}
|
|
|
|
if category:
|
|
json_dict['category'] = category
|
|
|
|
return json_dict
|
|
|
|
def copy(self):
|
|
""" Creates a copy of itself. """
|
|
return State(self.state, dict(self.attributes), self.last_changed)
|
|
|
|
@staticmethod
|
|
def from_json_dict(json_dict):
|
|
""" Static method to create a state from a dict.
|
|
Ensures: state == State.from_json_dict(state.to_json_dict()) """
|
|
|
|
try:
|
|
last_changed = json_dict.get('last_changed')
|
|
|
|
if last_changed:
|
|
last_changed = util.str_to_datetime(last_changed)
|
|
|
|
return State(json_dict['state'],
|
|
json_dict.get('attributes'),
|
|
last_changed)
|
|
except KeyError: # if key 'state' did not exist
|
|
return None
|
|
|
|
def __repr__(self):
|
|
return "{}({}, {})".format(
|
|
self.state, self.attributes,
|
|
util.datetime_to_str(self.last_changed))
|
|
|
|
|
|
class StateMachine(object):
|
|
""" Helper class that tracks the state of different categories. """
|
|
|
|
def __init__(self, bus):
|
|
self.states = dict()
|
|
self.bus = bus
|
|
self.lock = threading.Lock()
|
|
|
|
@property
|
|
def categories(self):
|
|
""" List of categories which states are being tracked. """
|
|
return self.states.keys()
|
|
|
|
def remove_category(self, category):
|
|
""" Removes a category from the state machine.
|
|
|
|
Returns boolean to indicate if a category was removed. """
|
|
try:
|
|
del self.states[category]
|
|
|
|
return True
|
|
|
|
except KeyError:
|
|
# if category does not exist
|
|
return False
|
|
|
|
def set_state(self, category, new_state, attributes=None):
|
|
""" Set the state of a category, add category if it does not exist.
|
|
|
|
Attributes is an optional dict to specify attributes of this state. """
|
|
|
|
attributes = attributes or {}
|
|
|
|
self.lock.acquire()
|
|
|
|
# Add category if it does not exist
|
|
if category not in self.states:
|
|
self.states[category] = State(new_state, attributes)
|
|
|
|
# Change state and fire listeners
|
|
else:
|
|
old_state = self.states[category]
|
|
|
|
if old_state.state != new_state or \
|
|
old_state.attributes != attributes:
|
|
|
|
self.states[category] = State(new_state, attributes)
|
|
|
|
self.bus.fire_event(EVENT_STATE_CHANGED,
|
|
{'category': category,
|
|
'old_state': old_state,
|
|
'new_state': self.states[category]})
|
|
|
|
self.lock.release()
|
|
|
|
def get_state(self, category):
|
|
""" Returns a dict (state, last_changed, attributes) describing
|
|
the state of the specified category. """
|
|
try:
|
|
# Make a copy so people won't mutate the state
|
|
return self.states[category].copy()
|
|
|
|
except KeyError:
|
|
# If category does not exist
|
|
return None
|
|
|
|
def is_state(self, category, state):
|
|
""" Returns True if category exists and is specified state. """
|
|
cur_state = self.get_state(category)
|
|
|
|
return cur_state and cur_state.state == state
|
|
|
|
|
|
class Timer(threading.Thread):
|
|
""" Timer will sent out an event every TIMER_INTERVAL seconds. """
|
|
|
|
def __init__(self, bus):
|
|
threading.Thread.__init__(self)
|
|
|
|
self.daemon = True
|
|
self.bus = bus
|
|
|
|
bus.listen_once_event(EVENT_HOMEASSISTANT_START,
|
|
lambda event: self.start())
|
|
|
|
def run(self):
|
|
""" Start the timer. """
|
|
|
|
logging.getLogger(__name__).info("Timer:starting")
|
|
|
|
last_fired_on_second = -1
|
|
|
|
while True:
|
|
now = dt.datetime.now()
|
|
|
|
# First check checks if we are not on a second matching the
|
|
# timer interval. Second check checks if we did not already fire
|
|
# this interval.
|
|
if now.second % TIMER_INTERVAL or \
|
|
now.second == last_fired_on_second:
|
|
|
|
# Sleep till it is the next time that we have to fire an event.
|
|
# Aim for halfway through the second that fits TIMER_INTERVAL.
|
|
# If TIMER_INTERVAL is 10 fire at .5, 10.5, 20.5, etc seconds.
|
|
# This will yield the best results because time.sleep() is not
|
|
# 100% accurate because of non-realtime OS's
|
|
slp_seconds = TIMER_INTERVAL - now.second % TIMER_INTERVAL + \
|
|
.5 - now.microsecond/1000000.0
|
|
|
|
time.sleep(slp_seconds)
|
|
|
|
now = dt.datetime.now()
|
|
|
|
last_fired_on_second = now.second
|
|
|
|
self.bus.fire_event(EVENT_TIME_CHANGED,
|
|
{'now': now})
|
|
|
|
|
|
class HomeAssistantException(Exception):
|
|
""" General Home Assistant exception occured. """
|
|
|
|
|
|
class ServiceDoesNotExistException(HomeAssistantException):
|
|
""" A service has been referenced that deos not exist. """
|