Update host.py

This commit is contained in:
Pascal Vizeli 2018-04-26 11:36:51 +02:00 committed by GitHub
parent f64da6a547
commit 1c2e0e5749
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -29,13 +29,13 @@ class APIHost(CoreSysAttributes):
"""Return host information.""" """Return host information."""
return { return {
ATTR_TYPE: None, ATTR_TYPE: None,
ATTR_CHASSIS: self.sys_host.local.chassis, ATTR_CHASSIS: self.sys_host.info.chassis,
ATTR_VERSION: None, ATTR_VERSION: None,
ATTR_LAST_VERSION: None, ATTR_LAST_VERSION: None,
ATTR_FEATURES: self.sys_host.supperted_features, ATTR_FEATURES: self.sys_host.supperted_features,
ATTR_HOSTNAME: self.sys_host.local.hostname, ATTR_HOSTNAME: self.sys_host.info.hostname,
ATTR_OPERATING_SYSTEM: self.sys_host.local.operating_system, ATTR_OPERATING_SYSTEM: self.sys_host.info.operating_system,
ATTR_KERNEL: self.sys_host.local.kernel, ATTR_KERNEL: self.sys_host.info.kernel,
} }
@api_process @api_process
@ -45,31 +45,26 @@ class APIHost(CoreSysAttributes):
# hostname # hostname
if ATTR_HOSTNAME in body: if ATTR_HOSTNAME in body:
await self.sys_host.local.set_hostname(body[ATTR_HOSTNAME]) await self.sys_host.control.set_hostname(body[ATTR_HOSTNAME])
@api_process @api_process
def reboot(self, request): def reboot(self, request):
"""Reboot host.""" """Reboot host."""
return self.sys_host.power.reboot() return self.sys_host.control.reboot()
@api_process @api_process
def shutdown(self, request): def shutdown(self, request):
"""Poweroff host.""" """Poweroff host."""
return self.sys_host.power.shutdown() return self.sys_host.control.shutdown()
@api_process @api_process
def reload(self, request): def reload(self, request):
"""Reload host data.""" """Reload host data."""
return self._host_control.load() return self.sys_host.load()
@api_process @api_process
async def update(self, request): async def update(self, request):
"""Update host OS.""" """Update host OS."""
body = await api_validate(SCHEMA_VERSION, request) pass
version = body.get(ATTR_VERSION, self._host_control.last_version) #body = await api_validate(SCHEMA_VERSION, request)
#version = body.get(ATTR_VERSION, self.sys_host.last_version)
if version == self._host_control.version:
raise RuntimeError(f"Version {version} is already in use")
return await asyncio.shield(
self._host_control.update(version=version))