fix:优化代码逻辑

This commit is contained in:
Azide 2022-07-10 12:51:54 +08:00
parent a1e246e1b4
commit f6ff760c56

View File

@ -31,13 +31,11 @@ class _CustomPost(BasePost):
if message_segment.type == "text": if message_segment.type == "text":
md += "{}<br>".format(message_segment.data.get("text", "")) md += "{}<br>".format(message_segment.data.get("text", ""))
elif message_segment.type == "image": elif message_segment.type == "image":
try: # 先尝试获取file的值没有再尝试获取url的值都没有则为空
# 先尝试获取file的值没有再尝试获取url的值都没有则为空 pic_res = message_segment.data.get("file") or message_segment.data.get(
pic_res = message_segment.data.get( "url", ""
"file", message_segment.data.get("url", "") )
) if not pic_res:
assert pic_res
except AssertionError:
logger.warning("无法获取到图片资源:MessageSegment.image中file/url字段均为空") logger.warning("无法获取到图片资源:MessageSegment.image中file/url字段均为空")
else: else:
md += "![Image]({})\n".format(pic_res) md += "![Image]({})\n".format(pic_res)