✨ TOS 查数据使用图片回复

nonebot_plugin_tetris_stats/utils/render/__init__.py

@@ -6,6 +6,8 @@ from nonebot.compat import PYDANTIC_V2
 from ..templates import templates_dir
 from .schemas.bind import Bind
 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(path=render_type, data=data.json())
+        return await env.get_template('index.html').render_async(
+            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']

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

@@ -6,7 +6,7 @@ from pydantic import BaseModel
 
 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

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