使用satori时维护一个有昵称的用户列表

get_plugin_session_enable 判断当前使用的适配器

liteyuki/utils/base/ly_typing.py

@@ -1,7 +1,8 @@
 from nonebot.adapters.onebot import v11, v12
+from nonebot.adapters import satori
 
-T_Bot = v11.Bot | v12.Bot
-T_GroupMessageEvent = v11.GroupMessageEvent | v12.GroupMessageEvent
-T_PrivateMessageEvent = v11.PrivateMessageEvent | v12.PrivateMessageEvent
-T_MessageEvent = v11.MessageEvent | v12.MessageEvent
-T_Message = v11.Message | v12.Message
+T_Bot = v11.Bot | v12.Bot | satori.Bot
+T_GroupMessageEvent = v11.GroupMessageEvent | v12.GroupMessageEvent | satori.MessageEvent
+T_PrivateMessageEvent = v11.PrivateMessageEvent | v12.PrivateMessageEvent | satori.MessageEvent
+T_MessageEvent = v11.MessageEvent | v12.MessageEvent | satori.MessageEvent
+T_Message = v11.Message | v12.Message | satori.Message

liteyuki/utils/satori_utils/__init__.py

@@ -0,0 +1 @@
+from .user_info import user_infos

liteyuki/utils/satori_utils/user_info.py

@@ -0,0 +1,35 @@
+from nonebot.adapters import satori
+from nonebot.adapters.satori.models import User
+
+
+class UserInfo:
+    user_infos: dict = {}
+
+    async def load_friends(self, bot: satori.Bot):
+        print("LoadFriends")
+        friend_response = await bot.friend_list()
+        while friend_response.next is not None:
+            for i in friend_response.data:
+                i: User = i
+                self.user_infos[str(i.id)] = i
+            friend_response = await bot.friend_list(next_token=friend_response.next)
+
+        for i in friend_response.data:
+            i: User = i
+            self.user_infos[str(i.id)] = i
+            print(i)
+
+    async def get(self, uid: int | str) -> User | None:
+        try:
+            return self.user_infos[str(uid)]
+        except KeyError:
+            return None
+
+    async def put(self, user: User):
+        self.user_infos[str(user.id)] = user
+
+    def __init__(self):
+        pass
+
+
+user_infos = UserInfo()