diff options
| -rw-r--r-- | swiftstory/__main__.py | 5 | ||||
| -rw-r--r-- | swiftstory/client.py | 2 |
2 files changed, 4 insertions, 3 deletions
diff --git a/swiftstory/__main__.py b/swiftstory/__main__.py index 00c6e39..99176a0 100644 --- a/swiftstory/__main__.py +++ b/swiftstory/__main__.py @@ -7,7 +7,8 @@ import asyncio import contextlib import json import logging -import websockets +import websockets.server +import websockets.exceptions import swiftstory.game_manager from swiftstory.exception import WrongAction, JoinError @@ -90,7 +91,7 @@ def main(): logging.basicConfig(level=logging.INFO) - start_server = websockets.serve(connection_handler, args['listen'], args['port']) + start_server = websockets.server.serve(connection_handler, args['listen'], args['port']) asyncio.get_event_loop().run_until_complete(start_server) asyncio.get_event_loop().run_forever() diff --git a/swiftstory/client.py b/swiftstory/client.py index 52957c2..527c9c6 100644 --- a/swiftstory/client.py +++ b/swiftstory/client.py @@ -2,7 +2,7 @@ import asyncio import logging from typing import Optional -import websockets +import websockets.exceptions from swiftstory.game import Game from swiftstory.exception import WrongAction, UnsupportedLanguage, JoinError |