mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-02-03 18:45:32 +08:00
f0b1a1c4f5
* ➕ 添加依赖 nonebot-plugin-uninfo * ♻️ 使用 nonebot_plugin_uninfo 替代 nonebot_plugin_session nonebot_plugin_session_orm nonebot_plugin_userinfo * 🗃️ 迁移数据库数据至 nonebot_plugin_uninfo * ➖ 删除依赖 nonebot-plugin-session nonebot-plugin-session-orm nonebot-plugin-userinfo * ➕ 添加依赖 nonebot-session-to-uninfo
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
from base64 import b64encode
|
|
from io import BytesIO
|
|
from typing import Literal, overload
|
|
|
|
from nonebot_plugin_uninfo import User
|
|
from PIL import Image
|
|
from yarl import URL
|
|
|
|
from ..config.config import config
|
|
from .request import Request
|
|
|
|
request = Request(config.tetris.proxy.main)
|
|
|
|
|
|
@overload
|
|
async def get_avatar(user: User, scheme: Literal['Data URI'], default: str | None) -> str:
|
|
"""获取用户头像的指定格式
|
|
|
|
Args:
|
|
user (User): 要获取的用户
|
|
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: User, scheme: Literal['bytes'], default: str | None) -> bytes:
|
|
"""获取用户头像的指定格式
|
|
|
|
Args:
|
|
user (User): 要获取的用户
|
|
scheme (Literal['bytes']): 格式
|
|
default (str | None): 获取不到时的默认值
|
|
|
|
Returns:
|
|
bytes: bytes 格式的头像
|
|
"""
|
|
|
|
|
|
async def get_avatar(user: User, scheme: Literal['Data URI', 'bytes'], default: str | None) -> str | bytes:
|
|
if user.avatar is None:
|
|
if default is None:
|
|
msg = "Can't get avatar"
|
|
raise TypeError(msg)
|
|
return default
|
|
avatar = await request.request(URL(user.avatar), is_json=False)
|
|
if scheme == 'Data URI':
|
|
avatar_format = Image.open(BytesIO(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(avatar).decode()}'
|
|
return avatar
|
|
|
|
|
|
def img_to_png(image: bytes) -> bytes:
|
|
"""将图片转换为 PNG 格式"""
|
|
result = BytesIO()
|
|
with Image.open(BytesIO(image)) as img:
|
|
img.save(result, 'PNG')
|
|
return result.getvalue()
|