ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-02-18 00:30:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ implement at parser

Browse Source
This commit is contained in:
StarHeartHunt 2021-07-08 22:30:39 +08:00
parent 15ab958a70
commit 67770ffa6f
3 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
packages/nonebot-adapter-feishu/nonebot/adapters/feishu/bot.py
View File

@ -293,7 +293,7 @@ class Bot(BaseBot):
at_sender = at_sender and bool(event.get_user_id())
if at_sender and receive_id_type != "union_id":
msg += MessageSegment.at(event.get_user_id(), "StarHeart") + " "
msg = MessageSegment.at(event.get_user_id(), "StarHeart") + " " + msg
msg_type, content = MessageSerializer(msg).serialize()

7
packages/nonebot-adapter-feishu/nonebot/adapters/feishu/event.py
View File

@ -97,10 +97,9 @@ class EventMessage(BaseModel):
@root_validator(pre=True)
def parse_message(cls, values: dict):
#TODO:解析mentions替换message的user_id，传入deserializer
values["content"] = MessageDeserializer(
values["message_type"],
json.loads(values["content"])).deserialize()
values["message_type"], json.loads(values["content"]),
values.get("mentions")).deserialize()
return values
@ -154,7 +153,7 @@ class MessageEvent(Event):
@overrides(Event)
def get_user_id(self) -> str:
return self.event.sender.sender_id.user_id
return self.event.sender.sender_id.open_id
@overrides(Event)
def get_session_id(self) -> str:

15
packages/nonebot-adapter-feishu/nonebot/adapters/feishu/message.py
View File

@ -2,7 +2,7 @@ import json
import itertools
from dataclasses import dataclass
from typing import Any, Dict, List, Tuple, Type, Union, Mapping, Iterable
from typing import Any, Dict, List, Optional, Tuple, Type, Union, Mapping, Iterable
from nonebot.adapters import Message as BaseMessage, MessageSegment as BaseMessageSegment
from nonebot.typing import overrides
@ -214,15 +214,28 @@ class MessageDeserializer:
"""
type: str
data: Dict[str, Any]
mentions: Optional[List[dict]]
def deserialize(self) -> Message:
dict_mention = {}
if self.type == "post":
if self.mentions:
for mention in self.mentions:
dict_mention[mention["key"]] = mention
msg = Message()
if self.data["title"] != "":
msg += MessageSegment("text", {'text': self.data["title"]})
for seg in itertools.chain(*self.data["content"]):
tag = seg.pop("tag")
if tag == "at":
seg["user_name"] = dict_mention[seg["user_id"]]["name"]
seg["user_id"] = dict_mention[
seg["user_id"]]["id"]["open_id"]
msg += MessageSegment(tag if tag != "img" else "image", seg)
return msg._merge()
else: