mirror of
https://github.com/LiteyukiStudio/LiteyukiBot.git
synced 2025-07-28 08:00:56 +00:00
使用satori时维护一个有昵称的用户列表
get_plugin_session_enable 判断当前使用的适配器
This commit is contained in:
35
liteyuki/utils/satori_utils/user_info.py
Normal file
35
liteyuki/utils/satori_utils/user_info.py
Normal file
@ -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()
|
Reference in New Issue
Block a user