35 lines
887 B
Python
35 lines
887 B
Python
import base64
|
|
|
|
from classes.Message import Message
|
|
|
|
|
|
class ShuffleMessage(Message):
|
|
def __init__(self, name: str, cards: list[bytes], stage: str):
|
|
super().__init__()
|
|
self.message_fields['type'] = 'shuffle'
|
|
self.set_name(name)
|
|
self.set_cards(cards)
|
|
self.set_stage(stage)
|
|
|
|
def set_name(self, name: str):
|
|
self.message_fields['name'] = name
|
|
|
|
def get_name(self) -> str:
|
|
return self.message_fields['name']
|
|
|
|
def set_cards(self, cards: list[bytes]):
|
|
self.message_fields['cards'] = []
|
|
for card in cards:
|
|
self.message_fields['cards'].append(base64.b64encode(card).decode('utf-8'))
|
|
|
|
def get_cards(self) -> list[bytes]:
|
|
cards = []
|
|
for card in self.message_fields['cards']:
|
|
cards.append(base64.b64decode(card))
|
|
return cards
|
|
|
|
def set_stage(self, stage: str):
|
|
self.message_fields['stage'] = stage
|
|
|
|
def get_stage(self) -> str:
|
|
return self.message_fields['stage'] |