🎨 重命名一些模块

nonebot_plugin_tetris_stats/games/tetrio/api/__init__.py

@@ -0,0 +1,7 @@
+from .player import Player
+from .schemas.user import User
+from .schemas.user_info import UserInfoSuccess
+from .schemas.user_records import UserRecordsSuccess
+from .tetra_league import full_export as tetra_league_full_export
+
+__all__ = ['Player', 'User', 'UserInfoSuccess', 'UserRecordsSuccess', 'tetra_league_full_export']

nonebot_plugin_tetris_stats/games/tetrio/api/cache.py

@@ -0,0 +1,35 @@
+from asyncio import Lock
+from datetime import datetime, timezone
+from typing import ClassVar
+from weakref import WeakValueDictionary
+
+from aiocache import Cache as ACache  # type: ignore[import-untyped]
+from nonebot.compat import type_validate_json
+from nonebot.log import logger
+
+from ....utils.request import Request
+from .schemas.base import FailedModel, SuccessModel
+
+UTC = timezone.utc
+
+
+class Cache:
+    cache = ACache(ACache.MEMORY)
+    task: ClassVar[WeakValueDictionary[str, Lock]] = WeakValueDictionary()
+
+    @classmethod
+    async def get(cls, url: str) -> bytes:
+        lock = cls.task.setdefault(url, Lock())
+        async with lock:
+            if (cached_data := await cls.cache.get(url)) is not None:
+                logger.debug(f'{url}: Cache hit!')
+                return cached_data
+            response_data = await Request.request(url)
+            parsed_data: SuccessModel | FailedModel = type_validate_json(SuccessModel | FailedModel, response_data)  # type: ignore[arg-type]
+            if isinstance(parsed_data, SuccessModel):
+                await cls.cache.add(
+                    url,
+                    response_data,
+                    (parsed_data.cache.cached_until - datetime.now(UTC)).total_seconds(),
+                )
+            return response_data

nonebot_plugin_tetris_stats/games/tetrio/api/models.py

