mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-04 02:26:11 +08:00
32e3bcc022
* 🐛 调整ruff的pytest警告 * 🐛 调整导入关系警告 * 🐛 删除奇怪无用的赋值和取值逻辑 * ✅ 不同测试部分所用变量应加以区分 * 🐛 subs_io model添加默认值 * 🐛 修完所有的 ruff PT001 警告 * 🔧 按ruff建议修改ruff配置 warning: The top-level linter settings are deprecated in favour of their counterparts in the `lint` section. Please update the following options in `pyproject.toml`: - 'ignore' -> 'lint.ignore' - 'select' -> 'lint.select' --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
118 lines
3.9 KiB
Python
118 lines
3.9 KiB
Python
import asyncio
|
|
from datetime import datetime
|
|
|
|
from nonebot import on_command
|
|
from nonebot.typing import T_State
|
|
from nonebot.matcher import Matcher
|
|
from nonebot.rule import Rule, to_me
|
|
from nonebot.permission import SUPERUSER
|
|
from nonebot_plugin_saa import TargetQQGroup
|
|
from nonebot.params import ArgStr, ArgPlainText
|
|
from nonebot.adapters import Bot, MessageTemplate
|
|
from nonebot.adapters.onebot.v11.event import PrivateMessageEvent
|
|
|
|
from .add_sub import do_add_sub
|
|
from .del_sub import do_del_sub
|
|
from .query_sub import do_query_sub
|
|
from .utils import common_platform, admin_permission, gen_handle_cancel, configurable_to_me, set_target_user_info
|
|
|
|
add_sub_matcher = on_command(
|
|
"添加订阅",
|
|
rule=configurable_to_me,
|
|
permission=admin_permission(),
|
|
priority=5,
|
|
block=True,
|
|
)
|
|
add_sub_matcher.handle()(set_target_user_info)
|
|
do_add_sub(add_sub_matcher)
|
|
|
|
|
|
query_sub_matcher = on_command("查询订阅", rule=configurable_to_me, priority=5, block=True)
|
|
query_sub_matcher.handle()(set_target_user_info)
|
|
do_query_sub(query_sub_matcher)
|
|
|
|
|
|
del_sub_matcher = on_command(
|
|
"删除订阅",
|
|
rule=configurable_to_me,
|
|
permission=admin_permission(),
|
|
priority=5,
|
|
block=True,
|
|
)
|
|
del_sub_matcher.handle()(set_target_user_info)
|
|
do_del_sub(del_sub_matcher)
|
|
|
|
group_manage_matcher = on_command("群管理", rule=to_me(), permission=SUPERUSER, priority=4, block=True)
|
|
|
|
group_handle_cancel = gen_handle_cancel(group_manage_matcher, "已取消")
|
|
|
|
|
|
@group_manage_matcher.handle()
|
|
async def send_group_list(bot: Bot, event: PrivateMessageEvent, state: T_State):
|
|
groups = await bot.call_api("get_group_list")
|
|
res_text = "请选择需要管理的群：\n"
|
|
group_number_idx = {}
|
|
for idx, group in enumerate(groups, 1):
|
|
group_number_idx[idx] = group["group_id"]
|
|
res_text += f'{idx}. {group["group_id"]} - {group["group_name"]}\n'
|
|
res_text += "请输入左侧序号\n中止操作请输入'取消'"
|
|
# await group_manage_matcher.send(res_text)
|
|
state["_prompt"] = res_text
|
|
state["group_number_idx"] = group_number_idx
|
|
|
|
|
|
@group_manage_matcher.got("group_idx", MessageTemplate("{_prompt}"), [group_handle_cancel])
|
|
async def do_choose_group_number(state: T_State, event: PrivateMessageEvent, group_idx: str = ArgPlainText()):
|
|
group_number_idx: dict[int, int] = state["group_number_idx"]
|
|
assert group_number_idx
|
|
idx = int(group_idx)
|
|
if idx not in group_number_idx.keys():
|
|
await group_manage_matcher.reject("请输入正确序号")
|
|
group_id = group_number_idx[idx]
|
|
state["target_user_info"] = TargetQQGroup(group_id=group_id)
|
|
|
|
|
|
@group_manage_matcher.got("command", "请输入需要使用的命令：添加订阅，查询订阅，删除订阅，取消", [group_handle_cancel])
|
|
async def do_dispatch_command(
|
|
bot: Bot,
|
|
event: PrivateMessageEvent,
|
|
state: T_State,
|
|
matcher: Matcher,
|
|
command: str = ArgStr(),
|
|
):
|
|
if command not in {"添加订阅", "查询订阅", "删除订阅", "取消"}:
|
|
await group_manage_matcher.reject("请输入正确的命令")
|
|
permission = await matcher.update_permission(bot, event)
|
|
new_matcher = Matcher.new(
|
|
"message",
|
|
Rule(),
|
|
permission,
|
|
handlers=None,
|
|
temp=True,
|
|
priority=0,
|
|
block=True,
|
|
plugin=matcher.plugin,
|
|
module=matcher.module,
|
|
expire_time=datetime.now(),
|
|
default_state=matcher.state,
|
|
default_type_updater=matcher.__class__._default_type_updater,
|
|
default_permission_updater=matcher.__class__._default_permission_updater,
|
|
)
|
|
if command == "查询订阅":
|
|
do_query_sub(new_matcher)
|
|
elif command == "添加订阅":
|
|
do_add_sub(new_matcher)
|
|
else:
|
|
do_del_sub(new_matcher)
|
|
new_matcher_ins = new_matcher()
|
|
asyncio.create_task(new_matcher_ins.run(bot, event, state))
|
|
|
|
|
|
__all__ = [
|
|
"common_platform",
|
|
"add_sub_matcher",
|
|
"query_sub_matcher",
|
|
"del_sub_matcher",
|
|
"group_manage_matcher",
|
|
]
|