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'])