icalingua-python-bot/data_struct.py

103 lines
2.8 KiB
Python
Raw Normal View History

2024-01-07 18:48:44 +08:00
from typing import List, Optional, Union, Literal, Tuple
import qtoml
from lib_not_dr.types import Options
class AtElement(Options):
text: str
id: Union[int, Literal['all']] = 'all'
class ReplyMessage(Options):
id: str
username: str = ''
content: str = ''
files: list = []
def to_json(self) -> dict:
return {
'_id': self.id,
'username': self.username,
'content': self.content,
'files': self.files
}
2024-02-02 21:44:15 +08:00
class SendMessage(Options):
2024-01-07 18:48:44 +08:00
content: str
room_id: Optional[int] = None
room: Optional[int] = None # room id 和 room 二选一 ( 实际上直接填 room id 就行了 )
file: None = None # TODO: 上传文件
reply_to: Optional[ReplyMessage] = None # 源码 给了一个 any TODO: 回复消息
b64_img: Optional[str] = None # TODO: 发送图片
at: Optional[List[AtElement]] = [] # TODO: @某人
sticker: Optional[None] = None # TODO: 发送表情
message_type: Optional[str] = None # TODO: 消息类型
def to_json(self) -> dict:
return {
'content': self.content,
'roomId': self.room_id,
'room': self.room,
'file': self.file,
'replyMessage': self.reply_to.to_json() if self.reply_to else None,
'b64img': self.b64_img,
'at': self.at,
'sticker': self.sticker,
'messageType': self.message_type
}
2024-01-25 22:08:02 +08:00
2024-02-02 21:44:15 +08:00
def to_content(self, content: str) -> "SendMessage":
2024-01-25 22:08:02 +08:00
self.content = content
return self
2024-01-07 18:48:44 +08:00
2024-02-02 21:44:15 +08:00
class NewMessage(Options):
sender_id: int
2024-02-02 22:23:51 +08:00
sender_name: str
2024-02-02 21:44:15 +08:00
room_id: int
content: str
msg_id: str
2024-02-02 22:23:51 +08:00
data: dict
def init(self, **kwargs) -> None:
data = kwargs.pop('data')
self.sender_name = data["message"]["username"]
self.sender_id = data["message"]["senderId"]
self.content = data["message"]["content"]
self.room_id = data["roomId"]
self.msg_id = data["message"]["_id"]
2024-02-02 21:44:15 +08:00
2024-02-02 22:23:51 +08:00
def is_self(self, self_id: int) -> bool:
return self.sender_id == self_id
2024-02-02 21:44:15 +08:00
2024-01-07 18:48:44 +08:00
class BotConfig(Options):
name = 'icalingua bot config'
# _check_filled = True
private_key: str
host: str
2024-01-07 18:48:44 +08:00
self_id: int
notice_room: List[int]
notice_start: bool = False
admin_list: List[int]
py_plugin_path: str
2024-01-07 18:48:44 +08:00
def init(self, **kwargs) -> None:
if self.notice_room is None:
self.notice_start = False
def get_config(config_path: str = 'config.toml') -> BotConfig:
with open(config_path, 'r', encoding='utf-8') as f:
config = qtoml.decoder.load(f)
return BotConfig(**config)
class BotStatus(Options):
inited: bool = False
running: bool = False
rooms: List[int] = []