mirror of https://github.com/Mai-with-u/MaiBot.git
108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
from typing import Optional
|
|
from dataclasses import dataclass, field
|
|
|
|
|
|
@dataclass
|
|
class DatabaseUserInfo:
|
|
platform: str = field(default_factory=str)
|
|
user_id: str = field(default_factory=str)
|
|
user_nickname: str = field(default_factory=str)
|
|
user_cardname: Optional[str] = None
|
|
|
|
|
|
@dataclass
|
|
class DatabaseGroupInfo:
|
|
group_id: str = field(default_factory=str)
|
|
group_name: str = field(default_factory=str)
|
|
group_platform: Optional[str] = None
|
|
|
|
|
|
@dataclass
|
|
class DatabaseChatInfo:
|
|
stream_id: str = field(default_factory=str)
|
|
platform: str = field(default_factory=str)
|
|
create_time: float = field(default_factory=float)
|
|
last_active_time: float = field(default_factory=float)
|
|
user_info: DatabaseUserInfo = field(default_factory=DatabaseUserInfo)
|
|
group_info: Optional[DatabaseGroupInfo] = None
|
|
|
|
|
|
@dataclass
|
|
class DatabaseMessages:
|
|
chat_info: DatabaseChatInfo
|
|
user_info: DatabaseUserInfo
|
|
group_info: Optional[DatabaseGroupInfo] = None
|
|
|
|
message_id: str = field(default_factory=str)
|
|
time: float = field(default_factory=float)
|
|
chat_id: str = field(default_factory=str)
|
|
reply_to: Optional[str] = None
|
|
interest_value: Optional[float] = None
|
|
|
|
key_words: Optional[str] = None
|
|
key_words_lite: Optional[str] = None
|
|
is_mentioned: Optional[bool] = None
|
|
|
|
# 从 chat_info 扁平化而来的字段
|
|
chat_info_stream_id: str = field(default_factory=str)
|
|
chat_info_platform: str = field(default_factory=str)
|
|
chat_info_user_platform: str = field(default_factory=str)
|
|
chat_info_user_id: str = field(default_factory=str)
|
|
chat_info_user_nickname: str = field(default_factory=str)
|
|
chat_info_user_cardname: Optional[str] = None
|
|
chat_info_group_platform: Optional[str] = None
|
|
chat_info_group_id: Optional[str] = None
|
|
chat_info_group_name: Optional[str] = None
|
|
chat_info_create_time: float = field(default_factory=float)
|
|
chat_info_last_active_time: float = field(default_factory=float)
|
|
|
|
# 从顶层 user_info 扁平化而来的字段 (消息发送者信息)
|
|
user_platform: str = field(default_factory=str)
|
|
user_id: str = field(default_factory=str)
|
|
user_nickname: str = field(default_factory=str)
|
|
user_cardname: Optional[str] = None
|
|
|
|
processed_plain_text: Optional[str] = None # 处理后的纯文本消息
|
|
display_message: Optional[str] = None # 显示的消息
|
|
|
|
priority_mode: Optional[str] = None
|
|
priority_info: Optional[str] = None
|
|
|
|
additional_config: Optional[str] = None
|
|
is_emoji: bool = False
|
|
is_picid: bool = False
|
|
is_command: bool = False
|
|
is_notify: bool = False
|
|
|
|
selected_expressions: Optional[str] = None
|
|
|
|
def __post_init__(self):
|
|
self.user_info = DatabaseUserInfo(
|
|
user_id=self.user_id,
|
|
user_nickname=self.user_nickname,
|
|
user_cardname=self.user_cardname,
|
|
platform=self.user_platform,
|
|
)
|
|
|
|
if self.chat_info_group_id and self.chat_info_group_name:
|
|
self.group_info = DatabaseGroupInfo(
|
|
group_id=self.chat_info_group_id,
|
|
group_name=self.chat_info_group_name,
|
|
group_platform=self.chat_info_group_platform,
|
|
)
|
|
|
|
chat_user_info = DatabaseUserInfo(
|
|
user_id=self.chat_info_user_id,
|
|
user_nickname=self.chat_info_user_nickname,
|
|
user_cardname=self.chat_info_user_cardname,
|
|
platform=self.chat_info_user_platform,
|
|
)
|
|
self.chat_info = DatabaseChatInfo(
|
|
stream_id=self.chat_info_stream_id,
|
|
platform=self.chat_info_platform,
|
|
create_time=self.chat_info_create_time,
|
|
last_active_time=self.chat_info_last_active_time,
|
|
user_info=chat_user_info,
|
|
group_info=self.group_info,
|
|
)
|