2017-04-13 23:09:46 +02:00

62 lines
1.6 KiB
Python

"""Init file for HassIO host rest api."""
import logging
import voluptuous as vol
from .util import api_process_hostcontroll, api_process, api_validate
from ..const import ATTR_VERSION
_LOGGER = logging.getLogger(__name__)
UNKNOWN = 'unknown'
SCHEMA_VERSION = vol.Schema({
vol.Optional(ATTR_VERSION): vol.Coerce(str),
})
class APIHost(object):
"""Handle rest api for host functions."""
def __init__(self, config, loop, host_controll):
"""Initialize host rest api part."""
self.config = config
self.loop = loop
self.host_controll = host_controll
@api_process
async def info(self, request):
"""Return host information."""
if not self.host_controll.active:
info = {
'os': UNKNOWN,
'version': UNKNOWN,
'current': UNKNOWN,
'level': 0,
'hostname': UNKNOWN,
}
return info
return await self.host_controll.info()
@api_process_hostcontroll
def reboot(self, request):
"""Reboot host."""
return self.host_controll.reboot()
@api_process_hostcontroll
def shutdown(self, request):
"""Poweroff host."""
return self.host_controll.shutdown()
@api_process_hostcontroll
async def update(self, request):
"""Update host OS."""
body = await api_validate(SCHEMA_VERSION, request)
version = body.get(ATTR_VERSION)
if version == self.host_controll.version:
raise RuntimeError("Version is already in use")
return await self.host_controll.host_update(version=version)