mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
c8832bd1c9
* 🔧 启用一些 ruff 的新规则 * ➕ 添加开发依赖 nonebot-adapter-qq * ➕ 添加依赖 nonebot-plugin-session * ➕ 添加依赖 nonebot-plugin-session-orm * 🔧 忽略 ruff 规则 ISC001 format 冲突风险 * 🚨 修复 ruff 警报 * ♻️ 重构！ * ♻️ 恢复定时获取 TetraLeague 数据的功能 * ✨ 统一处理需要捕获的错误 * ✨ 记录用户触发的指令
55 lines
1.7 KiB
Python
55 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:
|
|
msg = "Can't get avatar"
|
|
raise TypeError(msg)
|
|
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:
|
|
msg = "Can't get avatar format"
|
|
raise TypeError(msg)
|
|
return f'data:{Image.MIME[avatar_format]};base64,{b64encode(bot_avatar).decode()}'
|
|
return bot_avatar
|