Mirrors/chatmail-core
1
0
Fork 0
mirror of https://github.com/chatmail/core.git synced 2026-05-08 09:26:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

small refactor

Browse Source
This commit is contained in:
holger krekel
2024-12-16 20:01:37 +01:00
parent 4747d7abc2
commit 90f49a04e8
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
python/tests/test_5_simulation.py
View File

@@ -42,18 +42,24 @@ class Message:
nums = repr_peers(self.recipients) nums = repr_peers(self.recipients)
return f"<{self.__class__.__name__} {self.sender.id}->{nums} {self.payload}" return f"<{self.__class__.__name__} {self.sender.id}->{nums} {self.payload}"
def send(self):
for peer in self.sender.members:
peer.from2mailbox.setdefault(self.sender, []).append(self)
class AddMemberMessage(Message): class AddMemberMessage(Message):
def receive_imf(self, peer): def receive_imf(self, peer):
peer.members.add(self.payload["newmember"]) peer.members.add(self.payload["newmember"])
peer.members.update(self.recipients) peer.members.update(self.recipients)
class DelMemberMessage(Message): class DelMemberMessage(Message):
def receive_imf(self, peer): def receive_imf(self, peer):
member = self.payload["member"] member = self.payload["member"]
if member in peer.members: if member in peer.members:
peer.members.remove(member) peer.members.remove(member)
class ChatMessage(Message): class ChatMessage(Message):
def receive_imf(self, peer): def receive_imf(self, peer):
peer.members.update(self.recipients) peer.members.update(self.recipients)
@@ -87,30 +93,26 @@ class Peer:
def add_member(self, newmember): def add_member(self, newmember):
self.members.add(newmember) self.members.add(newmember)
message = AddMemberMessage(self, newmember=newmember) message = AddMemberMessage(self, newmember=newmember)
self.queue_message(message) message.send()
def del_member(self, member): def del_member(self, member):
message = DelMemberMessage(self, member=member) message = DelMemberMessage(self, member=member)
self.queue_message(message) message.send()
self.members.remove(member) self.members.remove(member)
def send_chatmessage(self): def send_chatmessage(self):
message = ChatMessage(self) message = ChatMessage(self)
self.queue_message(message) message.send()
def queue_message(self, message):
for peer in self.members:
peer.from2mailbox.setdefault(self, []).append(message)
### processing group membership message ### processing group membership message
def drain_mailbox(peer, from_peer): def drain_mailbox(peer, from_peer):
for msg in peer.from2mailbox.get(from_peer, []): for msg in peer.from2mailbox.get(from_peer, []):
msg.receive_imf(peer) msg.receive_imf(peer)
### Tests ### Tests