mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
import time
|
|
|
|
from nonebot_plugin_alconna import At
|
|
from nonebot_plugin_uninfo import Uninfo
|
|
|
|
from zhenxun.models.level_user import LevelUser
|
|
from zhenxun.models.plugin_info import PluginInfo
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.utils import EntityIDs, get_entity_ids
|
|
|
|
from .config import LOGGER_COMMAND, WARNING_THRESHOLD
|
|
from .context import PermissionContext
|
|
from .data_provider import DEFAULT_PERMISSION_DATA_PROVIDER, LevelUserSnapshot
|
|
from .exception import SkipPluginException
|
|
|
|
|
|
async def auth_admin(
|
|
plugin: PluginInfo,
|
|
session: Uninfo,
|
|
cached_levels: tuple[
|
|
LevelUser | LevelUserSnapshot | None, LevelUser | LevelUserSnapshot | None
|
|
]
|
|
| None = None,
|
|
*,
|
|
context: PermissionContext | None = None,
|
|
entity: EntityIDs | None = None,
|
|
):
|
|
"""管理员命令 个人权限
|
|
|
|
参数:
|
|
plugin: PluginInfo
|
|
session: Uninfo
|
|
"""
|
|
start_time = time.time()
|
|
|
|
if not plugin.admin_level:
|
|
return
|
|
|
|
try:
|
|
if context is not None:
|
|
entity = context.entity
|
|
if cached_levels is None:
|
|
cached_levels = context.admin_levels
|
|
if entity is None:
|
|
entity = get_entity_ids(session)
|
|
|
|
global_user: LevelUser | LevelUserSnapshot | None = None
|
|
group_users: LevelUser | LevelUserSnapshot | None = None
|
|
|
|
if cached_levels is not None:
|
|
global_user, group_users = cached_levels
|
|
else:
|
|
(
|
|
global_user,
|
|
group_users,
|
|
) = await DEFAULT_PERMISSION_DATA_PROVIDER.get_admin_levels(
|
|
entity.user_id, entity.group_id
|
|
)
|
|
|
|
user_level = global_user.user_level if global_user else 0
|
|
if entity.group_id and group_users:
|
|
user_level = max(user_level, group_users.user_level)
|
|
|
|
if user_level < plugin.admin_level:
|
|
raise SkipPluginException(
|
|
f"{plugin.name}({plugin.module}) 管理员权限不足...",
|
|
tip_message=[
|
|
At(flag="user", target=entity.user_id),
|
|
f"你的权限不足喔，该功能需要的权限等级: {plugin.admin_level}",
|
|
],
|
|
tip_check_tag=entity.user_id,
|
|
tip_background=True,
|
|
)
|
|
elif global_user:
|
|
if global_user.user_level < plugin.admin_level:
|
|
raise SkipPluginException(
|
|
f"{plugin.name}({plugin.module}) 管理员权限不足...",
|
|
tip_message=(
|
|
f"你的权限不足喔，该功能需要的权限等级: {plugin.admin_level}"
|
|
),
|
|
tip_background=True,
|
|
)
|
|
finally:
|
|
# 记录执行时间
|
|
elapsed = time.time() - start_time
|
|
if elapsed > WARNING_THRESHOLD: # 记录耗时超过500ms的检查
|
|
logger.warning(
|
|
f"auth_admin 耗时: {elapsed:.3f}s, plugin={plugin.module}",
|
|
LOGGER_COMMAND,
|
|
session=session,
|
|
)
|