agatha/maubot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add edit function to MaubotMessageEvent

Browse Source
This commit is contained in:
Tulir Asokan 2019-08-30 01:27:40 +03:00
parent aaf5e43d9f
commit eb353c4964
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
maubot/matrix.py
View File

@ -50,14 +50,17 @@ class MaubotMessageEvent(MessageEvent):
def respond(self, content: Union[str, MessageEventContent], def respond(self, content: Union[str, MessageEventContent],
event_type: EventType = EventType.ROOM_MESSAGE, markdown: bool = True, event_type: EventType = EventType.ROOM_MESSAGE, markdown: bool = True,
html_in_markdown: bool = False, reply: bool = False) -> Awaitable[EventID]: html_in_markdown: bool = False, reply: bool = False,
edits: Optional[Union[EventID, MessageEvent]] = None) -> Awaitable[EventID]:
if isinstance(content, str): if isinstance(content, str):
content = TextMessageEventContent(msgtype=MessageType.NOTICE, body=content) content = TextMessageEventContent(msgtype=MessageType.NOTICE, body=content)
if markdown: if markdown:
content.format = Format.HTML content.format = Format.HTML
content.body, content.formatted_body = parse_markdown(content.body, content.body, content.formatted_body = parse_markdown(content.body,
allow_html=html_in_markdown) allow_html=html_in_markdown)
if reply and not self.disable_reply: if edits:
content.set_edit(edits)
elif reply and not self.disable_reply:
content.set_reply(self) content.set_reply(self)
return self.client.send_message_event(self.room_id, event_type, content) return self.client.send_message_event(self.room_id, event_type, content)
@ -73,6 +76,12 @@ class MaubotMessageEvent(MessageEvent):
def react(self, key: str) -> Awaitable[None]: def react(self, key: str) -> Awaitable[None]:
return self.client.react(self.room_id, self.event_id, key) return self.client.react(self.room_id, self.event_id, key)
def edit(self, content: Union[str, MessageEventContent],
event_type: EventType = EventType.ROOM_MESSAGE, markdown: bool = True,
html_in_markdown: bool = False) -> Awaitable[EventID]:
return self.respond(content, event_type, markdown, edits=self,
html_in_markdown=html_in_markdown)
class MaubotMatrixClient(MatrixClient): class MaubotMatrixClient(MatrixClient):
def send_markdown(self, room_id: RoomID, markdown: str, msgtype: MessageType = MessageType.TEXT, def send_markdown(self, room_id: RoomID, markdown: str, msgtype: MessageType = MessageType.TEXT,