🔖 1.2.10

player实例被alconna缓存

nonebot_plugin_tetris_stats/__init__.py

@@ -8,6 +8,11 @@ require('nonebot_plugin_orm')
 require('nonebot_plugin_session')
 require('nonebot_plugin_session_orm')
 
+from nonebot_plugin_alconna import namespace  # noqa: E402
+
+with namespace('tetris_stats') as ns:
+    ns.enable_message_cache = False
+
 from .config.config import migrations  # noqa: E402
 
 __plugin_meta__ = PluginMetadata(
@@ -21,5 +26,4 @@ __plugin_meta__ = PluginMetadata(
     },
 )
 
-from . import game_data_processor  # noqa: F401, E402
+from . import games  # noqa: F401, E402
-from .utils import host  # noqa: F401, E402

nonebot_plugin_tetris_stats/db/__init__.py

@@ -1,3 +1,4 @@
+from asyncio import Lock
 from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
 from datetime import datetime, timezone
@@ -15,9 +16,9 @@ from .models import Bind, TriggerHistoricalData
 UTC = timezone.utc
 
 if TYPE_CHECKING:
-    from ..game_data_processor.io_data_processor.api.models import TETRIOHistoricalData
+    from ..games.tetrio.api.models import TETRIOHistoricalData
-    from ..game_data_processor.top_data_processor.api.models import TOPHistoricalData
+    from ..games.top.api.models import TOPHistoricalData
-    from ..game_data_processor.tos_data_processor.api.models import TOSHistoricalData
+    from ..games.tos.api.models import TOSHistoricalData
 
 
 class BindStatus(Enum):
