Add WrongAction exception and use it instaed of returning error

Signed-off-by: Olivier Gayot <olivier.gayot@sigexec.com>

1 files changed, 3 insertions, 0 deletions

diff --git a/swiftstory/SwiftStory.py b/swiftstory/SwiftStory.py
index f9799ef..b81b0c3 100644
--- a/swiftstory/SwiftStory.py
+++ b/swiftstory/SwiftStory.py
@@ -8,6 +8,7 @@ import logging
 import websockets
 
 import swiftstory.GameManager
+from swiftstory.exception import WrongAction
 from swiftstory.Client import Client
 from swiftstory.Status import error
 
@@ -62,6 +63,8 @@ def message_received_handler(client, message):
         return opcodes_map[json_msg["op"]]()
     except (KeyError, TypeError):
         return error('invalid command')
+    except WrongAction as e:
+        return error(str(e))
 
 
 async def connection_handler(websocket, path):