mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-03 10:40:02 +08:00
refactor code
This commit is contained in:
hibiki
+74
-83
@@ -1,13 +1,25 @@
|
||||
from nonebot.matcher import Matcher
|
||||
from nonebot.message import run_preprocessor, IgnoredException
|
||||
from nonebot.typing import T_State
|
||||
from nonebot.adapters.cqhttp import Bot, MessageEvent, PrivateMessageEvent, GroupMessageEvent
|
||||
from configs.config import BAN_RESULT, admin_plugins_auth, MALICIOUS_BAN_TIME, MALICIOUS_CHECK_TIME, MALICIOUS_BAN_COUNT
|
||||
from nonebot.adapters.cqhttp import (
|
||||
Bot,
|
||||
MessageEvent,
|
||||
PrivateMessageEvent,
|
||||
GroupMessageEvent,
|
||||
)
|
||||
from configs.config import (
|
||||
BAN_RESULT,
|
||||
admin_plugins_auth,
|
||||
MALICIOUS_BAN_TIME,
|
||||
MALICIOUS_CHECK_TIME,
|
||||
MALICIOUS_BAN_COUNT,
|
||||
)
|
||||
from models.ban_user import BanUser
|
||||
from utils.utils import is_number, static_flmt, BanCheckLimiter
|
||||
from utils.init_result import at
|
||||
from utils.message_builder import at
|
||||
from services.log import logger
|
||||
from models.level_user import LevelUser
|
||||
|
||||
try:
|
||||
import ujson as json
|
||||
except ModuleNotFoundError:
|
||||
@@ -19,32 +31,39 @@ except ModuleNotFoundError:
|
||||
async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
|
||||
if not isinstance(event, MessageEvent):
|
||||
return
|
||||
if matcher.type == 'message' and matcher.priority not in [1, 9]:
|
||||
if await BanUser.isban(event.user_id) and str(event.user_id) not in bot.config.superusers:
|
||||
if matcher.type == "message" and matcher.priority not in [1, 9]:
|
||||
if (
|
||||
await BanUser.is_ban(event.user_id)
|
||||
and str(event.user_id) not in bot.config.superusers
|
||||
):
|
||||
time = await BanUser.check_ban_time(event.user_id)
|
||||
if is_number(time):
|
||||
time = abs(int(time))
|
||||
if time < 60:
|
||||
time = str(time) + ' 秒'
|
||||
time = str(time) + " 秒"
|
||||
else:
|
||||
time = str(int(time / 60)) + ' 分'
|
||||
time = str(int(time / 60)) + " 分"
|
||||
else:
|
||||
time = str(time) + ' 分'
|
||||
if event.message_type == 'group':
|
||||
time = str(time) + " 分"
|
||||
if event.message_type == "group":
|
||||
if not static_flmt.check(event.user_id):
|
||||
raise IgnoredException('用户处于黑名单中')
|
||||
raise IgnoredException("用户处于黑名单中")
|
||||
static_flmt.start_cd(event.user_id)
|
||||
if matcher.priority != 9:
|
||||
await bot.send_group_msg(group_id=event.group_id,
|
||||
message=at(event.user_id) + BAN_RESULT + f' 在..在 {time}后才会理你喔')
|
||||
await bot.send_group_msg(
|
||||
group_id=event.group_id,
|
||||
message=at(event.user_id) + BAN_RESULT + f" 在..在 {time}后才会理你喔",
|
||||
)
|
||||
else:
|
||||
if not static_flmt.check(event.user_id):
|
||||
raise IgnoredException('用户处于黑名单中')
|
||||
raise IgnoredException("用户处于黑名单中")
|
||||
static_flmt.start_cd(event.user_id)
|
||||
if matcher.priority != 9:
|
||||
await bot.send_private_msg(user_id=event.user_id,
|
||||
message=at(event.user_id) + BAN_RESULT + f' 在..在 {time}后才会理你喔')
|
||||
raise IgnoredException('用户处于黑名单中')
|
||||
await bot.send_private_msg(
|
||||
user_id=event.user_id,
|
||||
message=at(event.user_id) + BAN_RESULT + f" 在..在 {time}后才会理你喔",
|
||||
)
|
||||
raise IgnoredException("用户处于黑名单中")
|
||||
|
||||
|
||||
_blmt = BanCheckLimiter(MALICIOUS_CHECK_TIME, MALICIOUS_BAN_COUNT)
|
||||
@@ -55,104 +74,76 @@ _blmt = BanCheckLimiter(MALICIOUS_CHECK_TIME, MALICIOUS_BAN_COUNT)
|
||||
async def _(matcher: Matcher, bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||
if not isinstance(event, MessageEvent):
|
||||
return
|
||||
if matcher.type == 'message' and matcher.priority not in [1, 9]:
|
||||
if matcher.type == "message" and matcher.priority not in [1, 9]:
|
||||
if state["_prefix"]["raw_command"]:
|
||||
# print(state["_prefix"]["raw_command"])
|
||||
if _blmt.check(f'{event.user_id}{state["_prefix"]["raw_command"]}'):
|
||||
if await BanUser.ban(event.user_id, 9, MALICIOUS_BAN_TIME * 60):
|
||||
logger.info(f'USER {event.user_id} 触发了恶意触发检测')
|
||||
logger.info(f"USER {event.user_id} 触发了恶意触发检测")
|
||||
# await update_img.finish('检测到恶意触发命令，您将被封禁 30 分钟', at_sender=True)
|
||||
if event.message_type == 'group':
|
||||
if event.message_type == "group":
|
||||
if not static_flmt.check(event.user_id):
|
||||
return
|
||||
static_flmt.start_cd(event.user_id)
|
||||
await bot.send_group_msg(group_id=event.group_id,
|
||||
message=at(event.user_id) + '检测到恶意触发命令，您将被封禁 30 分钟')
|
||||
await bot.send_group_msg(
|
||||
group_id=event.group_id,
|
||||
message=at(event.user_id) + "检测到恶意触发命令，您将被封禁 30 分钟",
|
||||
)
|
||||
else:
|
||||
if not static_flmt.check(event.user_id):
|
||||
return
|
||||
static_flmt.start_cd(event.user_id)
|
||||
await bot.send_private_msg(user_id=event.user_id,
|
||||
message=at(event.user_id) + '检测到恶意触发命令，您将被封禁 30 分钟')
|
||||
raise IgnoredException('检测到恶意触发命令')
|
||||
await bot.send_private_msg(
|
||||
user_id=event.user_id,
|
||||
message=at(event.user_id) + "检测到恶意触发命令，您将被封禁 30 分钟",
|
||||
)
|
||||
raise IgnoredException("检测到恶意触发命令")
|
||||
_blmt.add(f'{event.user_id}{state["_prefix"]["raw_command"]}')
|
||||
|
||||
|
||||
# 权限检测
|
||||
@run_preprocessor
|
||||
async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
|
||||
if not isinstance(event, MessageEvent):
|
||||
return
|
||||
if await BanUser.isban(event.user_id):
|
||||
if not isinstance(event, MessageEvent) or await BanUser.is_ban(event.user_id):
|
||||
return
|
||||
if matcher.module in admin_plugins_auth.keys() and matcher.priority not in [1, 9]:
|
||||
if event.message_type == 'group':
|
||||
if not await LevelUser.check_level(event.user_id, event.group_id, admin_plugins_auth[matcher.module]):
|
||||
await bot.send_group_msg(group_id=event.group_id,
|
||||
message=f'{at(event.user_id)}你的权限不足喔，该功能需要的权限等级：'
|
||||
f'{admin_plugins_auth[matcher.module]}')
|
||||
raise IgnoredException('权限不足')
|
||||
if event.message_type == "group":
|
||||
if not await LevelUser.check_level(
|
||||
event.user_id, event.group_id, admin_plugins_auth[matcher.module]
|
||||
):
|
||||
await bot.send_group_msg(
|
||||
group_id=event.group_id,
|
||||
message=f"{at(event.user_id)}你的权限不足喔，该功能需要的权限等级："
|
||||
f"{admin_plugins_auth[matcher.module]}",
|
||||
)
|
||||
raise IgnoredException("权限不足")
|
||||
else:
|
||||
if not await LevelUser.check_level(event.user_id, 0, admin_plugins_auth[matcher.module]):
|
||||
await bot.send_private_msg(user_id=event.user_id,
|
||||
message=f'你的权限不足喔，该功能需要的权限等级：{admin_plugins_auth[matcher.module]}')
|
||||
raise IgnoredException('权限不足')
|
||||
if not await LevelUser.check_level(
|
||||
event.user_id, 0, admin_plugins_auth[matcher.module]
|
||||
):
|
||||
await bot.send_private_msg(
|
||||
user_id=event.user_id,
|
||||
message=f"你的权限不足喔，该功能需要的权限等级：{admin_plugins_auth[matcher.module]}",
|
||||
)
|
||||
raise IgnoredException("权限不足")
|
||||
|
||||
|
||||
# 为什么AI会自己和自己聊天
|
||||
@run_preprocessor
|
||||
async def _(matcher: Matcher, bot: Bot, event: PrivateMessageEvent, state: T_State):
|
||||
if not isinstance(event, MessageEvent):
|
||||
return
|
||||
if event.user_id == int(bot.self_id):
|
||||
raise IgnoredException('为什么AI会自己和自己聊天')
|
||||
raise IgnoredException("为什么AI会自己和自己聊天")
|
||||
|
||||
|
||||
# 有命令就别说话了
|
||||
@run_preprocessor
|
||||
async def _(matcher: Matcher, bot: Bot, event: MessageEvent, state: T_State):
|
||||
if matcher.type == 'message':
|
||||
if state["_prefix"]["raw_command"] and matcher.module == 'ai':
|
||||
raise IgnoredException('有命令就别说话了')
|
||||
|
||||
|
||||
# _ulmt = UserExistLimiter()
|
||||
#
|
||||
#
|
||||
# # 是否有命令正在处理
|
||||
# @run_preprocessor
|
||||
# async def _(matcher: Matcher, bot: Bot, event: Event, state: T_State):
|
||||
# if matcher.module == 'admin_bot_manage':
|
||||
# return
|
||||
# if event.user_id == bot.self_id:
|
||||
# raise IgnoredException('自己和自己聊天？')
|
||||
# print(state["_prefix"]["raw_command"])
|
||||
# print(matcher.type)
|
||||
# if (event.is_tome() and (state["_prefix"]["raw_command"] and matcher.type == 'message')
|
||||
# and matcher.module == 'ai') or not static_group_dict[event.group_id]['总开关']:
|
||||
# raise IgnoredException('Ai给爷爬？')
|
||||
# # if matcher.module in ['ai']:
|
||||
# # raise IgnoredException('Ai给爷爬？')
|
||||
# if matcher.type == 'message' and matcher.priority not in [1, 8, 9]:
|
||||
# if _ulmt.check(event.user_id):
|
||||
# if event.message_type == 'group':
|
||||
# await bot.send_group_msg(group_id=event.group_id, message=at(event.user_id) + '您有命令正在处理，等一哈！')
|
||||
# else:
|
||||
# await bot.send_private_msg(user_id=event.user_id, message='您有命令正在处理，等一哈！')
|
||||
# raise IgnoredException('有命令正在处理')
|
||||
# _ulmt.set_True(event.user_id)
|
||||
#
|
||||
#
|
||||
# # 结束正在处理的命令
|
||||
# @run_postprocessor
|
||||
# async def _(matcher: Matcher, bot: Bot, event: Event, state: T_State):
|
||||
# _ulmt.set_False(event.user_id)
|
||||
#
|
||||
#
|
||||
# # 结束正在处理的命令
|
||||
# @event_preprocessor
|
||||
# async def _(bot: Bot, event: Event, state: T_State):
|
||||
# try:
|
||||
# _ulmt.set_False(event.user_id)
|
||||
# except AttributeError:
|
||||
# pass
|
||||
if not isinstance(event, MessageEvent):
|
||||
return
|
||||
if matcher.type == "message":
|
||||
if state["_prefix"]["raw_command"] and matcher.module == "ai":
|
||||
raise IgnoredException("有命令就别说话了")
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user