From c54514b283c47f774b0b6e9e22449ee6876ffc51 Mon Sep 17 00:00:00 2001
From: Olivier Gayot <olivier.gayot@sigexec.com>
Date: Sun, 13 Sep 2020 00:16:32 +0200
Subject: Handle exception when JSON received is not an object

Signed-off-by: Olivier Gayot <olivier.gayot@sigexec.com>
---
 swiftstory/SwiftStory.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'swiftstory')

diff --git a/swiftstory/SwiftStory.py b/swiftstory/SwiftStory.py
index 770e347..dc55e2a 100644
--- a/swiftstory/SwiftStory.py
+++ b/swiftstory/SwiftStory.py
@@ -20,7 +20,11 @@ def message_received_handler(client, message):
     except json.JSONDecodeError:
         return error('badly formatted json')
 
-    op = json_msg.get('op')
+    try:
+        op = json_msg['op']
+    except (KeyError, TypeError):
+        op = None
+
     if op == 'join_game':
         try:
             game_name = json_msg['game_name']
-- 
cgit v1.2.3