@@ -0,0 +1,17 @@
+from datetime import datetime
+from typing import Literal
+
+from nonebot_plugin_orm import Model
+from sqlalchemy import DateTime, String
+from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
+
+from ....db.models import PydanticType
+from .schemas.base import SuccessModel
+
+
+class TETRIOHistoricalData(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    user_unique_identifier: Mapped[str] = mapped_column(String(24), index=True)
+    api_type: Mapped[Literal['User Info', 'User Records']] = mapped_column(String(16), index=True)
+    data: Mapped[SuccessModel] = mapped_column(PydanticType(get_model=[SuccessModel.__subclasses__], models=set()))
+    update_time: Mapped[datetime] = mapped_column(DateTime, index=True)

nonebot_plugin_tetris_stats/games/tetrio/api/player.py

@@ -0,0 +1,102 @@
+from typing import overload
+
+from nonebot.compat import type_validate_json
+
+from ....db import anti_duplicate_add
+from ....utils.exception import RequestError
+from ....utils.request import splice_url
+from ..constant import BASE_URL, USER_ID, USER_NAME
+from .cache import Cache
+from .models import TETRIOHistoricalData
+from .schemas.base import FailedModel
+from .schemas.user import User
+from .schemas.user_info import UserInfo, UserInfoSuccess
+from .schemas.user_records import UserRecords, UserRecordsSuccess
+
+
+class Player:
+    @overload
+    def __init__(self, *, user_id: str, trust: bool = False): ...
+    @overload
+    def __init__(self, *, user_name: str, trust: bool = False): ...
+    def __init__(self, *, user_id: str | None = None, user_name: str | None = None, trust: bool = False):
+        self.user_id = user_id
+        self.user_name = user_name
+        if not trust:
+            if self.user_id is not None:
+                if not USER_ID.match(self.user_id):
+                    msg = 'Invalid user id'
+                    raise ValueError(msg)
+            elif self.user_name is not None:
+                if not USER_NAME.match(self.user_name):
+                    msg = 'Invalid user name'
+                    raise ValueError(msg)
+            else:
+                msg = 'Invalid user'
+                raise ValueError(msg)
+        self.__user: User | None = None
+        self._user_info: UserInfoSuccess | None = None
+        self._user_records: UserRecordsSuccess | None = None
+
+    @property
+    def _request_user_parameter(self) -> str:
+        if self.user_id is not None:
+            return self.user_id
+        if self.user_name is not None:
+            return self.user_name.lower()
+        msg = 'Invalid user'
+        raise ValueError(msg)
+
+    @property
+    async def user(self) -> User:
+        if self.__user is None:
+            user_info = await self.get_info()
+            self.__user = User(
+                ID=user_info.data.user.id,
+                name=user_info.data.user.username,
+            )
+            self.user_id = user_info.data.user.id
+            self.user_name = user_info.data.user.username
+        return self.__user
+
+    async def get_info(self) -> UserInfoSuccess:
+        """Get User Info"""
+        if self._user_info is None:
+            raw_user_info = await Cache.get(splice_url([BASE_URL, 'users/', f'{self._request_user_parameter}']))
+            user_info: UserInfo = type_validate_json(UserInfo, raw_user_info)  # type: ignore[arg-type]
+            if isinstance(user_info, FailedModel):
+                msg = f'用户信息请求错误:\n{user_info.error}'
+                raise RequestError(msg)
+            self._user_info = user_info
+            await anti_duplicate_add(
+                TETRIOHistoricalData,
+                TETRIOHistoricalData(
+                    user_unique_identifier=(await self.user).unique_identifier,
+                    api_type='User Info',
+                    data=user_info,
+                    update_time=user_info.cache.cached_at,
+                ),
+            )
+        return self._user_info
+
+    async def get_records(self) -> UserRecordsSuccess:
+        """Get User Records"""
+        if self._user_records is None:
+            raw_user_records = await Cache.get(
+                splice_url([BASE_URL, 'users/', f'{self._request_user_parameter}/', 'records'])
+            )
+            user_records: UserRecords = type_validate_json(UserRecords, raw_user_records)  # type: ignore[arg-type]
+            if isinstance(user_records, FailedModel):
+                msg = f'用户Solo数据请求错误:\n{user_records.error}'
+                raise RequestError(msg)
+            self._user_records = user_records
+            await anti_duplicate_add(
+                TETRIOHistoricalData,
+                TETRIOHistoricalData(
+                    user_unique_identifier=(await self.user).unique_identifier,
+                    api_type='User Records',
+                    data=user_records,
+                    update_time=user_records.cache.cached_at,
+                ),
+            )
+        return self._user_records

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

@@ -0,0 +1,20 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel
+
+
+class Cache(BaseModel):
+    status: str
+    cached_at: datetime
+    cached_until: datetime
+
+
+class SuccessModel(BaseModel):
+    success: Literal[True]
+    cache: Cache
+
+
+class FailedModel(BaseModel):
+    success: Literal[False]
+    error: str

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

@@ -0,0 +1,59 @@
+from pydantic import BaseModel, Field
+
+from ..typing import Rank
+from .base import FailedModel
+from .base import SuccessModel as BaseSuccessModel
+
+
+class _User(BaseModel):
+    id: str = Field(..., alias='_id')
+    username: str
+    role: str
+    xp: float
+    supporter: bool
+    verified: bool
+    country: str | None = None
+
+
+class _League(BaseModel):
+    gamesplayed: int
+    gameswon: int
+    rating: float
+    rank: Rank
+    bestrank: Rank
+    decaying: bool
+
+
+class ValidLeague(_League):
+    glicko: float
+    rd: float
+    apm: float
+    pps: float
+    vs: float
+
+
+class ValidUser(_User):
+    league: ValidLeague
+
+
+class InvalidLeague(_League):
+    glicko: float | None = None
+    rd: float | None = None
+    apm: float | None = None
+    pps: float | None = None
+    vs: float | None = None
+
+
+class InvalidUser(_User):
+    league: InvalidLeague
+
+
+class Data(BaseModel):
+    users: list[ValidUser | InvalidUser]
+
+
+class TetraLeagueSuccess(BaseSuccessModel):
+    data: Data
+
+
+TetraLeague = TetraLeagueSuccess | FailedModel

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

@@ -0,0 +1,18 @@
+from typing import Literal
+
+from typing_extensions import override
+
+from ....schemas import BaseUser
+from ...constant import GAME_TYPE
+
+
+class User(BaseUser):
+    platform: Literal['IO'] = GAME_TYPE
+
+    ID: str
+    name: str
+
+    @property
+    @override
+    def unique_identifier(self) -> str:
+        return self.ID

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

@@ -0,0 +1,133 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+from ..typing import Rank
+from .base import FailedModel
+from .base import SuccessModel as BaseSuccessModel
+
+
+class Badge(BaseModel):
+    id: str
+    label: str
+    group: str | None = None
+    ts: datetime | Literal[False] | None = None
+
+
+class MetaLeague(BaseModel):
+    decaying: bool
+
+
+class NeverPlayedLeague(MetaLeague):
+    gamesplayed: Literal[0]
+    gameswon: Literal[0]
+    rating: Literal[-1]
+    rank: Literal['z']
+    standing: Literal[-1]
+    standing_local: Literal[-1]
+    next_rank: None
+    prev_rank: None
+    next_at: Literal[-1]
+    prev_at: Literal[-1]
+    percentile: Literal[-1]
+    percentile_rank: Literal['z']
+    apm: None = None
+    pps: None = None
+    vs: None = None
+
+
+class NeverRatedLeague(MetaLeague):
+    gamesplayed: Literal[1, 2, 3, 4, 5, 6, 7, 8, 9]
+    gameswon: int
+    rating: Literal[-1]
+    rank: Literal['z']
+    standing: Literal[-1]
+    standing_local: Literal[-1]
+    next_rank: None
+    prev_rank: None
+    next_at: Literal[-1]
+    prev_at: Literal[-1]
+    percentile: Literal[-1]
+    percentile_rank: Literal['z']
+    apm: float
+    pps: float
+    vs: float | None = None
+
+
+class RatedLeague(MetaLeague):
+    gamesplayed: int
+    gameswon: int
+    rating: float
+    rank: Rank
+    bestrank: Rank
+    standing: int
+    standing_local: int
+    next_rank: Rank | None = None
+    prev_rank: Rank | None = None
+    next_at: int
+    prev_at: int
+    percentile: float
+    percentile_rank: str
+    glicko: float
+    rd: float
+    apm: float
+    pps: float
+    vs: float | None = None
+
+
+class Discord(BaseModel):
+    id: str
+    username: str
+
+
+class Connections(BaseModel):
+    discord: Discord | None = None
+
+
+class Distinguishment(BaseModel):
+    type: str
+
+
+class User(BaseModel):
+    id: str = Field(..., alias='_id')
+    username: str
+    role: Literal['anon', 'user', 'bot', 'halfmod', 'mod', 'admin', 'sysop', 'banned']
+    ts: datetime | None = None
+    botmaster: str | None = None
+    badges: list[Badge]
+    xp: float
+    gamesplayed: int
+    gameswon: int
+    gametime: float
+    country: str | None = None
+    badstanding: bool | None = None
+    supporter: bool | None = None  # osk说是必有, 但实际上不是 fkosk
+    supporter_tier: int
+    verified: bool
+    league: NeverPlayedLeague | NeverRatedLeague | RatedLeague
+    avatar_revision: int | None = None
+    """This user's avatar ID. Get their avatar at
+
+    https://tetr.io/user-content/avatars/{ USERID }.jpg?rv={ AVATAR_REVISION }"""
+    banner_revision: int | None = None
+    """This user's banner ID. Get their banner at
+
+    https://tetr.io/user-content/banners/{ USERID }.jpg?rv={ BANNER_REVISION }
+
+    Ignore this field if the user is not a supporter."""
+    bio: str | None = None
+    connections: Connections
+    friend_count: int | None = None
+    distinguishment: Distinguishment | None = None
+
+
+class Data(BaseModel):
+    user: User
+
+
+class UserInfoSuccess(BaseSuccessModel):
+    data: Data
+
+
+UserInfo = UserInfoSuccess | FailedModel

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

@@ -0,0 +1,122 @@
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+from .....utils.typing import Number
+from .base import FailedModel
+from .base import SuccessModel as BaseSuccessModel
+
+
+class Time(BaseModel):
+    start: int
+    zero: bool
+    locked: bool
+    prev: int
+    frameoffset: int | None = None
+
+
+class Clears(BaseModel):
+    singles: int
+    doubles: int
+    triples: int
+    quads: int
+    pentas: int | None = None
+    realtspins: int
+    minitspins: int
+    minitspinsingles: int
+    tspinsingles: int
+    minitspindoubles: int
+    tspindoubles: int
+    tspintriples: int
+    tspinquads: int
+    allclear: int
+
+
+class Garbage(BaseModel):
+    sent: int
+    received: int
+    attack: int | None = None
+    cleared: int | None = None
+
+
+class Finesse(BaseModel):
+    combo: int
+    faults: int
+    perfectpieces: int
+
+
+class EndContext(BaseModel):
+    seed: Number
+    lines: int
+    level_lines: int
+    level_lines_needed: int
+    inputs: int
+    holds: int | None = None
+    time: Time
+    score: int
+    zenlevel: int | None = None
+    zenprogress: int | None = None
+    level: int
+    combo: int
+    currentcombopower: int | None = None  # WTF
+    topcombo: int
+    btb: int
+    topbtb: int
+    currentbtbchainpower: int | None = None  # WTF * 2
+    tspins: int
+    piecesplaced: int
+    clears: Clears
+    garbage: Garbage
+    kills: int
+    finesse: Finesse
+    final_time: float = Field(..., alias='finalTime')
+    gametype: str
+
+
+class _User(BaseModel):
+    id: str = Field(..., alias='_id')
+    username: str
+
+
+class _Record(BaseModel):
+    id: str = Field(..., alias='_id')
+    stream: str
+    replayid: str
+    user: _User
+    ts: datetime
+    ismulti: bool | None = None
+
+
+class SoloRecord(_Record):
+    endcontext: EndContext
+
+
+class MultiRecord(_Record):
+    endcontext: list[EndContext]
+
+
+class SoloModeRecord(BaseModel):
+    record: SoloRecord
+    rank: int | None = None
+
+
+class Records(BaseModel):
+    sprint: SoloModeRecord = Field(..., alias='40l')
+    blitz: SoloModeRecord
+
+
+class Zen(BaseModel):
+    level: int
+    score: int
+
+
+class Data(BaseModel):
+    records: Records
+    zen: Zen
+
+
+class UserRecordsSuccess(BaseSuccessModel):
+    data: Data
+
+
+UserRecords = UserRecordsSuccess | FailedModel

nonebot_plugin_tetris_stats/games/tetrio/api/tetra_league.py

@@ -0,0 +1,36 @@
+from typing import Literal, NamedTuple, overload
+
+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 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

nonebot_plugin_tetris_stats/games/tetrio/api/typing.py

@@ -0,0 +1,22 @@
+from typing import Literal
+
+Rank = Literal[
+    'x',
+    'u',
+    'ss',
+    's+',
+    's',
+    's-',
+    'a+',
+    'a',
+    'a-',
+    'b+',
+    'b',
+    'b-',
+    'c+',
+    'c',
+    'c-',
+    'd+',
+    'd',
+    'z',  # 未定级
+]