✨ 部分功能适配QQ BOT (#1701)

zhenxun/utils/rules.py

@@ -1,10 +1,12 @@
 from nonebot.internal.rule import Rule
 from nonebot.adapters import Bot, Event
 from nonebot.permission import SUPERUSER
-from nonebot_plugin_session import EventSession, SessionLevel
+from nonebot_plugin_uninfo import Uninfo
+from nonebot_plugin_session import EventSession
 
 from zhenxun.configs.config import Config
 from zhenxun.models.level_user import LevelUser
+from zhenxun.utils.platform import PlatformUtils
 
 
 def admin_check(a: int | str, key: str | None = None) -> Rule:
@@ -19,33 +21,38 @@ def admin_check(a: int | str, key: str | None = None) -> Rule:
         Rule: Rule
     """
 
-    async def _rule(bot: Bot, event: Event, session: EventSession) -> bool:
+    async def _rule(bot: Bot, event: Event, session: Uninfo) -> bool:
         if await SUPERUSER(bot, event):
             return True
-        if session.id1 and session.id2:
+        if PlatformUtils.is_qbot(session):
+            """官bot接口，放弃所有权限检查"""
+            return False
+        if session.id and session.group:
             level = a
             if isinstance(a, str) and key:
                 level = Config.get_config(a, key)
             if level is not None:
                 return bool(
-                    await LevelUser.check_level(session.id1, session.id2, int(level))
+                    await LevelUser.check_level(
+                        session.id, session.group.id, int(level)
+                    )
                 )
         return False
 
     return Rule(_rule)
 
 
-def ensure_group(session: EventSession) -> bool:
+def ensure_group(session: Uninfo) -> bool:
     """
     是否在群聊中
 
     参数:
-        session: session
+        session: Uninfo
 
     返回:
         bool: bool
     """
-    return session.level in [SessionLevel.LEVEL2, SessionLevel.LEVEL3]
+    return bool(session.group)
 
 
 def ensure_private(session: EventSession) -> bool: