zhenxun_bot/zhenxun/builtin_plugins/hooks/auth/auth_group.py

import time

from nonebot_plugin_alconna import UniMsg

from zhenxun.models.group_console import GroupConsole
from zhenxun.models.plugin_info import PluginInfo
from zhenxun.services.log import logger

from .config import LOGGER_COMMAND, WARNING_THRESHOLD, SwitchEnum
from .exception import SkipPluginException


async def auth_group(
    plugin: PluginInfo,
    group: GroupConsole | None,
    message: UniMsg,
    group_id: str | None,
):
    """群黑名单检测 群总开关检测

    参数:
        plugin: PluginInfo
        group: GroupConsole
        message: UniMsg
    """
    if not group_id:
        return

    start_time = time.time()

    try:
        text = message.extract_plain_text()

        if not group:
            raise SkipPluginException("群组信息不存在...")
        if group.level < 0:
            raise SkipPluginException("群组黑名单, 目标群组群权限权限-1...")
        if text.strip() != SwitchEnum.ENABLE and not group.status:
            raise SkipPluginException("群组休眠状态...")
        if plugin.level > group.level:
            raise SkipPluginException(
                f"{plugin.name}({plugin.module}) 群等级限制，"
                f"该功能需要的群等级: {plugin.level}..."
            )
    finally:
        # 记录执行时间
        elapsed = time.time() - start_time
        if elapsed > WARNING_THRESHOLD:  # 记录耗时超过500ms的检查
            logger.warning(
                f"auth_group 耗时: {elapsed:.3f}s, plugin={plugin.module}",
                LOGGER_COMMAND,
                group_id=group_id,
            )