mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
65 lines
2.8 KiB
Python
65 lines
2.8 KiB
Python
from hashlib import md5
|
|
from urllib.parse import urlunparse
|
|
|
|
from nonebot.adapters import Bot, Event
|
|
from nonebot_plugin_alconna.uniseg import UniMessage
|
|
from nonebot_plugin_orm import get_session
|
|
from nonebot_plugin_session import EventSession # type: ignore[import-untyped]
|
|
from nonebot_plugin_session_orm import get_session_persist_id # type: ignore[import-untyped]
|
|
from nonebot_plugin_userinfo import BotUserInfo, UserInfo # type: ignore[import-untyped]
|
|
|
|
from ...db import BindStatus, create_or_update_bind, trigger
|
|
from ...utils.avatar import get_avatar
|
|
from ...utils.host import HostPage, get_self_netloc
|
|
from ...utils.platform import get_platform
|
|
from ...utils.render import Bind, render
|
|
from ...utils.render.schemas.base import Avatar, People
|
|
from ...utils.screenshot import screenshot
|
|
from . import alc
|
|
from .api import Player
|
|
from .constant import GAME_TYPE
|
|
|
|
|
|
@alc.assign('bind')
|
|
async def _(bot: Bot, event: Event, account: Player, event_session: EventSession, bot_info: UserInfo = BotUserInfo()): # noqa: B008
|
|
async with trigger(
|
|
session_persist_id=await get_session_persist_id(event_session),
|
|
game_platform=GAME_TYPE,
|
|
command_type='bind',
|
|
command_args=[],
|
|
):
|
|
user = await account.user
|
|
async with get_session() as session:
|
|
bind_status = await create_or_update_bind(
|
|
session=session,
|
|
chat_platform=get_platform(bot),
|
|
chat_account=event.get_user_id(),
|
|
game_platform=GAME_TYPE,
|
|
game_account=user.unique_identifier,
|
|
)
|
|
user_info = await account.get_info()
|
|
if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
|
|
async with HostPage(
|
|
await render(
|
|
'binding',
|
|
Bind(
|
|
platform='TETR.IO',
|
|
status='unknown',
|
|
user=People(
|
|
avatar=f'https://tetr.io/user-content/avatars/{user_info.data.user.id}.jpg?rv={user_info.data.user.avatar_revision}'
|
|
if user_info.data.user.avatar_revision is not None
|
|
else Avatar(type='identicon', hash=md5(user_info.data.user.id.encode()).hexdigest()), # noqa: S324
|
|
name=user_info.data.user.username.upper(),
|
|
),
|
|
bot=People(
|
|
avatar=await get_avatar(bot_info, 'Data URI', '../../static/logo/logo.svg'),
|
|
name=bot_info.user_name,
|
|
),
|
|
command='io查我',
|
|
),
|
|
)
|
|
) as page_hash:
|
|
await UniMessage.image(
|
|
raw=await screenshot(urlunparse(('http', get_self_netloc(), f'/host/{page_hash}.html', '', '', '')))
|
|
).finish()
|