Sneed-Group/pypush-plus-plus
2
0
Fork 0
mirror of https://github.com/Sneed-Group/pypush-plus-plus synced 2025-01-09 17:33:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

dataclasses and test effects

Browse source
This commit is contained in:
JJTech0130 2023-08-14 09:49:11 -04:00
parent 1d74e568cf
commit 10a5771791
No known key found for this signature in database
GPG key ID: 23C92EBCCF8F93D6
1 changed files with 13 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
imessage.py
View file

@ -132,13 +132,14 @@ class Attachment:
def __repr__(self): def __repr__(self):
return f'<Attachment name="{self.name}" type="{self.mime_type}">' return f'<Attachment name="{self.name}" type="{self.mime_type}">'
@dataclass
class Message: class Message:
def __init__(self, text: str, sender: str, participants: list[str], id: uuid.UUID, _raw: dict, _compressed: bool = True): text: str
self.text = text sender: str
self.sender = sender participants: list[str]
self.id = id id: uuid.UUID
self._raw = _raw _raw: dict
self._compressed = _compressed _compressed: bool = True
def from_raw(message: bytes, sender: str | None = None) -> "Message": def from_raw(message: bytes, sender: str | None = None) -> "Message":
"""Create a `Message` from raw message bytes""" """Create a `Message` from raw message bytes"""
@ -148,6 +149,7 @@ class Message:
def __str__(): def __str__():
raise NotImplementedError() raise NotImplementedError()
@dataclass
class SMSReflectedMessage(Message): class SMSReflectedMessage(Message):
def from_raw(message: bytes, sender: str | None = None) -> "SMSReflectedMessage": def from_raw(message: bytes, sender: str | None = None) -> "SMSReflectedMessage":
"""Create a `SMSIncomingMessage` from raw message bytes""" """Create a `SMSIncomingMessage` from raw message bytes"""
@ -173,6 +175,7 @@ class SMSReflectedMessage(Message):
def __str__(self): def __str__(self):
return f"[SMS {self.sender}] '{self.text}'" return f"[SMS {self.sender}] '{self.text}'"
@dataclass
class SMSIncomingMessage(Message): class SMSIncomingMessage(Message):
def from_raw(message: bytes, sender: str | None = None) -> "SMSIncomingMessage": def from_raw(message: bytes, sender: str | None = None) -> "SMSIncomingMessage":
"""Create a `SMSIncomingMessage` from raw message bytes""" """Create a `SMSIncomingMessage` from raw message bytes"""
@ -200,7 +203,10 @@ class SMSIncomingMessage(Message):
def __str__(self): def __str__(self):
return f"[SMS {self.sender}] '{self.text}'" return f"[SMS {self.sender}] '{self.text}'"
@dataclass
class iMessage(Message): class iMessage(Message):
effect: str | None = None
def from_raw(message: bytes, sender: str | None = None) -> "iMessage": def from_raw(message: bytes, sender: str | None = None) -> "iMessage":
"""Create a `iMessage` from raw message bytes""" """Create a `iMessage` from raw message bytes"""
@ -220,6 +226,7 @@ class iMessage(Message):
id=uuid.UUID(message["r"]), id=uuid.UUID(message["r"]),
_raw=message, _raw=message,
_compressed=compressed, _compressed=compressed,
effect=message["iid"] if "iid" in message else None,
) )
def __str__(self): def __str__(self):