From 22b98b6f3a47e2db674cd9b77608b7bd9b192a53 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=CA=99=E1=B4=80=E1=B4=85=20=E1=B4=8D=E1=B4=9C=C9=B4?=
 =?UTF-8?q?=E1=B4=85=E1=B4=80=20=F0=9F=8C=B8?=
 <154504921+Badhacker98@users.noreply.github.com>
Date: Fri, 18 Oct 2024 15:18:00 +0530
Subject: [PATCH] Create ults.py

---
 shizuchat/modules/helpers/ults.py | 8 ++++++++
 1 file changed, 8 insertions(+)
 create mode 100644 shizuchat/modules/helpers/ults.py

diff --git a/shizuchat/modules/helpers/ults.py b/shizuchat/modules/helpers/ults.py
new file mode 100644
index 0000000..b0a7605
--- /dev/null
+++ 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
+