Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

✨ 添加并修复webui对kook、discord的头像显示

Browse Source
This commit is contained in:
Hanako 2025-07-07 23:37:31 +08:00
parent 1e7ae38684
commit 5ec4baf089
3 changed files with 42 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
zhenxun/builtin_plugins/web_ui/api/tabs/dashboard/data_source.py
View File

@ -62,7 +62,12 @@ class ApiDataSource:
)
else:
nickname = bot.self_id
ava_url = ""
ava_url = (
await PlatformUtils.get_user_avatar_url(
bot.self_id, platform
)
or ""
)
bot_info = BotInfo(
self_id=bot.self_id, nickname=nickname, ava_url=ava_url, platform=platform
)

6
zhenxun/builtin_plugins/web_ui/api/tabs/main/data_source.py
View File

@ -76,11 +76,15 @@ class ApiDataSource:
login_info = await bot.get_login_info()
except Exception as e:
logger.warning("调用接口get_login_info失败", "WebUi", e=e)
platform = PlatformUtils.get_platform(bot) or ""
return TemplateBaseInfo(
bot=bot,
self_id=bot.self_id,
nickname=login_info["nickname"] if login_info else bot.self_id,
ava_url=AVA_URL.format(bot.self_id),
ava_url=await PlatformUtils.get_user_avatar_url(
bot.self_id, platform
)
or AVA_URL.format(bot.self_id),
)
@classmethod

37
zhenxun/utils/platform.py
View File

@ -233,7 +233,7 @@ class PlatformUtils:
return await AsyncHttpx.get_content(url) if url else None
@classmethod
def get_user_avatar_url(
async def get_user_avatar_url(
cls, user_id: str, platform: str, appid: str | None = None
) -> str | None:
"""快捷获取用户头像url
@ -242,12 +242,37 @@ class PlatformUtils:
user_id: 用户id
platform: 平台
"""
if platform != "qq":
return None
if user_id.isdigit():
return f"http://q1.qlogo.cn/g?b=qq&nk={user_id}&s=160"
if platform == "qq":
if user_id.isdigit():
return f"http://q1.qlogo.cn/g?b=qq&nk={user_id}&s=160"
else:
return f"https://q.qlogo.cn/qqapp/{appid}/{user_id}/640"
elif platform == "kaiheila":
if not hasattr(cls,'_kook_cache'):
params = {
'user_id': user_id
}
header={
'Authorization': f"Bot {BotConfig.kaiheila_bots[0].get('token')}"
}
result=await AsyncHttpx.get(url='https://www.kookapp.cn/api/v3/user/view',params=params,headers=header)
cls._kook_cache = result.json()['data']['avatar']
return cls._kook_cache
else:
return cls._kook_cache #使用临时缓存
elif platform == "discord":
if not hasattr(cls,'_discord_cache'):
header={
'Authorization': f"Bot {BotConfig.discord_bots[0].get('token')}"
}
result=await AsyncHttpx.get(url=f'https://discord.com/api/users/{user_id}',headers=header)
cls._discord_cache = f"https://cdn.discordapp.com/avatars/{user_id}/{result.json()['avatar']}.png?size=256"
return cls._discord_cache
else:
return cls._discord_cache #使用临时缓存
else:
return f"https://q.qlogo.cn/qqapp/{appid}/{user_id}/640"
return None
@classmethod
async def get_group_avatar(cls, gid: str, platform: str) -> bytes | None: