Fix check

mautrix · Mar 25, 2024 · aed4f6a · aed4f6a
1 parent 637534c
commit aed4f6a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/mautrix_telegram/portal_util/message_convert.py b/mautrix_telegram/portal_util/message_convert.py
@@ -209,13 +209,14 @@ async def convert(
             )
         return converted
 
-    def _should_convert_full_document(self, document, is_bot: bool, is_channel: bool) -> bool:
-        if not isinstance(document, MessageMediaDocument):
+    def _should_convert_full_document(self, media, is_bot: bool, is_channel: bool) -> bool:
+        if not isinstance(media, MessageMediaDocument):
             return True
+        size = media.document.size
         if is_bot and self.config["bridge.document_as_link_size.bot"]:
-            return document.size < self.config["bridge.document_as_link_size.bot"] * 1000**2
+            return size < self.config["bridge.document_as_link_size.bot"] * 1000**2
         if is_channel and self.config["bridge.document_as_link_size.channel"]:
-            return document.size < self.config["bridge.document_as_link_size.channel"] * 1000**2
+            return size < self.config["bridge.document_as_link_size.channel"] * 1000**2
         return True
 
     @staticmethod