From 7c6bf96f6fd3de56cc14155c11b4840a9f4478dc Mon Sep 17 00:00:00 2001 From: Pascal Vizeli Date: Thu, 3 May 2018 01:00:13 +0200 Subject: [PATCH] shield host functions --- hassio/api/host.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/hassio/api/host.py b/hassio/api/host.py index fdf8e2d45..94991a46b 100644 --- a/hassio/api/host.py +++ b/hassio/api/host.py @@ -1,4 +1,5 @@ """Init file for HassIO host rest api.""" +import asyncio import logging import voluptuous as vol @@ -45,22 +46,23 @@ class APIHost(CoreSysAttributes): # hostname if ATTR_HOSTNAME in body: - await self.sys_host.control.set_hostname(body[ATTR_HOSTNAME]) + await asyncio.shield( + self.sys_host.control.set_hostname(body[ATTR_HOSTNAME])) @api_process def reboot(self, request): """Reboot host.""" - return self.sys_host.control.reboot() + return asyncio.shield(self.sys_host.control.reboot()) @api_process def shutdown(self, request): """Poweroff host.""" - return self.sys_host.control.shutdown() + return asyncio.shield(self.sys_host.control.shutdown()) @api_process def reload(self, request): """Reload host data.""" - return self.sys_host.reload() + return asyncio.shield(self.sys_host.reload()) @api_process async def update(self, request):