Add helper method to get formatted body

mautrix · Nov 7, 2020 · 68dfd6b · 68dfd6b
1 parent cf79ae0
commit 68dfd6b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mautrix/types/event/message.py b/mautrix/types/event/message.py
@@ -336,6 +336,11 @@ def set_reply(self, reply_to: Union['MessageEvent', EventID],
                                    + self.formatted_body)
             self.body = reply_to.make_reply_fallback_text(displayname) + self.body
 
+    def formatted(self, format: Format) -> Optional[str]:
+        if self.format == format:
+            return self.formatted_body
+        return None
+
     def trim_reply_fallback(self) -> None:
         if self.get_reply_to():
             self._trim_reply_fallback_text()