mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
716e392a3a
* 🔥 删除现有模板 * ✨ 自动克隆模板仓库 * 🔥 删除 identicon 相关代码 * 🚚 修改静态文件路径 * ✨ 使用新模板进行渲染 * ✨ 每次渲染都获取一次模板, 以应对实时更新 * ✨ TETR.IO 绑定图使用新模板 * 🚚 修改网络路径 * ✨ TOP 绑定图使用新模板 * ✨ TOS 绑定图使用新模板 * 🐛 防止截图超时 * 🐛 Pydantic V1 会把 float 转换成 int * ✏️ 模板字段名写错了 * ✨ 兼容 Pydantic V1 * ✨ TETR.IO 查询图使用新模板 * 🐛 在查询的用户没有历史记录时不去查询更多记录
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
from base64 import b64encode
|
|
from io import BytesIO
|
|
from typing import Literal, overload
|
|
|
|
from nonebot_plugin_userinfo import UserInfo # type: ignore[import-untyped]
|
|
from PIL import Image
|
|
|
|
|
|
@overload
|
|
async def get_avatar(user: UserInfo, scheme: Literal['Data URI'], default: str | None) -> str:
|
|
"""获取用户头像的指定格式
|
|
|
|
Args:
|
|
user (UserInfo): 要获取的用户
|
|
scheme (Literal['Data URI']): 格式
|
|
default (str | None): 获取不到时的默认值
|
|
|
|
Raises:
|
|
TypeError: Can't get avatar: 当获取不到头像并且没有设置默认值时抛出
|
|
TypeError: Can't get avatar format: 当获取到的头像无法识别格式时抛出
|
|
|
|
Returns:
|
|
str: Data URI 格式的头像
|
|
"""
|
|
|
|
|
|
@overload
|
|
async def get_avatar(user: UserInfo, scheme: Literal['bytes'], default: str | None) -> bytes:
|
|
"""获取用户头像的指定格式
|
|
|
|
Args:
|
|
user (UserInfo): 要获取的用户
|
|
scheme (Literal['bytes']): 格式
|
|
default (str | None): 获取不到时的默认值
|
|
|
|
Returns:
|
|
bytes: bytes 格式的头像
|
|
"""
|
|
|
|
|
|
async def get_avatar(user: UserInfo, scheme: Literal['Data URI', 'bytes'], default: str | None) -> str | bytes:
|
|
if user.user_avatar is None:
|
|
if default is None:
|
|
raise TypeError("Can't get avatar")
|
|
return default
|
|
bot_avatar = await user.user_avatar.get_image()
|
|
if scheme == 'Data URI':
|
|
avatar_format = Image.open(BytesIO(bot_avatar)).format
|
|
if avatar_format is None:
|
|
raise TypeError("Can't get avatar format")
|
|
return f'data:{Image.MIME[avatar_format]};base64,{b64encode(bot_avatar).decode()}'
|
|
return bot_avatar
|