From c0d055b8251c0bf4474e1a9693356bdf1eb0e39f Mon Sep 17 00:00:00 2001
From: HibiKier <775757368@qq.com>
Date: Sat, 21 May 2022 13:20:58 +0800
Subject: [PATCH] add get_message_img_file

---
 utils/utils.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/utils/utils.py b/utils/utils.py
index d7137941..7133da6a 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -229,6 +229,25 @@ def get_message_img(data: Union[str, Message]) -> List[str]:
     return img_list
 
 
+def get_message_img_file(data: Union[str, Message]) -> List[str]:
+    """
+    说明：
+        获取消息中所有的 图片file
+    参数：
+        :param data: event.json()
+    """
+    file_list = []
+    if isinstance(data, str):
+        data = json.loads(data)
+        for msg in data["message"]:
+            if msg["type"] == "image":
+                file_list.append(msg["data"]["file"])
+    else:
+        for seg in data["image"]:
+            file_list.append(seg.data["file"])
+    return file_list
+
+
 def get_message_text(data: Union[str, Message]) -> str:
     """
     说明：