mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-05 11:26:43 +08:00
0201d5b316
* 🚧 第一次尝试 50/82 * 62/82 调整了清除数据库的位置 * 🚧 pytest-mock * 🚧 fix test_send * 🚧 intruduce app request * 🚧 close and remove tinydb after each test * 🚧 clean ScheduleTimeWeight table * 🚧 reload http module to test proxy * ✅ 合并 main 的代码 * 🚧 在每次测试结束后关闭browser * 🧑💻 在mcbbsnews渲染异常时添加logger --------- Co-authored-by: hemengyang <hmy0119@gmail.com> Co-authored-by: Azide <rukuy@qq.com>
154 lines
5.3 KiB
Python
154 lines
5.3 KiB
Python
from typing import TYPE_CHECKING, TypedDict
|
||
|
||
from typing_extensions import Literal
|
||
|
||
if TYPE_CHECKING:
|
||
from nonebot.adapters.onebot.v11 import GroupMessageEvent, PrivateMessageEvent
|
||
|
||
|
||
class AppReq(TypedDict, total=False):
|
||
refresh_bot: bool
|
||
no_init_db: bool
|
||
|
||
|
||
def fake_group_message_event(**field) -> "GroupMessageEvent":
|
||
from nonebot.adapters.onebot.v11 import GroupMessageEvent, Message
|
||
from nonebot.adapters.onebot.v11.event import Sender
|
||
from pydantic import create_model
|
||
|
||
_Fake = create_model("_Fake", __base__=GroupMessageEvent)
|
||
|
||
class FakeEvent(_Fake):
|
||
time: int = 1000000
|
||
self_id: int = 1
|
||
post_type: Literal["message"] = "message"
|
||
sub_type: str = "normal"
|
||
user_id: int = 10
|
||
message_type: Literal["group"] = "group"
|
||
group_id: int = 10000
|
||
message_id: int = 1
|
||
message: Message = Message("test")
|
||
raw_message: str = "test"
|
||
font: int = 0
|
||
sender: Sender = Sender(
|
||
card="",
|
||
nickname="test",
|
||
role="member",
|
||
)
|
||
to_me: bool = False
|
||
|
||
class Config:
|
||
extra = "forbid"
|
||
|
||
return FakeEvent(**field)
|
||
|
||
|
||
def fake_private_message_event(**field) -> "PrivateMessageEvent":
|
||
from nonebot.adapters.onebot.v11 import Message, PrivateMessageEvent
|
||
from nonebot.adapters.onebot.v11.event import Sender
|
||
from pydantic import create_model
|
||
|
||
_Fake = create_model("_Fake", __base__=PrivateMessageEvent)
|
||
|
||
class FakeEvent(_Fake):
|
||
time: int = 1000000
|
||
self_id: int = 1
|
||
post_type: Literal["message"] = "message"
|
||
sub_type: str = "friend"
|
||
user_id: int = 10
|
||
message_type: Literal["private"] = "private"
|
||
message_id: int = 1
|
||
message: Message = Message("test")
|
||
raw_message: str = "test"
|
||
font: int = 0
|
||
sender: Sender = Sender(nickname="test")
|
||
to_me: bool = False
|
||
|
||
class Config:
|
||
extra = "forbid"
|
||
|
||
return FakeEvent(**field)
|
||
|
||
|
||
from nonebot.adapters.onebot.v11.event import Sender
|
||
|
||
fake_admin_user = Sender(nickname="test", role="admin")
|
||
fake_superuser = Sender(user_id=10001, nickname="superuser")
|
||
|
||
|
||
class BotReply:
|
||
@staticmethod
|
||
def add_reply_on_platform(platform_manager, common_platform):
|
||
return (
|
||
"请输入想要订阅的平台，目前支持，请输入冒号左边的名称：\n"
|
||
+ "".join(
|
||
[
|
||
"{}：{}\n".format(
|
||
platform_name, platform_manager[platform_name].name
|
||
)
|
||
for platform_name in common_platform
|
||
]
|
||
)
|
||
+ "要查看全部平台请输入：“全部”\n中止订阅过程请输入：“取消”"
|
||
)
|
||
|
||
@staticmethod
|
||
def add_reply_on_platform_input_allplatform(platform_manager):
|
||
return "全部平台\n" + "\n".join(
|
||
[
|
||
"{}：{}".format(platform_name, platform.name)
|
||
for platform_name, platform in platform_manager.items()
|
||
]
|
||
)
|
||
|
||
@staticmethod
|
||
def add_reply_on_id_input_search():
|
||
search_url = "https://nonebot-bison.netlify.app/usage/#%E6%89%80%E6%94%AF%E6%8C%81%E5%B9%B3%E5%8F%B0%E7%9A%84-uid"
|
||
search_title = "Bison所支持的平台UID"
|
||
search_content = "查询相关平台的uid格式或获取方式"
|
||
search_image = "https://s3.bmp.ovh/imgs/2022/03/ab3cc45d83bd3dd3.jpg"
|
||
type = "share"
|
||
data = {
|
||
"url": search_url,
|
||
"title": search_title,
|
||
"content": search_content,
|
||
"image": search_image,
|
||
}
|
||
msg = [type, data]
|
||
return msg
|
||
|
||
@staticmethod
|
||
def add_reply_on_target_confirm(platform, name, id):
|
||
return f"即将订阅的用户为:{platform} {name} {id}\n如有错误请输入“取消”重新订阅"
|
||
|
||
@staticmethod
|
||
def add_reply_on_cats(platform_manager, platform: str):
|
||
return "请输入要订阅的类别，以空格分隔，支持的类别有：{}".format(
|
||
" ".join(list(platform_manager[platform].categories.values()))
|
||
)
|
||
|
||
@staticmethod
|
||
def add_reply_on_cats_input_error(cat: str):
|
||
return "不支持 {}".format(cat)
|
||
|
||
@staticmethod
|
||
def add_reply_subscribe_success(name):
|
||
return "添加 {} 成功".format(name)
|
||
|
||
@staticmethod
|
||
def add_reply_on_id(platform: object) -> str:
|
||
base_text = "请输入订阅用户的id\n查询id获取方法请回复:“查询”"
|
||
extra_text = (
|
||
("1." + platform.parse_target_promot + "\n2.")
|
||
if platform.parse_target_promot
|
||
else ""
|
||
)
|
||
return extra_text + base_text
|
||
|
||
add_reply_on_id_input_error = "id输入错误"
|
||
add_reply_on_target_parse_input_error = "不能从你的输入中提取出id，请检查你输入的内容是否符合预期"
|
||
add_reply_on_platform_input_error = "平台输入错误"
|
||
add_reply_on_tags = '请输入要订阅/屏蔽的标签(不含#号)\n多个标签请使用空格隔开\n订阅所有标签输入"全部标签"\n具体规则回复"详情"'
|
||
add_reply_on_tags_need_more_info = "订阅标签直接输入标签内容\n屏蔽标签请在标签名称前添加~号\n详见https://nonebot-bison.netlify.app/usage/#%E5%B9%B3%E5%8F%B0%E8%AE%A2%E9%98%85%E6%A0%87%E7%AD%BE-tag"
|
||
add_reply_abort = "已中止订阅"
|