mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-04-20 19:27:16 +00:00
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
"""Multible tasks."""
|
|
import asyncio
|
|
from datetime import datetime
|
|
import logging
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def api_sessions_cleanup(config):
|
|
"""Create scheduler task for cleanup api sessions."""
|
|
async def _api_sessions_cleanup():
|
|
"""Cleanup old api sessions."""
|
|
now = datetime.now()
|
|
for session, until_valid in config.security_sessions.items():
|
|
if now >= until_valid:
|
|
config.security_sessions = (session, None)
|
|
|
|
return _api_sessions_cleanup
|
|
|
|
|
|
def addons_update(loop, addons):
|
|
"""Create scheduler task for auto update addons."""
|
|
async def _addons_update():
|
|
"""Check if a update is available of a addon and update it."""
|
|
tasks = []
|
|
for addon in addons.list_addons:
|
|
if not addon.is_installed or not addon.auto_update:
|
|
continue
|
|
|
|
if addon.version_installed != addon.version:
|
|
tasks.append(addon.update())
|
|
|
|
if tasks:
|
|
_LOGGER.info("Addon auto update process %d tasks", len(tasks))
|
|
await asyncio.wait(tasks, loop=loop)
|
|
|
|
return _addons_update
|
|
|
|
|
|
def hassio_update(config, supervisor, websession):
|
|
"""Create scheduler task for update of supervisor hassio."""
|
|
async def _hassio_update():
|
|
"""Check and run update of supervisor hassio."""
|
|
await config.fetch_update_infos(websession)
|
|
if config.last_hassio == supervisor.version:
|
|
return
|
|
|
|
# don't perform a update on beta/dev channel
|
|
if config.upstream_beta:
|
|
_LOGGER.warning("Ignore Hass.IO update on beta upstream!")
|
|
return
|
|
|
|
_LOGGER.info("Found new HassIO version %s.", config.last_hassio)
|
|
await supervisor.update(config.last_hassio)
|
|
|
|
return _hassio_update
|
|
|
|
|
|
def homeassistant_watchdog(loop, homeassistant):
|
|
"""Create scheduler task for montoring running state."""
|
|
async def _homeassistant_watchdog():
|
|
"""Check running state and start if they is close."""
|
|
if homeassistant.in_progress or await homeassistant.is_running():
|
|
return
|
|
|
|
loop.create_task(homeassistant.run())
|
|
|
|
return _homeassistant_watchdog
|
|
|
|
|
|
async def homeassistant_setup(config, loop, homeassistant, websession):
|
|
"""Install a homeassistant docker container."""
|
|
while True:
|
|
# read homeassistant tag and install it
|
|
if not config.last_homeassistant:
|
|
await config.fetch_update_infos(websession)
|
|
|
|
tag = config.last_homeassistant
|
|
if tag and await homeassistant.install(tag):
|
|
break
|
|
_LOGGER.warning("Error on setup HomeAssistant. Retry in 60.")
|
|
await asyncio.sleep(60, loop=loop)
|
|
|
|
# store version
|
|
_LOGGER.info("HomeAssistant docker now installed.")
|