Add type hinting everywhere so we can enable strict mypy options

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

1 files changed, 6 insertions, 5 deletions

diff --git a/swiftstory/interface/ws.py b/swiftstory/interface/ws.py
index 2485b4f..a86c730 100644
--- a/swiftstory/interface/ws.py
+++ b/swiftstory/interface/ws.py
@@ -1,6 +1,7 @@
 import contextlib
 import json
 import logging
+from typing import Union
 
 import websockets.exceptions
 from websockets.server import WebSocketServerProtocol
@@ -16,7 +17,7 @@ class WebsocketsInterface:
     def __init__(self, game_manager: GameManager):
         self.game_manager = game_manager
 
-    async def connection_handler(self, websocket: WebSocketServerProtocol, path: str):
+    async def connection_handler(self, websocket: WebSocketServerProtocol, path: str) -> None:
         client = Client(websocket, self.game_manager)
 
         with contextlib.suppress(websockets.exceptions.ConnectionClosed):
@@ -25,8 +26,8 @@ class WebsocketsInterface:
 
         client.disconnect()
 
-    def message_received_handler(self, client, message):
-        def join_game():
+    def message_received_handler(self, client: Client, message: Union[bytes, str]) -> str:
+        def join_game() -> str:
             try:
                 game_name = json_msg['game_name']
             except KeyError:
@@ -35,7 +36,7 @@ class WebsocketsInterface:
                 lang = json_msg.get('lang')
                 return client.join_game(game_name, lang)
 
-        def designate_card():
+        def designate_card() -> str:
             card_id = None
             try:
                 card_id = int(json_msg['card_id'])
@@ -44,7 +45,7 @@ class WebsocketsInterface:
             finally:
                 return client.designate_card(card_id)
 
-        def play_white_card():
+        def play_white_card() -> str:
             try:
                 card_id = int(json_msg['card_id'])
             except KeyError: