mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
from typing import Literal, NamedTuple, TypedDict, overload
|
|
from urllib.parse import urlencode
|
|
|
|
from nonebot.compat import type_validate_json
|
|
|
|
from ....utils.exception import RequestError
|
|
from ....utils.request import splice_url
|
|
from ..constant import BASE_URL
|
|
from .cache import Cache
|
|
from .schemas.base import FailedModel
|
|
from .schemas.tetra_league import TetraLeague, TetraLeagueSuccess
|
|
|
|
|
|
class Parameter(TypedDict, total=False):
|
|
after: float
|
|
before: float
|
|
limit: int
|
|
country: str
|
|
|
|
|
|
async def leaderboard(parameter: Parameter | None = None) -> TetraLeagueSuccess:
|
|
league: TetraLeague = type_validate_json(
|
|
TetraLeague, # type: ignore[arg-type]
|
|
(await Cache.get(splice_url([BASE_URL, 'users/lists/league', f'?{urlencode(parameter or {})}']))),
|
|
)
|
|
if isinstance(league, FailedModel):
|
|
msg = f'排行榜数据请求错误:\n{league.error}'
|
|
raise RequestError(msg)
|
|
return league
|
|
|
|
|
|
class FullExport(NamedTuple):
|
|
model: TetraLeagueSuccess
|
|
original: bytes
|
|
|
|
|
|
@overload
|
|
async def full_export(*, with_original: Literal[False]) -> TetraLeagueSuccess: ...
|
|
|
|
|
|
@overload
|
|
async def full_export(*, with_original: Literal[True]) -> FullExport: ...
|
|
|
|
|
|
async def full_export(*, with_original: bool) -> TetraLeagueSuccess | FullExport:
|
|
full: TetraLeague = type_validate_json(
|
|
TetraLeague, # type: ignore[arg-type]
|
|
(data := await Cache.get(splice_url([BASE_URL, 'users/lists/league/all']))),
|
|
)
|
|
if isinstance(full, FailedModel):
|
|
msg = f'排行榜数据请求错误:\n{full.error}'
|
|
raise RequestError(msg)
|
|
if with_original:
|
|
return FullExport(full, data)
|
|
return full
|