mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
Copaan
8afc8f8673
* bugfix:更换数据库初始化超时路径以修复连接超时问题 * 移除部分观测链路 * 细节修改 * 权限检查细节修改2 * bugfix:修复金币懒加载造成插件金币消耗不了的问题 * bugfix:整理鉴权逻辑 * 完善缓存系统 * 优化官端使用 * bugfix:增加官机和频道的前导统一剥离，新增获取群列表自愈 * bugfix:修复导入问题
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import time
|
|
|
|
from zhenxun.models.bot_console import BotConsole
|
|
from zhenxun.models.plugin_info import PluginInfo
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.common_utils import CommonUtils
|
|
|
|
from .config import LOGGER_COMMAND, WARNING_THRESHOLD
|
|
from .context import PermissionContext
|
|
from .data_provider import DEFAULT_PERMISSION_DATA_PROVIDER, BotSnapshot
|
|
from .exception import SkipPluginException
|
|
|
|
|
|
async def auth_bot(
|
|
plugin: PluginInfo,
|
|
bot_id: str,
|
|
bot_data: BotConsole | BotSnapshot | None = None,
|
|
skip_fetch: bool = False,
|
|
allow_sleep_bypass: bool = False,
|
|
*,
|
|
context: PermissionContext | None = None,
|
|
):
|
|
"""bot层面的权限检查
|
|
|
|
参数:
|
|
plugin: PluginInfo
|
|
bot_id: bot id
|
|
|
|
异常:
|
|
SkipPluginException: 忽略插件
|
|
SkipPluginException: 忽略插件
|
|
"""
|
|
start_time = time.time()
|
|
|
|
try:
|
|
provider = DEFAULT_PERMISSION_DATA_PROVIDER
|
|
if context is not None:
|
|
bot_id = context.event.bot_id
|
|
bot_data = context.bot_data
|
|
bot: BotConsole | BotSnapshot | None = bot_data
|
|
if bot is None and not skip_fetch:
|
|
bot = await provider.get_bot(bot_id)
|
|
|
|
if bot is None:
|
|
raise SkipPluginException("Bot不存在，阻断权限检测...")
|
|
|
|
if not bot.status and not allow_sleep_bypass:
|
|
raise SkipPluginException("Bot休眠中阻断权限检测...")
|
|
|
|
if CommonUtils.format(plugin.module) in bot.block_plugins:
|
|
raise SkipPluginException(
|
|
f"Bot插件 {plugin.name}({plugin.module}) 权限检查结果为关闭..."
|
|
)
|
|
finally:
|
|
elapsed = time.time() - start_time
|
|
if elapsed > WARNING_THRESHOLD:
|
|
logger.warning(
|
|
f"auth_bot 耗时: {elapsed:.3f}s, "
|
|
f"bot_id={bot_id}, plugin={plugin.module}",
|
|
LOGGER_COMMAND,
|
|
)
|