From 2abf9fd7f063baa70101eb5a5a4c6303a0e45e37 Mon Sep 17 00:00:00 2001
From: Sunpy <sunpy@osufx.com>
Date: Thu, 1 Aug 2019 19:06:01 +0200
Subject: [PATCH] Only allow bytes thru websocket

---
 handlers/mainHandler.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/handlers/mainHandler.py b/handlers/mainHandler.py
index dcf5681..08a595f 100644
--- a/handlers/mainHandler.py
+++ b/handlers/mainHandler.py
@@ -134,8 +134,10 @@ async def handle(ws, path):
             await subscribe_flight(struct.pack(b"<I?", int(path), True))
         while not cli.ws.closed:
             async for data in cli.ws:
-                #data = data.encode() # For some reason or another... websockets module returns a string instead of bytes?
-                await query(data)
+                if type(data) != bytes:
+                    await send_error(b"Bytes are required")
+                else:
+                    await query(data)
             
             await asyncio.sleep(.1)
     finally: