diff options
Diffstat (limited to 'swiftstory/client.py')
-rw-r--r-- | swiftstory/client.py | 6 |
1 files changed, 5 insertions, 1 deletions
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) |