martin/DecentraSanta
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Looks like it's working, at least in my test setup

Browse Source
This commit is contained in:
Martin Asprusten
2025-04-18 03:08:54 +02:00
parent e8e1fce791
commit d0b99c47a4
6 changed files with 388 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files
classes/MessageTypes/Announcement.py
+14 -7
View File
@@ -4,11 +4,12 @@ from classes.Message import Message
class AnnouncementMessage(Message):
def __init__(self, name: str, announcement: bytes):
def __init__(self, name: str, encrypted_announcement: bytes, announcement_hash: bytes):
super().__init__()
self.message_fields['type'] = 'announcement'
self.message_fields['name'] = name
self.message_fields['announcement'] = announcement
self.set_name(name)
self.set_encrypted_announcement(encrypted_announcement)
self.set_announcement_hash(announcement_hash)
def set_name(self, name: str):
self.message_fields['name'] = name
@@ -16,8 +17,14 @@ class AnnouncementMessage(Message):
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 set_encrypted_announcement(self, announcement: bytes):
self.message_fields['encrypted_announcement'] = base64.b64encode(announcement).decode('utf-8')
def get_announcement(self) -> bytes:
return base64.b64decode(self.message_fields['announcement'])
def get_encrypted_announcement(self) -> bytes:
return base64.b64decode(self.message_fields['encrypted_announcement'])
def set_announcement_hash(self, announcement_hash: bytes):
self.message_fields['announcement_hash'] = base64.b64encode(announcement_hash).decode('utf-8')
def get_announcement_hash(self) -> bytes:
return base64.b64decode(self.message_fields['announcement_hash'])