diff --git a/app/demo.py b/app/demo.py index 7558f48..7b58d10 100644 --- a/app/demo.py +++ b/app/demo.py @@ -1,5 +1,6 @@ import os -import lark_oapi as lark +import json +from itertools import chain import lark_oapi as lark @@ -31,7 +32,26 @@ app = Flask(__name__) def do_p2_im_message_receive_v1(data: P2ImMessageReceiveV1) -> None: print(lark.JSON.marshal(data)) record = {} - record["问题概述"] = data.event.message.content + images = [] + message_content = "" + message_type = data.event.message.message_type + content = data.event.message.content + content = json.loads(content) + if message_type == "post": + content = content.get("content", []) + + for item in chain.from_iterable(content): + if item.get("tag") == "text": + content = message_content += item["text"] + break + elif item.get("tag") == "img": + images.append(item["image_key"]) + break + + if message_type == "text": + message_content = content["text"] + + record["问题概述"] = message_content record["创建人"] = data.event.sender.sender_id.open_id receive_id = data.event.sender.sender_id.open_id add_record(record)