From 96eaa6555ae44e30b3784576a6ecd0941523691f Mon Sep 17 00:00:00 2001 From: Olivier Gayot Date: Thu, 23 Dec 2021 15:47:13 +0100 Subject: Add required annotations to make sure mypy succeeds Signed-off-by: Olivier Gayot --- swiftstory/client.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'swiftstory/client.py') diff --git a/swiftstory/client.py b/swiftstory/client.py index 3000b79..52957c2 100644 --- a/swiftstory/client.py +++ b/swiftstory/client.py @@ -1,14 +1,16 @@ import asyncio import logging +from typing import Optional import websockets +from swiftstory.game import Game from swiftstory.exception import WrongAction, UnsupportedLanguage, JoinError class Client: def __init__(self, socket, game_manager): - self.game = None + self.game: Optional[Game] = None self.game_manager = game_manager self.socket = socket @@ -75,4 +77,6 @@ class Client: def disconnect(self): if self.player is not None: + if self.game is None: + raise ValueError("Disconnect from inexistent game.") self.game.disconnect(self.player) -- cgit v1.2.3