From 95ca11e98c47c3e8e57093c37134a6f51bcb6f30 Mon Sep 17 00:00:00 2001 From: Olivier Gayot Date: Wed, 29 Dec 2021 17:05:21 +0100 Subject: Add type hinting everywhere so we can enable strict mypy options Signed-off-by: Olivier Gayot --- swiftstory/interface/ws.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'swiftstory/interface/ws.py') 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: -- cgit v1.2.3