83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
|
"""Support for restoring entity states on startup."""
|
||
|
|
import asyncio
|
||
|
|
import logging
|
||
|
|
from datetime import timedelta
|
||
|
|
|
||
|
|
from homeassistant.core import HomeAssistant, CoreState, callback
|
||
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_START
|
||
|
|
from homeassistant.components.history import get_states, last_recorder_run
|
||
|
|
from homeassistant.components.recorder import DOMAIN as _RECORDER
|
||
|
|
import homeassistant.util.dt as dt_util
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
DATA_RESTORE_CACHE = 'restore_state_cache'
|
||
|
|
_LOCK = 'restore_lock'
|
||
|
|
|
||
|
|
|
||
|
|
def _load_restore_cache(hass: HomeAssistant):
|
||
|
|
"""Load the restore cache to be used by other components."""
|
||
|
|
@callback
|
||
|
|
def remove_cache(event):
|
||
|
|
"""Remove the states cache."""
|
||
|
|
hass.data.pop(DATA_RESTORE_CACHE, None)
|
||
|
|
|
||
|
|
hass.bus.listen_once(EVENT_HOMEASSISTANT_START, remove_cache)
|
||
|
|
|
||
|
|
last_run = last_recorder_run()
|
||
|
|
|
||
|
|
if last_run is None or last_run.end is None:
|
||
|
|
_LOGGER.debug('Not creating cache - no suitable last run found: %s',
|
||
|
|
last_run)
|
||
|
|
hass.data[DATA_RESTORE_CACHE] = {}
|
||
|
|
return
|
||
|
|
|
||
|
|
last_end_time = last_run.end - timedelta(seconds=1)
|
||
|
|
# Unfortunately the recorder_run model do not return offset-aware time
|
||
|
|
last_end_time = last_end_time.replace(tzinfo=dt_util.UTC)
|
||
|
|
_LOGGER.debug("Last run: %s - %s", last_run.start, last_end_time)
|
||
|
|
|
||
|
|
states = get_states(last_end_time, run=last_run)
|
||
|
|
|
||
|
|
# Cache the states
|
||
|
|
hass.data[DATA_RESTORE_CACHE] = {
|
||
|
|
state.entity_id: state for state in states}
|
||
|
|
_LOGGER.debug('Created cache with %s', list(hass.data[DATA_RESTORE_CACHE]))
|
||
|
|
|
||
|
|
|
||
|
|
@asyncio.coroutine
|
||
|
|
def async_get_last_state(hass, entity_id: str):
|
||
|
|
"""Helper to restore state."""
|
||
|
|
if (_RECORDER not in hass.config.components or
|
||
|
|
hass.state != CoreState.starting):
|
||
|
|
return None
|
||
|
|
|
||
|
|
if DATA_RESTORE_CACHE in hass.data:
|
||
|
|
return hass.data[DATA_RESTORE_CACHE].get(entity_id)
|
||
|
|
|
||
|
|
if _LOCK not in hass.data:
|
||
|
|
hass.data[_LOCK] = asyncio.Lock(loop=hass.loop)
|
||
|
|
|
||
|
|
with (yield from hass.data[_LOCK]):
|
||
|
|
if DATA_RESTORE_CACHE not in hass.data:
|
||
|
|
yield from hass.loop.run_in_executor(
|
||
|
|
None, _load_restore_cache, hass)
|
||
|
|
|
||
|
|
return hass.data[DATA_RESTORE_CACHE].get(entity_id)
|
||
|
|
|
||
|
|
|
||
|
|
@asyncio.coroutine
|
||
|
|
def async_restore_state(entity, extract_info):
|
||
|
|
"""Helper to call entity.async_restore_state with cached info."""
|
||
|
|
if entity.hass.state != CoreState.starting:
|
||
|
|
_LOGGER.debug("Not restoring state: State is not starting: %s",
|
||
|
|
entity.hass.state)
|
||
|
|
return
|
||
|
|
|
||
|
|
state = yield from async_get_last_state(entity.hass, entity.entity_id)
|
||
|
|
|
||
|
|
if not state:
|
||
|
|
return
|
||
|
|
|
||
|
|
yield from entity.async_restore_state(**extract_info(state))
|