63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
from dataclasses import dataclass
|
|
from typing import List, Optional, Dict, Any
|
|
from .sender import Sender # 导入上面的 Sender
|
|
|
|
@dataclass
|
|
class MessageSegment:
|
|
type: str
|
|
data: Dict[str, Any]
|
|
|
|
@property
|
|
def text(self) -> str:
|
|
"""如果是文本段，返回文本内容，否则返回空字符串"""
|
|
return self.data.get("text", "") if self.type == "text" else ""
|
|
|
|
@property
|
|
def image_url(self) -> str:
|
|
"""如果是图片段，返回图片 URL"""
|
|
return self.data.get("url", "") if self.type == "image" else ""
|
|
|
|
def is_at(self, user_id: int = None) -> bool:
|
|
"""判断是否是 @某人"""
|
|
if self.type != "at":
|
|
return False
|
|
if user_id is None:
|
|
return True
|
|
return str(self.data.get("qq")) == str(user_id)
|
|
|
|
def __repr__(self):
|
|
return f"[MS:{self.type}:{self.data}]"
|
|
|
|
@dataclass
|
|
class Event:
|
|
post_type: str
|
|
message_type: str # group 或 private
|
|
user_id: int
|
|
self_id: int
|
|
raw_message: str
|
|
message: List[MessageSegment]
|
|
sender: Sender
|
|
time: int
|
|
group_id: Optional[int] = None
|
|
target_id: Optional[int] = None
|
|
|
|
@classmethod
|
|
def from_dict(cls, data: dict):
|
|
raw_msg_array = data.get("message", [])
|
|
segments = [
|
|
MessageSegment(type=seg["type"], data=seg["data"])
|
|
for seg in raw_msg_array
|
|
]
|
|
|
|
data_copy = data.copy()
|
|
data_copy["message"] = segments
|
|
|
|
sender_data = data.get("sender", {})
|
|
sender_obj = Sender(**{k: v for k, v in sender_data.items() if k in Sender.__annotations__})
|
|
|
|
data_copy = data.copy()
|
|
data_copy["message"] = segments
|
|
data_copy["sender"] = sender_obj # 关键点：把对象塞进去
|
|
|
|
valid_data = {k: v for k, v in data_copy.items() if k in cls.__annotations__}
|
|
return cls(**valid_data) |