ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2024-11-24 17:15:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

✨ implement group and private message

Browse Source
This commit is contained in:
StarHeartHunt 2021-07-05 11:07:33 +08:00
parent 150ef30b73
commit b6979594a7
1 changed files with 33 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,7 +1,7 @@
import inspect import inspect
import json import json
from typing import Any, List, Literal, Optional, Type from typing import Any, List, Literal, Optional, Type, Union
from pygtrie import StringTrie from pygtrie import StringTrie
from pydantic import BaseModel, root_validator, Field from pydantic import BaseModel, root_validator, Field
@ -101,11 +101,27 @@ class EventMessage(BaseModel):
return values return values
class GroupEventMessage(EventMessage):
chat_type: Literal["group"]
class PrivateEventMessage(EventMessage):
chat_type: Literal["p2p"]
class MessageEventDetail(BaseModel): class MessageEventDetail(BaseModel):
sender: Sender sender: Sender
message: EventMessage message: EventMessage
class GroupMessageEventDetail(MessageEventDetail):
message: GroupEventMessage
class PrivateMessageEventDetail(MessageEventDetail):
message: PrivateEventMessage
class MessageEvent(Event): class MessageEvent(Event):
__event__ = "im.message.receive_v1" __event__ = "im.message.receive_v1"
event: MessageEventDetail event: MessageEventDetail
@ -116,13 +132,15 @@ class MessageEvent(Event):
@overrides(Event) @overrides(Event)
def get_event_name(self) -> str: def get_event_name(self) -> str:
return f"{self.get_type()}.{super().get_type()}" return f"{self.get_type()}.{self.event.message.chat_type}"
@overrides(Event) @overrides(Event)
def get_event_description(self) -> str: def get_event_description(self) -> str:
#TODO:换成GroupId
return ( return (
f"Message[{super().get_type()}]" f"Message[{super().get_type()}]"
f" {self.event.message.message_id} from {self.event.sender.sender_id.user_id}" f" {self.event.message.message_id} from {self.get_user_id()}"
f"@[{self.event.message.chat_type}:{self.event.message.chat_id}]"
f" {self.event.message.content}") f" {self.event.message.content}")
@overrides(Event) @overrides(Event)
@ -135,11 +153,21 @@ class MessageEvent(Event):
@overrides(Event) @overrides(Event)
def get_user_id(self) -> str: def get_user_id(self) -> str:
return self.event.sender.sender_id.user_id return self.event.sender.sender_id.union_id
@overrides(Event) @overrides(Event)
def get_session_id(self) -> str: def get_session_id(self) -> str:
return self.event.sender.sender_id.user_id return f"{self.event.message.chat_type}_{self.event.message.chat_id}_{self.get_user_id()}"
class GroupMessageEvent(MessageEvent):
__event__ = "im.message.receive_v1.group"
event: GroupMessageEventDetail
class PrivateMessageEvent(MessageEvent):
__event__ = "im.message.receive_v1.private"
event: PrivateMessageEventDetail
class MessageReader(BaseModel): class MessageReader(BaseModel):