🎨 优化检测代码结构

zhenxun/builtin_plugins/hooks/auth_checker.py

@@ -1,3 +1,5 @@
+import contextlib
+
 from nonebot.adapters import Bot, Event
 from nonebot.exception import IgnoredException
 from nonebot.matcher import Matcher
@@ -7,8 +9,8 @@ from tortoise.exceptions import IntegrityError
 
 from zhenxun.models.plugin_info import PluginInfo
 from zhenxun.models.user_console import UserConsole
-from zhenxun.services.log import logger
 from zhenxun.services.cache import Cache
+from zhenxun.services.log import logger
 from zhenxun.utils.enum import (
     CacheType,
     GoldHandle,
@@ -53,16 +55,22 @@ async def auth(
             group_id = session.group.id
     is_ignore = False
     cost_gold = 0
-    try:
+    with contextlib.suppress(ImportError):
         from nonebot.adapters.onebot.v11 import PokeNotifyEvent
 
         if matcher.type == "notice" and not isinstance(event, PokeNotifyEvent):
             """过滤除poke外的notice"""
             return
-    except ImportError:
-        pass
     user_cache = Cache[UserConsole](CacheType.USERS)
     if matcher.plugin and (module := matcher.plugin.name):
+        plugin = await Cache[PluginInfo](CacheType.PLUGINS).get(module)
+        if not plugin:
+            return logger.debug(f"插件:{module} 数据不存在，已跳过权限检查...")
+        if plugin.plugin_type == PluginType.HIDDEN:
+            return logger.debug(
+                f"插件: {plugin.name}:{plugin.module} 为HIDDEN，已跳过权限检查..."
+            )
+        user = None
         try:
             user = await user_cache.get(session.user.id)
         except IntegrityError as e:
@@ -72,15 +80,10 @@ async def auth(
                 session=session,
                 e=e,
             )
-            return
-        plugin = await Cache[PluginInfo](CacheType.PLUGINS).get(module)
-        if user and plugin:
-            if plugin.plugin_type == PluginType.HIDDEN:
-                logger.debug(
-                    f"插件: {plugin.name}:{plugin.module} "
-                    "为HIDDEN，已跳过权限检查..."
+        if not user:
+            return logger.debug(
+                "用户数据不存在，已跳过权限检查...", "AuthChecker", session=session
             )
-                return
         try:
             cost_gold = await auth_cost(user, plugin, session)
             if session.user.id in bot.config.superusers: