Handle both types of incoming websocket data

girlpunk · Mar 17, 2021 · 673ac49 · 673ac49
1 parent 5d6994e
commit 673ac49
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/app/YtManagerApp/consumers.py b/app/YtManagerApp/consumers.py
@@ -25,7 +25,10 @@ class EventConsumer(WebsocketConsumer):
     http_user = True
 
     def receive(self, text=None, bytes=None, **kwargs):
-        text_data_json = json.loads(text)
+        if text:
+            text_data_json = json.loads(text)
+        elif bytes:
+            text_data_json = json.loads(bytes)
         request = text_data_json['request']
 
         if request == "jobs":