mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 16:50:02 +08:00
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
import time
|
|
|
|
from nonebot.adapters import Bot, Event
|
|
from nonebot.exception import IgnoredException
|
|
from nonebot.matcher import Matcher
|
|
from nonebot.message import run_postprocessor, run_preprocessor
|
|
from nonebot_plugin_alconna import UniMsg
|
|
from nonebot_plugin_uninfo import Uninfo
|
|
|
|
from zhenxun.services.auth_snapshot.checker import optimized_auth_checker
|
|
from zhenxun.services.auth_snapshot.exception import (
|
|
PermissionExemption,
|
|
SkipPluginException,
|
|
)
|
|
from zhenxun.services.log import logger
|
|
|
|
from .auth.auth_limit import LimitManager
|
|
from .auth.config import LOGGER_COMMAND
|
|
|
|
|
|
# # 权限检测
|
|
@run_preprocessor
|
|
async def _(matcher: Matcher, event: Event, bot: Bot, session: Uninfo, message: UniMsg):
|
|
start_time = time.time()
|
|
# await _auth_checker.check(
|
|
# matcher,
|
|
# event,
|
|
# bot,
|
|
# session,
|
|
# message,
|
|
# )
|
|
try:
|
|
await optimized_auth_checker.check(matcher, event, bot, session, message)
|
|
except SkipPluginException as e:
|
|
logger.info(str(e), LOGGER_COMMAND, session=session)
|
|
raise IgnoredException(str(e))
|
|
except PermissionExemption as e:
|
|
logger.info(
|
|
str(e) or "超级用户跳过权限检测...", LOGGER_COMMAND, session=session
|
|
)
|
|
raise IgnoredException(str(e))
|
|
except Exception as e:
|
|
logger.error(f"权限检测异常: {e}", LOGGER_COMMAND, session=session, e=e)
|
|
raise SkipPluginException("权限检测异常") from e
|
|
logger.debug(f"权限检测耗时：{time.time() - start_time}秒", LOGGER_COMMAND)
|
|
|
|
|
|
# 解除命令block阻塞
|
|
@run_postprocessor
|
|
async def _(matcher: Matcher, session: Uninfo):
|
|
user_id = session.user.id
|
|
group_id = None
|
|
channel_id = None
|
|
if session.group:
|
|
if session.group.parent:
|
|
group_id = session.group.parent.id
|
|
channel_id = session.group.id
|
|
else:
|
|
group_id = session.group.id
|
|
if user_id and matcher.plugin:
|
|
module = matcher.plugin.name
|
|
LimitManager.unblock(module, user_id, group_id, channel_id)
|