🔥 小型重构

src/utils/satori_utils/__init__.py

@@ -0,0 +1,3 @@
+from .user_info import user_infos
+from .count_friends import count_friends
+from .count_groups import count_groups

src/utils/satori_utils/count_friends.py

@@ -0,0 +1,13 @@
+from nonebot.adapters import satori
+
+
+async def count_friends(bot: satori.Bot) -> int:
+    cnt: int = 0
+
+    friend_response = await bot.friend_list()
+    while friend_response.next is not None:
+        cnt += len(friend_response.data)
+        friend_response = await bot.friend_list(next_token=friend_response.next)
+
+    cnt += len(friend_response.data)
+    return cnt - 1

src/utils/satori_utils/count_groups.py

@@ -0,0 +1,13 @@
+from nonebot.adapters import satori
+
+
+async def count_groups(bot: satori.Bot) -> int:
+    cnt: int = 0
+
+    group_response = await bot.guild_list()
+    while group_response.next is not None:
+        cnt += len(group_response.data)
+        group_response = await bot.friend_list(next_token=group_response.next)
+
+    cnt += len(group_response.data)
+    return cnt - 1

src/utils/satori_utils/user_info.py

@@ -0,0 +1,64 @@
+import nonebot
+
+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):
+        nonebot.logger.info("Update user info from friends")
+        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
+
+        nonebot.logger.info("Finish update user info")
+
+    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) -> bool:
+        """
+        向用户信息数据库中添加/修改一项，返回值仅代表数据是否变更，不代表操作是否成功
+        Args:
+            user: 要加入数据库的用户
+
+        Returns: 当数据库中用户信息发生变化时返回 True, 否则返回 False
+
+        """
+        try:
+            old_user: User = self.user_infos[str(user.id)]
+            attr_edited = False
+            if user.name is not None:
+                if old_user.name != user.name:
+                    attr_edited = True
+                    self.user_infos[str(user.id)].name = user.name
+            if user.nick is not None:
+                if old_user.nick != user.nick:
+                    attr_edited = True
+                    self.user_infos[str(user.id)].nick = user.nick
+            if user.avatar is not None:
+                if old_user.avatar != user.avatar:
+                    attr_edited = True
+                    self.user_infos[str(user.id)].avatar = user.avatar
+            return attr_edited
+        except KeyError:
+            self.user_infos[str(user.id)] = user
+            return True
+
+    def __init__(self):
+        pass
+
+
+user_infos = UserInfo()