add get imgfile

This commit is contained in:
yajiwa
2022-05-14 15:41:44 +08:00
parent 03a51ca5dd
commit 98afbb54b6
+20 -5
View File
@@ -16,7 +16,6 @@ try:
except ModuleNotFoundError: except ModuleNotFoundError:
import json import json
scheduler = require("nonebot_plugin_apscheduler").scheduler scheduler = require("nonebot_plugin_apscheduler").scheduler
@@ -203,10 +202,7 @@ def get_message_at(data: Union[str, Message]) -> List[int]:
for msg in data["message"]: for msg in data["message"]:
if msg["type"] == "at": if msg["type"] == "at":
qq_list.append(int(msg["data"]["qq"])) qq_list.append(int(msg["data"]["qq"]))
else:
for seg in data:
if seg.type == "image":
qq_list.append(seg.data["url"])
return qq_list return qq_list
@@ -229,6 +225,25 @@ def get_message_img(data: Union[str, Message]) -> List[str]:
return img_list 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: def get_message_text(data: Union[str, Message]) -> str:
""" """
说明: 说明: