Soulter
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
import time
|
|
from typing import List
|
|
from dataclasses import dataclass
|
|
from astrbot.core.message.components import BaseMessageComponent
|
|
from .message_type import MessageType
|
|
|
|
|
|
@dataclass
|
|
class MessageMember:
|
|
user_id: str # 发送者id
|
|
nickname: str = None
|
|
|
|
def __str__(self):
|
|
# 使用 f-string 来构建返回的字符串表示形式
|
|
return (
|
|
f"User ID: {self.user_id},"
|
|
f"Nickname: {self.nickname if self.nickname else 'N/A'}"
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class Group:
|
|
group_id: str
|
|
"""群号"""
|
|
group_name: str = None
|
|
"""群名称"""
|
|
group_avatar: str = None
|
|
"""群头像"""
|
|
group_owner: str = None
|
|
"""群主 id"""
|
|
group_admins: List[str] = None
|
|
"""群管理员 id"""
|
|
members: List[MessageMember] = None
|
|
"""所有群成员"""
|
|
|
|
def __str__(self):
|
|
# 使用 f-string 来构建返回的字符串表示形式
|
|
return (
|
|
f"Group ID: {self.group_id}\n"
|
|
f"Name: {self.group_name if self.group_name else 'N/A'}\n"
|
|
f"Avatar: {self.group_avatar if self.group_avatar else 'N/A'}\n"
|
|
f"Owner ID: {self.group_owner if self.group_owner else 'N/A'}\n"
|
|
f"Admin IDs: {self.group_admins if self.group_admins else 'N/A'}\n"
|
|
f"Members Len: {len(self.members) if self.members else 0}\n"
|
|
f"First Member: {self.members[0] if self.members else 'N/A'}\n"
|
|
)
|
|
|
|
|
|
class AstrBotMessage:
|
|
"""
|
|
AstrBot 的消息对象
|
|
"""
|
|
|
|
type: MessageType # 消息类型
|
|
self_id: str # 机器人的识别id
|
|
session_id: str # 会话id。取决于 unique_session 的设置。
|
|
message_id: str # 消息id
|
|
group_id: str = "" # 群组id，如果为私聊，则为空
|
|
sender: MessageMember # 发送者
|
|
message: List[BaseMessageComponent] # 消息链使用 Nakuru 的消息链格式
|
|
message_str: str # 最直观的纯文本消息字符串
|
|
raw_message: object
|
|
timestamp: int # 消息时间戳
|
|
|
|
def __init__(self) -> None:
|
|
self.timestamp = int(time.time())
|
|
|
|
def __str__(self) -> str:
|
|
return str(self.__dict__)
|