mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
5d92ccd3b0
* 性能优化 * 代码改进 * 优化浏览器代际切换逻辑 * 统一缓存与生命周期 * 添加aiomysql依赖 * 优化插件路径处理逻辑，简化条件判断；在虚拟环境包管理器中添加编码和错误处理参数以增强稳定性 * 🚨 auto fix by pre-commit hooks * 优化Windows下的关闭逻辑 * 代码优化 * bugfix:修复配置重载问题 * bugfix:修复插件加载启动竞态问题 * 收敛事件入口和权限上下文 * 优化 Windows launcher 关闭重启兜底 --------- Co-authored-by: HibiKier <775757368@qq.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
90 lines
3.0 KiB
Python
90 lines
3.0 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.cache.runtime_cache import LevelUserMemoryCache, LevelUserSnapshot
|
|
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 .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 LevelUserMemoryCache.get_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"你的权限不足喔，该功能需要的权限等级: "
|
|
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,
|
|
)
|