forked from LiteyukiStudio/LiteyukiBot
36 lines
949 B
Python
36 lines
949 B
Python
|
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()
|