blob: 527c9c6751e64f294c01d12216c508d3d488f66a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
import asyncio
import logging
from typing import Optional
import websockets.exceptions
from swiftstory.game import Game
from swiftstory.exception import WrongAction, UnsupportedLanguage, JoinError
class Client:
def __init__(self, socket, game_manager):
self.game: Optional[Game] = None
self.game_manager = game_manager
self.socket = socket
self.player = None
def join_game(self, game_name, lang):
if self.game is not None:
raise WrongAction('You are already in a game')
if lang is None:
lang = 'en'
try:
game = self.game_manager.join_game(game_name, lang)
except UnsupportedLanguage as e:
raise JoinError(f"unsupported language: {str(e)}") from e
# XXX self.game will be assigned by game.try_join()
return game.try_join(self)
def play_white_card(self, card_id):
if self.game is None:
raise WrongAction('You have to join a game first')
return self.game.try_play_card(self.player, card_id)
def pick_black_card(self):
if self.game is None:
raise WrongAction('You have to join a game first')
return self.game.try_become_judge(self.player)
def collect_cards(self):
if self.game is None:
raise WrongAction('You have to join a game first')
return self.game.try_collect_cards(self.player)
def designate_card(self, card_id):
if self.game is None:
raise WrongAction('You have to join a game first')
return self.game.try_designate_card(self.player, card_id)
def view_player_cards(self):
if self.game is None:
raise WrongAction('You have to join a game first')
return self.game.try_view_player_cards(self.player)
def view_played_cards(self):
if self.game is None:
raise WrongAction('You have to join a game first')
return self.game.try_view_played_cards(self.player)
def view_black_card(self):
if self.game is None:
raise WrongAction('You have to join a game first')
return self.game.try_view_black_card(self.player)
def register_notification(self, message):
async def f():
try:
await self.socket.send(message)
except websockets.exceptions.ConnectionClosed:
logging.warning("Recipient has disconnected.")
asyncio.create_task(f())
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)
|