summaryrefslogtreecommitdiff
path: root/ATRI/plugins/manage/modules/service.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/plugins/manage/modules/service.py')
-rw-r--r--ATRI/plugins/manage/modules/service.py160
1 files changed, 67 insertions, 93 deletions
diff --git a/ATRI/plugins/manage/modules/service.py b/ATRI/plugins/manage/modules/service.py
index 6489ad6..a3624df 100644
--- a/ATRI/plugins/manage/modules/service.py
+++ b/ATRI/plugins/manage/modules/service.py
@@ -5,7 +5,7 @@ from nonebot.adapters.cqhttp import (
Bot,
MessageEvent,
GroupMessageEvent,
- PrivateMessageEvent
+ PrivateMessageEvent,
)
from ATRI.service import Service as sv
@@ -21,46 +21,40 @@ __doc__ = """
"""
cur_service_ena = sv.on_command(
- cmd="启用功能",
- docs=__doc__,
- permission=SUPERUSER|GROUP_ADMIN|GROUP_OWNER
+ cmd="启用功能", docs=__doc__, permission=SUPERUSER | GROUP_ADMIN | GROUP_OWNER
)
+
@cur_service_ena.args_parser # type: ignore
-async def _cur_ena_load(bot: Bot,
- event: GroupMessageEvent,
- state: T_State) -> None:
+async def _cur_ena_load(bot: Bot, event: GroupMessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
- quit_list = ['算了', '罢了', '取消']
+ quit_list = ["算了", "罢了", "取消"]
if msg in quit_list:
- await cur_service_ena.finish('好吧...')
+ await cur_service_ena.finish("好吧...")
if not msg:
- await cur_service_ena.reject('请告诉咱目标命令！')
+ await cur_service_ena.reject("请告诉咱目标命令！")
else:
- state['service_e'] = msg
+ state["service_e"] = msg
+
@cur_service_ena.handle()
-async def _cur_ena(bot: Bot,
- event: GroupMessageEvent,
- state: T_State) -> None:
+async def _cur_ena(bot: Bot, event: GroupMessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
if msg:
- state['service_e'] = msg
+ state["service_e"] = msg
-@cur_service_ena.got('service_e', prompt='请告诉咱目标命令！')
-async def _deal_cur_ena(bot: Bot,
- event: GroupMessageEvent,
- state: T_State) -> None:
- cmd = state['service_e']
+
+@cur_service_ena.got("service_e", prompt="请告诉咱目标命令！")
+async def _deal_cur_ena(bot: Bot, event: GroupMessageEvent, state: T_State) -> None:
+ cmd = state["service_e"]
group = str(event.group_id)
sv.control_service(cmd, False, True, group=group)
- await cur_service_ena.finish(f'成功！本群已启用：{cmd}')
+ await cur_service_ena.finish(f"成功！本群已启用：{cmd}")
+
@cur_service_ena.handle()
-async def _refuse_cur_ena(bot: Bot,
- event: PrivateMessageEvent,
- state: T_State) -> None:
- await cur_service_ena.finish('只能在群聊中决定哦...')
+async def _refuse_cur_ena(bot: Bot, event: PrivateMessageEvent, state: T_State) -> None:
+ await cur_service_ena.finish("只能在群聊中决定哦...")
__doc__ = """
@@ -73,46 +67,40 @@ __doc__ = """
"""
cur_service_dis = sv.on_command(
- cmd="禁用功能",
- docs=__doc__,
- permission=SUPERUSER|GROUP_ADMIN|GROUP_OWNER
+ cmd="禁用功能", docs=__doc__, permission=SUPERUSER | GROUP_ADMIN | GROUP_OWNER
)
+
@cur_service_dis.args_parser # type: ignore
-async def _cur_dis_load(bot: Bot,
- event: GroupMessageEvent,
- state: T_State) -> None:
+async def _cur_dis_load(bot: Bot, event: GroupMessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
- quit_list = ['算了', '罢了', '取消']
+ quit_list = ["算了", "罢了", "取消"]
if msg in quit_list:
- await cur_service_dis.finish('好吧...')
+ await cur_service_dis.finish("好吧...")
if not msg:
- await cur_service_dis.reject('请告诉咱目标命令！')
+ await cur_service_dis.reject("请告诉咱目标命令！")
else:
- state['service_d'] = msg
+ state["service_d"] = msg
+
@cur_service_dis.handle()
-async def _cur_dis(bot: Bot,
- event: GroupMessageEvent,
- state: T_State) -> None:
+async def _cur_dis(bot: Bot, event: GroupMessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
if msg:
- state['service_d'] = msg
+ state["service_d"] = msg
-@cur_service_dis.got('service_d', prompt='请告诉咱目标命令！')
-async def _deal_cur_dis(bot: Bot,
- event: GroupMessageEvent,
- state: T_State) -> None:
- cmd = state['service_d']
+
+@cur_service_dis.got("service_d", prompt="请告诉咱目标命令！")
+async def _deal_cur_dis(bot: Bot, event: GroupMessageEvent, state: T_State) -> None:
+ cmd = state["service_d"]
group = str(event.group_id)
sv.control_service(cmd, False, False, group=group)
- await cur_service_dis.finish(f'成功！本群已禁用：{cmd}')
+ await cur_service_dis.finish(f"成功！本群已禁用：{cmd}")
+
@cur_service_dis.handle()
-async def _refuse_cur_dis(bot: Bot,
- event: PrivateMessageEvent,
- state: T_State) -> None:
- await cur_service_dis.finish('只能在群聊中决定哦...')
+async def _refuse_cur_dis(bot: Bot, event: PrivateMessageEvent, state: T_State) -> None:
+ await cur_service_dis.finish("只能在群聊中决定哦...")
__doc__ = """
@@ -124,40 +112,33 @@ __doc__ = """
全局启用 以图搜图
"""
-glo_service_ena = sv.on_command(
- cmd="全局启用",
- docs=__doc__,
- permission=SUPERUSER
-)
+glo_service_ena = sv.on_command(cmd="全局启用", docs=__doc__, permission=SUPERUSER)
+
@glo_service_ena.args_parser # type: ignore
-async def _glo_ena_load(bot: Bot,
- event: MessageEvent,
- state: T_State) -> None:
+async def _glo_ena_load(bot: Bot, event: MessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
- quit_list = ['算了', '罢了', '取消']
+ quit_list = ["算了", "罢了", "取消"]
if msg in quit_list:
- await glo_service_ena.finish('好吧...')
+ await glo_service_ena.finish("好吧...")
if not msg:
- await glo_service_ena.reject('请告诉咱目标命令！')
+ await glo_service_ena.reject("请告诉咱目标命令！")
else:
- state['service_e_g'] = msg
+ state["service_e_g"] = msg
+
@glo_service_ena.handle()
-async def _glo_ena(bot: Bot,
- event: MessageEvent,
- state: T_State) -> None:
+async def _glo_ena(bot: Bot, event: MessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
if msg:
- state['service_e_g'] = msg
+ state["service_e_g"] = msg
+
-@glo_service_ena.got('service_e_g', prompt='请告诉咱目标命令！')
-async def _deal_glo_ena(bot: Bot,
- event: MessageEvent,
- state: T_State) -> None:
- cmd = state['service_e_g']
+@glo_service_ena.got("service_e_g", prompt="请告诉咱目标命令！")
+async def _deal_glo_ena(bot: Bot, event: MessageEvent, state: T_State) -> None:
+ cmd = state["service_e_g"]
sv.control_service(cmd, True, True)
- await glo_service_ena.finish(f'成功！已全局启用：{cmd}')
+ await glo_service_ena.finish(f"成功！已全局启用：{cmd}")
__doc__ = """
@@ -169,37 +150,30 @@ __doc__ = """
全局禁用 以图搜图
"""
-glo_service_dis = sv.on_command(
- cmd="全局禁用",
- docs=__doc__,
- permission=SUPERUSER
-)
+glo_service_dis = sv.on_command(cmd="全局禁用", docs=__doc__, permission=SUPERUSER)
+
@glo_service_dis.args_parser # type: ignore
-async def _glo_dis_load(bot: Bot,
- event: MessageEvent,
- state: T_State) -> None:
+async def _glo_dis_load(bot: Bot, event: MessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
- quit_list = ['算了', '罢了', '取消']
+ quit_list = ["算了", "罢了", "取消"]
if msg in quit_list:
- await glo_service_dis.finish('好吧...')
+ await glo_service_dis.finish("好吧...")
if not msg:
- await glo_service_dis.reject('请告诉咱目标命令！')
+ await glo_service_dis.reject("请告诉咱目标命令！")
else:
- state['service_d_g'] = msg
+ state["service_d_g"] = msg
+
@glo_service_dis.handle()
-async def _glo_dis(bot: Bot,
- event: MessageEvent,
- state: T_State) -> None:
+async def _glo_dis(bot: Bot, event: MessageEvent, state: T_State) -> None:
msg = str(event.message).strip()
if msg:
- state['service_d_g'] = msg
+ state["service_d_g"] = msg
+
-@glo_service_dis.got('service_d_g', prompt='请告诉咱目标命令！')
-async def _deal_glo_dis(bot: Bot,
- event: MessageEvent,
- state: T_State) -> None:
- cmd = state['service_d_g']
+@glo_service_dis.got("service_d_g", prompt="请告诉咱目标命令！")
+async def _deal_glo_dis(bot: Bot, event: MessageEvent, state: T_State) -> None:
+ cmd = state["service_d_g"]
sv.control_service(cmd, True, False)
- await glo_service_dis.finish(f'成功！已全局禁用：{cmd}')
+ await glo_service_dis.finish(f"成功！已全局禁用：{cmd}")
Powered by cgit, Theme cgit-dark created by Ted Yin, updated by Mole Shang.