@@ -72,9 +73,11 @@ async def create_or_update_bind(
 
 T = TypeVar('T', 'TETRIOHistoricalData', 'TOPHistoricalData', 'TOSHistoricalData')
 
+lock = Lock()
+
 
 async def anti_duplicate_add(cls: type[T], model: T) -> None:
-    async with get_session() as session:
+    async with lock, get_session() as session:
         result = (
             await session.scalars(
                 select(cls)
@@ -88,8 +91,8 @@ async def anti_duplicate_add(cls: type[T], model: T) -> None:
                 if i.data == model.data:
                     logger.debug('Anti duplicate successfully')
                     return
-            session.add(model)
+        session.add(model)
-            await session.commit()
+        await session.commit()
 
 
 @asynccontextmanager

nonebot_plugin_tetris_stats/db/models.py

@@ -23,9 +23,8 @@ class PydanticType(TypeDecorator):
         *args: Any,
         **kwargs: Any,
     ):
-        for i in get_model:
+        self.get_model = get_model
-            models.update(i())
+        self._models = models
-        self.models = models
         super().__init__(*args, **kwargs)
 
     if PYDANTIC_V2:
@@ -56,6 +55,14 @@ class PydanticType(TypeDecorator):
                     ...
         raise ValueError
 
+    @property
+    def models(self) -> tuple[type[BaseModel], ...]:
+        models: set[type[BaseModel]] = set()
+        for i in self.get_model:
+            models.update(i())
+        models.update(self._models)
+        return tuple(models)
+
 
 class Bind(MappedAsDataclass, Model):
     id: Mapped[int] = mapped_column(init=False, primary_key=True)

nonebot_plugin_tetris_stats/games/__init__.py

@@ -32,11 +32,7 @@ def add_default_handlers(matcher: type[AlconnaMatcher]) -> None:
         raise FinishedException
 
 
-from . import (  # noqa: F401, E402
+from . import tetrio, top, tos  # noqa: F401, E402
-    io_data_processor,
-    top_data_processor,
-    tos_data_processor,
-)
 
 
 @run_postprocessor

nonebot_plugin_tetris_stats/games/tetrio/__init__.py

@@ -1,6 +1,7 @@
 from arclet.alconna import Alconna, AllParam, Arg, ArgFlag, Args, CommandMeta, Option
 from nonebot_plugin_alconna import At, on_alconna
 
+from ... import ns
 from ...utils.exception import MessageFormatError
 from ...utils.typing import Me
 from .. import add_default_handlers
@@ -72,6 +73,7 @@ alc = on_alconna(
             compact=True,
             fuzzy_match=True,
         ),
+        namespace=ns,
     ),
     skip_for_unmatch=False,
     auto_send_output=True,

nonebot_plugin_tetris_stats/games/tetrio/api/player.py

@@ -43,7 +43,7 @@ class Player:
         if self.user_id is not None:
             return self.user_id
         if self.user_name is not None:
-            return self.user_name
+            return self.user_name.lower()
         msg = 'Invalid user'
         raise ValueError(msg)
 

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/user_records.py

@@ -36,7 +36,7 @@ class Garbage(BaseModel):
     sent: int
     received: int
     attack: int | None = None
-    cleared: int
+    cleared: int | None = None
 
 
 class Finesse(BaseModel):
@@ -96,7 +96,7 @@ class MultiRecord(_Record):
 
 
 class SoloModeRecord(BaseModel):
-    record: SoloRecord
+    record: SoloRecord | None = None
     rank: int | None = None
 
 

nonebot_plugin_tetris_stats/games/tetrio/query.py

@@ -1,39 +1,51 @@
 import contextlib
 from asyncio import gather
-from datetime import datetime, timedelta, timezone
+from collections import defaultdict
+from datetime import date, datetime, timedelta, timezone
 from hashlib import md5
 from math import ceil, floor
+from typing import ClassVar
 from urllib.parse import urlunparse
 from zoneinfo import ZoneInfo
 
+from aiofiles import open
+from nonebot import get_driver
 from nonebot.adapters import Bot, Event
+from nonebot.compat import type_validate_json
 from nonebot.matcher import Matcher
 from nonebot_plugin_alconna import At
 from nonebot_plugin_alconna.uniseg import UniMessage
+from nonebot_plugin_apscheduler import scheduler  # type: ignore[import-untyped]
+from nonebot_plugin_localstore import get_data_file  # type: ignore[import-untyped]
 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 sqlalchemy import select
+from zstandard import ZstdDecompressor
 
 from ...db import query_bind_info, trigger
 from ...utils.host import HostPage, get_self_netloc
 from ...utils.platform import get_platform
-from ...utils.render import TETRIOInfo, render
+from ...utils.render import render
-from ...utils.render.schemas.base import Avatar
+from ...utils.render.schemas.base import Avatar, Ranking
-from ...utils.render.schemas.tetrio_info import Data, Radar, Ranking, TetraLeague, TetraLeagueHistory
+from ...utils.render.schemas.tetrio_info import Data, Info, Radar, TetraLeague, TetraLeagueHistory
 from ...utils.render.schemas.tetrio_info import User as TemplateUser
 from ...utils.screenshot import screenshot
-from ...utils.typing import Me
+from ...utils.typing import Me, Number
 from ..constant import CANT_VERIFY_MESSAGE
 from . import alc
 from .api import Player, User, UserInfoSuccess
 from .api.models import TETRIOHistoricalData
+from .api.schemas.tetra_league import TetraLeagueSuccess
 from .api.schemas.user_info import NeverPlayedLeague, NeverRatedLeague, RatedLeague
 from .api.schemas.user_records import SoloModeRecord, SoloRecord
 from .constant import GAME_TYPE, TR_MAX, TR_MIN
+from .model import IORank
 
 UTC = timezone.utc
 
+driver = get_driver()
+
 
 @alc.assign('query')
 async def _(bot: Bot, event: Event, matcher: Matcher, target: At | Me, event_session: EventSession):
@@ -156,7 +168,8 @@ async def query_historical_data(user: User, user_info: UserInfoSuccess) -> list[
             if extra is not None:
                 historical_data = list(historical_data)
                 historical_data.append(extra)
-    if not historical_data:
+    full_export_data = FullExport.get_data(user.unique_identifier)
+    if not historical_data and not full_export_data:
         return [
             Data(record_at=today - forward, tr=user_info.data.user.league.rating),
             Data(record_at=today.replace(microsecond=1000), tr=user_info.data.user.league.rating),
@@ -168,13 +181,13 @@ async def query_historical_data(user: User, user_info: UserInfoSuccess) -> list[
         )
         for i in historical_data
         if isinstance(i.data, UserInfoSuccess) and isinstance(i.data.data.user.league, RatedLeague)
-    ]
+    ] + full_export_data
 
     # 按照时间排序
     histories = sorted(histories, key=lambda x: x.record_at)
     for index, value in enumerate(histories):
-        # 在历史记录里找有没有今天0点后的数据
+        # 在历史记录里找有没有今天0点后的数据, 并且至少要有两个数据点
-        if value.record_at > today:
+        if value.record_at > today and len(histories) >= 2:  # noqa: PLR2004
             histories = histories[:index] + [
                 get_specified_point(histories[index - 1], histories[index], today.replace(microsecond=1000))
             ]
@@ -217,7 +230,7 @@ async def make_query_image(
     async with HostPage(
         await render(
             'tetrio/info',
-            TETRIOInfo(
+            Info(
                 user=TemplateUser(
                     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
@@ -294,3 +307,82 @@ def make_query_text(user_info: UserInfoSuccess, sprint: SoloModeRecord, blitz: S
         message += f'\nBlitz: {blitz.record.endcontext.score}'
         message += f' ( #{blitz.rank} )' if blitz.rank is not None else ''
     return UniMessage(message)
+
+
+class FullExport:
+    cache: ClassVar[defaultdict[str, set[tuple[datetime, Number]]]] = defaultdict(set)
+    latest_update: ClassVar[date | None] = None
+
+    @classmethod
+    async def init(cls) -> None:
+        async with get_session() as session:
+            full_exports = (await session.scalars(select(IORank).where(IORank.update_time >= cls.start_time()))).all()
+        await gather(
+            *[
+                cls._load(update_time, file_hash)
+                for file_hash, update_time in {
+                    i.file_hash: i.update_time for i in full_exports if i.file_hash is not None
+                }.items()
+            ]
+        )
+
+    @classmethod
+    async def update(cls) -> None:
+        if cls.latest_update == datetime.now(tz=ZoneInfo('Asia/Shanghai')).date():
+            return
+        start_time = cls.start_time()
+        for i in cls.cache:
+            cls.cache[i] = {j for j in cls.cache[i] if j[0] >= start_time}
+        latest_time = max(cls.cache)
+        async with get_session() as session:
+            full_exports = (await session.scalars(select(IORank).where(IORank.update_time > latest_time))).all()
+        await gather(
+            *[
+                cls._load(update_time, file_hash)
+                for file_hash, update_time in {
+                    i.file_hash: i.update_time for i in full_exports if i.file_hash is not None
+                }.items()
+            ]
+        )
+        cls.latest_update = datetime.now(tz=ZoneInfo('Asia/Shanghai')).date()
+
+    @classmethod
+    def get_data(cls, unique_identifier: str) -> list[Data]:
+        return [Data(record_at=i[0], tr=i[1]) for i in cls.cache[unique_identifier]]
+
+    @classmethod
+    def start_time(cls) -> datetime:
+        return (
+            datetime.now(ZoneInfo('Asia/Shanghai')).replace(hour=0, minute=0, second=0, microsecond=0)
+            - timedelta(days=9)
+        ).astimezone(UTC)
+
+    @classmethod
+    async def _load(cls, update_time: datetime, file_hash: str) -> None:
+        try:
+            users = type_validate_json(TetraLeagueSuccess, await cls.decompress(file_hash)).data.users
+        except FileNotFoundError:
+            await cls.clear_invalid(file_hash)
+            return
+        update_time = update_time.astimezone(ZoneInfo('Asia/Shanghai'))
+        for i in users:
+            cls.cache[i.id].add((update_time, i.league.rating))
+
+    @classmethod
+    async def decompress(cls, file_hash: str) -> bytes:
+        async with open(get_data_file('nonebot_plugin_tetris_stats', f'{file_hash}.json.zst'), mode='rb') as file:
+            return ZstdDecompressor().decompress(await file.read())
+
+    @classmethod
+    async def clear_invalid(cls, file_hash: str) -> None:
+        async with get_session() as session:
+            full_exports = (await session.scalars(select(IORank).where(IORank.file_hash == file_hash))).all()
+            for i in full_exports:
+                i.file_hash = None
+            await session.commit()
+
+
+@driver.on_startup
+async def _():
+    await FullExport.init()
+    scheduler.add_job(FullExport.update, 'interval', hours=1)

nonebot_plugin_tetris_stats/games/top/__init__.py

@@ -1,6 +1,7 @@
 from arclet.alconna import Alconna, AllParam, Arg, ArgFlag, Args, CommandMeta, Option
 from nonebot_plugin_alconna import At, on_alconna
 
+from ... import ns
 from ...utils.exception import MessageFormatError
 from ...utils.typing import Me
 from .. import add_default_handlers
@@ -61,6 +62,7 @@ alc = on_alconna(
             compact=True,
             fuzzy_match=True,
         ),
+        namespace=ns,
     ),
     skip_for_unmatch=False,
     auto_send_output=True,

nonebot_plugin_tetris_stats/games/tos/__init__.py

@@ -1,6 +1,7 @@
 from arclet.alconna import Alconna, AllParam, Arg, ArgFlag, Args, CommandMeta, Option
 from nonebot_plugin_alconna import At, on_alconna
 
+from ... import ns
 from ...utils.exception import MessageFormatError
 from ...utils.typing import Me
 from .. import add_default_handlers
@@ -67,6 +68,7 @@ alc = on_alconna(
             compact=True,
             fuzzy_match=True,
         ),
+        namespace=ns,
     ),
     skip_for_unmatch=False,
     auto_send_output=True,

nonebot_plugin_tetris_stats/games/tos/query.py

@@ -1,22 +1,33 @@
 from asyncio import gather
-from dataclasses import dataclass
+from datetime import timedelta
-from typing import Literal
+from http import HTTPStatus
+from typing import Literal, NamedTuple
+from urllib.parse import urlunparse
 
 from nonebot.adapters import Bot, Event
 from nonebot.matcher import Matcher
 from nonebot_plugin_alconna import At
 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 import EventSession  # type: ignore[import-untyped]  # type: ignore[import-untyped]
 from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
+from nonebot_plugin_userinfo import EventUserInfo, UserInfo  # type: ignore[import-untyped]
 
 from ...db import query_bind_info, trigger
+from ...utils.avatar import get_avatar
+from ...utils.exception import RequestError
+from ...utils.host import HostPage, get_self_netloc
 from ...utils.metrics import TetrisMetricsProWithLPMADPM, get_metrics
 from ...utils.platform import get_platform
-from ...utils.typing import Me
+from ...utils.render import render
+from ...utils.render.schemas.base import People, Ranking
+from ...utils.render.schemas.tos_info import Info, Multiplayer, Radar
+from ...utils.screenshot import screenshot
+from ...utils.typing import Me, Number
 from ..constant import CANT_VERIFY_MESSAGE
 from . import alc
 from .api import Player
+from .api.schemas.user_info import UserInfoSuccess
 from .constant import GAME_TYPE
 
 
@@ -24,7 +35,12 @@ def add_special_handlers(
     teaid_prefix: Literal['onebot-', 'kook-', 'discord-', 'qqguild-'], match_event: type[Event]
 ) -> None:
     @alc.assign('query')
-    async def _(event: Event, target: At | Me, event_session: EventSession):
+    async def _(
+        event: Event,
+        target: At | Me,
+        event_session: EventSession,
+        event_user_info: UserInfo = EventUserInfo(),  # noqa: B008
+    ):
         if isinstance(event, match_event):
             async with trigger(
                 session_persist_id=await get_session_persist_id(event_session),
@@ -32,16 +48,23 @@ def add_special_handlers(
                 command_type='query',
                 command_args=[],
             ):
-                await (
+                player = Player(
-                    await make_query_text(
+                    teaid=f'{teaid_prefix}{target.target}'
-                        Player(
+                    if isinstance(target, At)
-                            teaid=f'{teaid_prefix}{target.target}'
+                    else f'{teaid_prefix}{event.get_user_id()}',
-                            if isinstance(target, At)
+                    trust=True,
-                            else f'{teaid_prefix}{event.get_user_id()}',
+                )
-                            trust=True,
+                try:
-                        )
+                    user_info, game_data = await gather(player.get_info(), get_game_data(player))
-                    )
+                    if game_data is not None:
-                ).finish()
+                        await UniMessage.image(
+                            raw=await make_query_image(user_info, game_data, event_user_info)
+                        ).finish()
+                    await (await make_query_text(user_info, game_data)).finish()
+                except RequestError as e:
+                    if e.status_code == HTTPStatus.BAD_REQUEST and '未找到此用户' in e.message:
+                        return
+                    raise
 
 
 try:
@@ -76,7 +99,14 @@ except ImportError:
 
 
 @alc.assign('query')
-async def _(bot: Bot, event: Event, matcher: Matcher, target: At | Me, event_session: EventSession):
+async def _(  # noqa: PLR0913
+    bot: Bot,
+    event: Event,
+    matcher: Matcher,
+    target: At | Me,
+    event_session: EventSession,
+    event_user_info: UserInfo = EventUserInfo(),  # noqa: B008
+):
     async with trigger(
         session_persist_id=await get_session_persist_id(event_session),
         game_platform=GAME_TYPE,
@@ -93,24 +123,35 @@ async def _(bot: Bot, event: Event, matcher: Matcher, target: At | Me, event_ses
         if bind is None:
             await matcher.finish('未查询到绑定信息')
         message = CANT_VERIFY_MESSAGE
-        await (message + await make_query_text(Player(teaid=bind.game_account, trust=True))).finish()
+        player = Player(teaid=bind.game_account, trust=True)
+        user_info, game_data = await gather(player.get_info(), get_game_data(player))
+        if game_data is not None:
+            await (
+                message + UniMessage.image(raw=await make_query_image(user_info, game_data, event_user_info))
+            ).finish()
+        await (message + await make_query_text(user_info, game_data)).finish()
 
 
 @alc.assign('query')
-async def _(account: Player, event_session: EventSession):
+async def _(account: Player, event_session: EventSession, event_user_info: UserInfo = EventUserInfo()):  # noqa: B008
     async with trigger(
         session_persist_id=await get_session_persist_id(event_session),
         game_platform=GAME_TYPE,
         command_type='query',
         command_args=[],
     ):
-        await (await make_query_text(account)).finish()
+        user_info, game_data = await gather(account.get_info(), get_game_data(account))
+        if game_data is not None:
+            await UniMessage.image(raw=await make_query_image(user_info, game_data, event_user_info)).finish()
+        await (await make_query_text(user_info, game_data)).finish()
 
 
-@dataclass
+class GameData(NamedTuple):
-class GameData:
     game_num: int
     metrics: TetrisMetricsProWithLPMADPM
+    OR: Number
+    dspp: Number
+    ge: Number
 
 
 async def get_game_data(player: Player, query_num: int = 50) -> GameData | None:
@@ -119,7 +160,7 @@ async def get_game_data(player: Player, query_num: int = 50) -> GameData | None:
     if user_profile.data == []:
         return None
     weighted_total_lpm = weighted_total_apm = weighted_total_adpm = total_time = 0.0
-    num = 0
+    total_attack = total_dig = total_offset = total_pieses = total_receive = num = 0
     for i in user_profile.data:
         # 排除单人局和时间为0的游戏
         # 茶: 不计算没挖掘的局, 即使apm和lpm也如此
@@ -133,6 +174,11 @@ async def get_game_data(player: Player, query_num: int = 50) -> GameData | None:
         weighted_total_lpm += lpm * time
         weighted_total_apm += apm * time
         weighted_total_adpm += adpm * time
+        total_attack += i.attack
+        total_dig += i.dig
+        total_offset += i.offset
+        total_pieses += i.pieces
+        total_receive += i.receive
         total_time += time
         num += 1
         if num >= query_num:
@@ -143,14 +189,51 @@ async def get_game_data(player: Player, query_num: int = 50) -> GameData | None:
     metrics = get_metrics(
         lpm=weighted_total_lpm / total_time, apm=weighted_total_apm / total_time, adpm=weighted_total_adpm / total_time
     )
-    lpm = weighted_total_lpm / total_time
+    return GameData(
-    apm = weighted_total_apm / total_time
+        game_num=num,
-    adpm = weighted_total_adpm / total_time
+        metrics=metrics,
-    return GameData(game_num=num, metrics=metrics)
+        OR=total_offset / total_receive * 100,
+        dspp=total_dig / total_pieses,
+        ge=2 * ((total_attack * total_dig) / total_pieses**2),
+    )
 
 
-async def make_query_text(player: Player) -> UniMessage:
+async def make_query_image(user_info: UserInfoSuccess, game_data: GameData, event_user_info: UserInfo) -> bytes:
-    user_info, game_data = await gather(player.get_info(), get_game_data(player))
+    metrics = game_data.metrics
+    duration = timedelta(milliseconds=float(user_info.data.pb_sprint)).total_seconds()
+    sprint_value = f'{duration:.1f}s' if duration < 60 else f'{duration // 60:.0f}m {duration % 60:.1f}s'  # noqa: PLR2004
+    async with HostPage(
+        await render(
+            'tos/info',
+            Info(
+                user=People(avatar=await get_avatar(event_user_info, 'Data URI', None), name=user_info.data.name),
+                ranking=Ranking(rating=float(user_info.data.ranking), rd=round(float(user_info.data.rd_now), 2)),
+                multiplayer=Multiplayer(
+                    pps=metrics.pps,
+                    lpm=metrics.lpm,
+                    apm=metrics.apm,
+                    apl=metrics.apl,
+                    vs=metrics.vs,
+                    adpm=metrics.adpm,
+                    adpl=metrics.adpl,
+                ),
+                radar=Radar(
+                    app=(app := (metrics.apm / (60 * metrics.pps))),
+                    OR=game_data.OR,
+                    dspp=game_data.dspp,
+                    ci=150 * game_data.dspp - 125 * app + 50 * (metrics.vs / metrics.apm) - 25,
+                    ge=game_data.ge,
+                ),
+                sprint=sprint_value,
+                challenge=f'{int(user_info.data.pb_challenge):,}' if user_info.data.pb_challenge != '0' else 'N/A',
+                marathon=f'{int(user_info.data.pb_marathon):,}' if user_info.data.pb_marathon != '0' else 'N/A',
+            ),
+        )
+    ) as page_hash:
+        return await screenshot(urlunparse(('http', get_self_netloc(), f'/host/{page_hash}.html', '', '', '')))
+
+
+async def make_query_text(user_info: UserInfoSuccess, game_data: GameData | None) -> UniMessage:
     user_data = user_info.data
     message = f'用户 {user_data.name} ({user_data.teaid}) '
     if user_data.ranked_games == '0':

nonebot_plugin_tetris_stats/utils/render/__init__.py

@@ -5,7 +5,9 @@ from nonebot.compat import PYDANTIC_V2
 
 from ..templates import templates_dir
 from .schemas.bind import Bind
-from .schemas.tetrio_info import TETRIOInfo
+from .schemas.tetrio_info import Info as TETRIOInfo
+from .schemas.top_info import Info as TOPInfo
+from .schemas.tos_info import Info as TOSInfo
 
 env = Environment(
     loader=FileSystemLoader(templates_dir), autoescape=True, trim_blocks=True, lstrip_blocks=True, enable_async=True
@@ -20,10 +22,22 @@ async def render(render_type: Literal['binding'], data: Bind) -> str: ...
 async def render(render_type: Literal['tetrio/info'], data: TETRIOInfo) -> str: ...
 
 
-async def render(render_type: Literal['binding', 'tetrio/info'], data: Bind | TETRIOInfo) -> str:
+@overload
+async def render(render_type: Literal['top/info'], data: TOPInfo) -> str: ...
+
+
+@overload
+async def render(render_type: Literal['tos/info'], data: TOSInfo) -> str: ...
+
+
+async def render(
+    render_type: Literal['binding', 'tetrio/info', 'top/info', 'tos/info'], data: Bind | TETRIOInfo | TOPInfo | TOSInfo
+) -> str:
     if PYDANTIC_V2:
-        return await env.get_template('index.html').render_async(path=render_type, data=data.model_dump_json())
+        return await env.get_template('index.html').render_async(
-    return await env.get_template('index.html').render_async(path=render_type, data=data.json())
+            path=render_type, data=data.model_dump_json(by_alias=True)
+        )
+    return await env.get_template('index.html').render_async(path=render_type, data=data.json(by_alias=True))
 
 
-__all__ = ['render', 'Bind', 'TETRIOInfo']
+__all__ = ['render', 'Bind']

nonebot_plugin_tetris_stats/utils/render/schemas/base.py

@@ -2,6 +2,8 @@ from typing import Literal
 
 from pydantic import BaseModel
 
+from ...typing import Number
+
 
 class Avatar(BaseModel):
     type: Literal['identicon']
@@ -11,3 +13,8 @@ class Avatar(BaseModel):
 class People(BaseModel):
     avatar: str | Avatar
     name: str
+
+
+class Ranking(BaseModel):
+    rating: Number
+    rd: Number

nonebot_plugin_tetris_stats/utils/render/schemas/tetrio_info.py

@@ -4,9 +4,9 @@ from typing import Annotated, ClassVar
 from nonebot.compat import PYDANTIC_V2
 from pydantic import BaseModel
 
-from ....game_data_processor.io_data_processor.api.typing import Rank
+from ....games.tetrio.api.typing import Rank
 from ...typing import Number
-from .base import People
+from .base import People, Ranking
 
 if PYDANTIC_V2:
     from pydantic import PlainSerializer
@@ -20,11 +20,6 @@ class User(People):
     bio: str | None
 
 
-class Ranking(BaseModel):
-    rating: Number
-    rd: Number
-
-
 class TetraLeague(BaseModel):
     rank: Rank
     tr: Number
@@ -62,7 +57,7 @@ class Radar(BaseModel):
     ge: Number
 
 
-class TETRIOInfo(BaseModel):
+class Info(BaseModel):
     user: User
     ranking: Ranking
     tetra_league: TetraLeague

nonebot_plugin_tetris_stats/utils/render/schemas/top_info.py

@@ -0,0 +1,17 @@
+from pydantic import BaseModel
+
+from ...typing import Number
+from .base import People
+
+
+class Data(BaseModel):
+    pps: Number
+    lpm: Number
+    apm: Number
+    apl: Number
+
+
+class Info(BaseModel):
+    user: People
+    today: Data
+    history: Data

nonebot_plugin_tetris_stats/utils/render/schemas/tos_info.py

@@ -0,0 +1,32 @@
+from pydantic import BaseModel, Field
+
+from ...typing import Number
+from .base import People, Ranking
+
+
+class Multiplayer(BaseModel):
+    pps: Number
+    lpm: Number
+    apm: Number
+    apl: Number
+    vs: Number
+    adpm: Number
+    adpl: Number
+
+
+class Radar(BaseModel):
+    app: Number
+    OR: Number = Field(serialization_alias='or')
+    dspp: Number
+    ci: Number
+    ge: Number
+
+
+class Info(BaseModel):
+    user: People
+    ranking: Ranking
+    multiplayer: Multiplayer
+    radar: Radar
+    sprint: str
+    challenge: str
+    marathon: str

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = 'nonebot-plugin-tetris-stats'
-version = '1.2.2'
+version = '1.2.10'
 description = '一款基于 NoneBot2 的用于查询 Tetris 相关游戏数据的插件'
 authors = ['scdhh <wallfjjd@gmail.com>']
 readme = 'README.md'
@@ -38,14 +38,16 @@ types-lxml = "^2024.2.9"
 types-pillow = "^10.2.0.20240423"
 types-ujson = '^5.9.0'
 pandas-stubs = '>=1.5.2,<3.0.0'
-nonebot-plugin-orm = { extras = ["default"], version = ">=0.3,<0.8" }
+nonebot2 = { extras = ["all"], version = "^2.3.0" }
 nonebot-adapter-discord = "^0.1.3"
 nonebot-adapter-kaiheila = "^0.3.4"
 nonebot-adapter-onebot = "^2.4.1"
 nonebot-adapter-qq = "^1.4.4"
 nonebot-adapter-satori = "^0.11.4"
+nonebot-plugin-orm = { extras = ["default"], version = ">=0.3,<0.8" }
 
 [tool.poetry.group.debug.dependencies]
+memory-profiler = "^0.61.0"
 objprint = '^0.2.2'
 viztracer = "^0.16.2"