71 lines
1.4 KiB
Python
71 lines
1.4 KiB
Python
from dataclasses import dataclass
|
|
from typing import Optional, List
|
|
|
|
from .client import Client
|
|
from .room import Room
|
|
from .content import ContentBase
|
|
|
|
|
|
@dataclass
|
|
class EventBase:
|
|
client: Client
|
|
content: ContentBase
|
|
type: str
|
|
sender: str
|
|
|
|
@classmethod
|
|
def from_dict(cls, client: Client, event_dict: dict):
|
|
from .content import content_dispatcher
|
|
if event_dict['type'] == 'm.room.message':
|
|
content_class = content_dispatcher[event_dict['content']['msgtype']]
|
|
else:
|
|
content_class = content_dispatcher[event_dict['type']]
|
|
|
|
content_dict = event_dict['content']
|
|
del event_dict['content']
|
|
|
|
return cls(
|
|
client=client,
|
|
content=content_class(content_dict),
|
|
**event_dict
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class UnsignedData:
|
|
age: int
|
|
redacted_because: Optional[EventBase] = None
|
|
transaction_id: Optional[str] = None
|
|
invite_room_state: Optional[List[EventBase]] = None
|
|
|
|
|
|
@dataclass
|
|
class RoomEvent(EventBase):
|
|
event_id: str
|
|
origin_server_ts: int
|
|
unsigned: UnsignedData
|
|
room: Room
|
|
user_id: str
|
|
|
|
|
|
@dataclass
|
|
class StateEvent(RoomEvent):
|
|
state_key: str
|
|
age: int
|
|
prev_content: Optional[EventBase] = None
|
|
|
|
|
|
@dataclass
|
|
class RedactionEvent(RoomEvent):
|
|
redacts: EventBase
|
|
|
|
|
|
@dataclass
|
|
class MessageEvent(RoomEvent):
|
|
pass
|
|
|
|
|
|
@dataclass
|
|
class PresenceEvent(EventBase):
|
|
pass
|