77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
from typing import Union, List, Callable
|
|
from dataclasses import dataclass
|
|
from nakuru.entities.components import Plain, Image
|
|
|
|
|
|
@dataclass
|
|
class CommandItem():
|
|
'''
|
|
用来描述单个指令
|
|
'''
|
|
|
|
command_name: Union[str, tuple] # 指令名
|
|
callback: Callable # 回调函数
|
|
description: str # 描述
|
|
origin: str # 注册来源
|
|
|
|
class CommandResult():
|
|
'''
|
|
用于在Command中返回多个值
|
|
'''
|
|
|
|
def __init__(self, hit: bool = True, success: bool = True, message_chain: list = [], command_name: str = "unknown_command") -> None:
|
|
self.hit = hit
|
|
self.success = success
|
|
self.message_chain = message_chain
|
|
self.command_name = command_name
|
|
self.is_use_t2i = None # default
|
|
|
|
def message(self, message: str):
|
|
'''
|
|
快捷回复消息。
|
|
|
|
CommandResult().message("Hello, world!")
|
|
'''
|
|
self.message_chain = [Plain(message), ]
|
|
return self
|
|
|
|
def error(self, message: str):
|
|
'''
|
|
快捷回复消息。
|
|
|
|
CommandResult().error("Hello, world!")
|
|
'''
|
|
self.success = False
|
|
self.message_chain = [Plain(message), ]
|
|
return self
|
|
|
|
def url_image(self, url: str):
|
|
'''
|
|
快捷回复图片(网络url的格式)。
|
|
|
|
CommandResult().image("https://example.com/image.jpg")
|
|
'''
|
|
self.message_chain = [Image.fromURL(url), ]
|
|
return self
|
|
|
|
def file_image(self, path: str):
|
|
'''
|
|
快捷回复图片(本地文件路径的格式)。
|
|
|
|
CommandResult().image("image.jpg")
|
|
'''
|
|
self.message_chain = [Image.fromFileSystem(path), ]
|
|
return self
|
|
|
|
# def use_t2i(self, use_t2i: bool):
|
|
# '''
|
|
# 设置是否使用文本转图片服务。如果不设置，则跟随用户的设置。
|
|
|
|
# CommandResult().use_t2i(False)
|
|
# '''
|
|
# self.is_use_t2i = use_t2i
|
|
# return self
|
|
|
|
def _result_tuple(self):
|
|
return (self.success, self.message_chain, self.command_name)
|