jeans/supervisor
jeans/supervisor
1
0
Fork 0
mirror of https://github.com/home-assistant/supervisor.git synced 2025-07-24 09:36:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix charset and timeout handling for HostControll

Browse Source
This commit is contained in:
pvizeli 2017-03-29 14:49:34 +02:00
parent 00dd19c292
commit 4fb2a88e1e
1 changed files with 25 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
hassio_api/hassio/host_controll.py
View File

@ -9,6 +9,8 @@ from .const import SOCKET_HC
_LOGGER = logging.getLogger(__name__)
TIMEOUT = 15
class HostControll(object):
"""Client for host controll."""
@ -33,21 +35,29 @@ class HostControll(object):
reader, writer = await asyncio.open_unix_connection(
SOCKET_HC, loop=self.loop)
# send
_LOGGER.info("Send '%s' to HostControll.", command)
writer.write(command.encode())
try:
# send
_LOGGER.info("Send '%s' to HostControll.", command)
writer.write("{}\n".format(command).encode())
# receive
data = await reader.readline()
response = data.decode().Upper()
_LOGGER.info("Receive from HostControll: %s.", response)
# receive
with async_timeout.timeout(TIMEOUT, loop=self.loop):
data = await reader.readline()
response = data.decode().Upper()
_LOGGER.info("Receive from HostControll: %s.", response)
if response == "OK":
return True
elif response == "ERROR":
return False
else:
return json.loads(response)
if response == "OK":
return True
elif response == "ERROR":
return False
else:
return json.loads(response)
except asyncio.TimeoutError:
_LOGGER.error("Timeout from HostControll!")
finally:
writer.close()
def info(self):
"""Return Info from host.
@ -76,7 +86,7 @@ class HostControll(object):
Return a coroutine.
"""
if version:
return self._send_command("host-update " + version)
return self._send_command("host-update {}".format(version))
return self._send_command("host-update")
def supervisor_update(self, version=None):
@ -85,5 +95,5 @@ class HostControll(object):
Return a coroutine.
"""
if version:
return self._send_command("supervisor-update " + version)
return self._send_command("supervisor-update {}".format(version))
return self._send_command("supervisor-update")