24 lines
676 B
Python
24 lines
676 B
Python
import base64
|
|
|
|
from classes.Message import Message
|
|
|
|
|
|
class AnnouncementMessage(Message):
|
|
def __init__(self, name: str, announcement: bytes):
|
|
super().__init__()
|
|
self.message_fields['type'] = 'announcement'
|
|
self.message_fields['name'] = name
|
|
self.message_fields['announcement'] = announcement
|
|
|
|
def set_name(self, name: str):
|
|
self.message_fields['name'] = name
|
|
|
|
def get_name(self) -> str:
|
|
return self.message_fields['name']
|
|
|
|
def set_announcement(self, announcement: bytes):
|
|
self.message_fields['announcement'] = base64.b64encode(announcement).decode('utf-8')
|
|
|
|
def get_announcement(self) -> bytes:
|
|
return base64.b64decode(self.message_fields['announcement'])
|