Create ults.py

Badhacker98 · Oct 18, 2024 · 22b98b6 · 22b98b6
1 parent 466bdd4
commit 22b98b6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/shizuchat/modules/helpers/ults.py b/shizuchat/modules/helpers/ults.py
@@ -0,0 +1,8 @@
+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
+