Update mongo.py

Badhacker98 · Oct 18, 2024 · fae934d · fae934d
1 parent b9f9c7b
commit fae934d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/shizuchat/modules/helpers/mongo.py b/shizuchat/modules/helpers/mongo.py
@@ -84,6 +84,14 @@ async def group_on():
         if not user:
             return await modedb.insert_one({"chat_id": chat_id})
 
+    def get_file_id(msg: Message):
+    if not msg.media: return None
+    for message_type in ("photo", "animation", "audio", "document", "video", "video_note", "voice", "sticker"):
+        obj = getattr(msg, message_type)
+        if obj:
+            setattr(obj, "message_type", message_type)
+            return obj
+
     async def group_off():
         chat_id = 123
         modelist[chat_id] = False