fixHEADmaster

1 files changed, 9 insertions, 3 deletions

diff --git a/main.py b/main.py
index 4534d62..74006a0 100755
--- a/main.py
+++ b/main.py
@@ -38,7 +38,7 @@ async def handle_client(reader, writer):
         except Exception:
             break
 
-        data = None
+        data = 'unknown'
         if request == 'on':
             await relay_set(ON)
             logger.info('set on')
@@ -54,9 +54,15 @@ async def handle_client(reader, writer):
             data = 'on' if status is True else 'off'
 
         writer.write((data + '\r\n').encode('utf-8'))
-        await writer.drain()
+        try:
+            await writer.drain()
+        except ConnectionError:
+            break
 
-    writer.close()
+    try:
+        writer.close()
+    except ConnectionError:
+        pass
 
 
 async def run_server(host, port):