🔖 1.5.5

Bumps [ruff](https://github.com/astral-sh/ruff) from 0.6.9 to 0.7.0.
- [Release notes](https://github.com/astral-sh/ruff/releases)
- [Changelog](https://github.com/astral-sh/ruff/blob/main/CHANGELOG.md)
- [Commits](https://github.com/astral-sh/ruff/compare/0.6.9...0.7.0)

---
updated-dependencies:
- dependency-name: ruff
  dependency-type: direct:development
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

.pre-commit-config.yaml

@@ -7,13 +7,13 @@ ci:
     autoupdate_commit_msg: ':arrow_up: auto update by pre-commit hooks'
 repos:
     - repo: https://github.com/astral-sh/ruff-pre-commit
-      rev: v0.6.4
+      rev: v0.6.9
       hooks:
           - id: ruff
             args: [--fix, --exit-non-zero-on-fix]
-            stages: [commit]
+            stages: [pre-commit]
           - id: ruff-format
-            stages: [commit]
+            stages: [pre-commit]
 
     - repo: https://github.com/nonebot/nonemoji
       rev: v0.1.4

nonebot_plugin_tetris_stats/config/migrations/3c25a5a8c050_refactor_historical.py

@@ -29,7 +29,7 @@ branch_labels: str | Sequence[str] | None = None
 depends_on: str | Sequence[str] | None = None
 
 
-def migrate_old_data() -> None:
+def migrate_old_data() -> None:  # noqa: C901
     from json import dumps, loads
 
     Base = automap_base()  # noqa: N806
@@ -47,6 +47,9 @@ def migrate_old_data() -> None:
             TimeRemainingColumn(),
         ) as progress,
     ):
+        if session.query(OldHistoricalData).count() == 0:
+            logger.info('空表, 跳过')
+            return
         task_id = progress.add_task('[cyan]Migrating:', total=session.query(OldHistoricalData).count())
         pointer = 0
         while pointer < session.query(OldHistoricalData).order_by(desc(OldHistoricalData.id)).limit(1).one().id:

nonebot_plugin_tetris_stats/config/migrations/8a91210ce14d_correct_the_data_in_historicaldata.py

@@ -28,12 +28,6 @@ depends_on: str | Sequence[str] | None = None
 def upgrade(name: str = '') -> None:  # noqa: C901
     if name:
         return
-    from nonebot_plugin_tetris_stats.version import __version__
-
-    if __version__ != '1.0.3':
-        msg = '本迁移需要1.0.3版本, 请先锁定版本至1.0.3版本再执行本迁移'
-        logger.critical(msg)
-        raise RuntimeError(msg)
 
     from nonebot.compat import PYDANTIC_V2, type_validate_json
     from pydantic import BaseModel, ValidationError
@@ -46,10 +40,6 @@ def upgrade(name: str = '') -> None:  # noqa: C901
         TimeRemainingColumn,
     )
 
-    from nonebot_plugin_tetris_stats.game_data_processor.schemas import (  # type: ignore[import-untyped]
-        BaseProcessedData,
-    )
-
     Base = automap_base()  # noqa: N806
     Base.prepare(autoload_with=op.get_bind())
     HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
@@ -62,18 +52,33 @@ def upgrade(name: str = '') -> None:  # noqa: C901
         def model_to_json(value: BaseModel) -> str:
             return value.json(by_alias=True)
 
-    models = BaseProcessedData.__subclasses__()
-
-    def json_to_model(value: str) -> BaseModel:
-        for i in models:
-            try:
-                return type_validate_json(i, value)
-            except ValidationError:  # noqa: PERF203
-                ...
-        raise ValueError
-
     with Session(op.get_bind()) as session:
         count = session.query(HistoricalData).count()
+        if count == 0:
+            logger.info('空表, 跳过')
+            return
+
+        from nonebot_plugin_tetris_stats.version import __version__
+
+        if __version__ != '1.0.3':
+            msg = '本迁移需要1.0.3版本, 请先锁定版本至1.0.3版本再执行本迁移'
+            logger.critical(msg)
+            raise RuntimeError(msg)
+
+        from nonebot_plugin_tetris_stats.game_data_processor.schemas import (  # type: ignore[import-untyped]
+            BaseProcessedData,
+        )
+
+        models = BaseProcessedData.__subclasses__()
+
+        def json_to_model(value: str) -> BaseModel:
+            for i in models:
+                try:
+                    return type_validate_json(i, value)
+                except ValidationError:  # noqa: PERF203
+                    ...
+            raise ValueError
+
         with Progress(
             TextColumn('[progress.description]{task.description}'),
             BarColumn(),

nonebot_plugin_tetris_stats/config/migrations/b7fbdafc339a_add_user_unique_identifier_field_to_.py

@@ -26,12 +26,7 @@ depends_on: str | Sequence[str] | None = None
 def upgrade(name: str = '') -> None:
     if name:
         return
-    from nonebot_plugin_tetris_stats.version import __version__
 
-    if __version__ != '1.0.4':
-        msg = '本迁移需要1.0.4版本, 请先锁定版本至1.0.4版本再执行本迁移'
-        logger.critical(msg)
-        raise RuntimeError(msg)
     from nonebot.compat import type_validate_json
     from pydantic import ValidationError
     from rich.progress import (
@@ -46,8 +41,6 @@ def upgrade(name: str = '') -> None:
     from sqlalchemy.ext.automap import automap_base
     from sqlalchemy.orm import Session
 
-    from nonebot_plugin_tetris_stats.game_data_processor.schemas import BaseUser  # type: ignore[import-untyped]
-
     with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
         batch_op.add_column(sa.Column('user_unique_identifier', sa.String(length=32), nullable=True))
         batch_op.create_index(
@@ -60,37 +53,48 @@ def upgrade(name: str = '') -> None:
     Base.prepare(autoload_with=connection)
     HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
 
-    models: list[type[BaseUser]] = BaseUser.__subclasses__()
-
-    def json_to_model(value: str) -> BaseUser:
-        for i in models:
-            try:
-                return type_validate_json(i, value)
-            except ValidationError:  # noqa: PERF203
-                ...
-        raise ValueError
-
     with Session(op.get_bind()) as session:
         count = session.query(HistoricalData).count()
-        with Progress(
+        if count == 0:
-            TextColumn('[progress.description]{task.description}'),
+            logger.info('空表, 跳过')
-            BarColumn(),
+        else:
-            MofNCompleteColumn(),
+            from nonebot_plugin_tetris_stats.version import __version__
-            TaskProgressColumn(),
+
-            TimeRemainingColumn(),
+            if __version__ != '1.0.4':
-        ) as progress:
+                msg = '本迁移需要1.0.4版本, 请先锁定版本至1.0.4版本再执行本迁移'
-            task_id = progress.add_task('[cyan]Updateing:', total=count)
+                logger.critical(msg)
-            for i in range(0, count, 100):
+                raise RuntimeError(msg)
-                for j in session.scalars(
+            from nonebot_plugin_tetris_stats.game_data_processor.schemas import BaseUser  # type: ignore[import-untyped]
-                    select(HistoricalData).where(HistoricalData.id > i).order_by(HistoricalData.id).limit(100)
+
-                ):
+            models: list[type[BaseUser]] = BaseUser.__subclasses__()
-                    model = json_to_model(j.game_user)
+
+            def json_to_model(value: str) -> BaseUser:
+                for i in models:
                     try:
-                        j.user_unique_identifier = model.unique_identifier
+                        return type_validate_json(i, value)
-                    except ValueError:
+                    except ValidationError:  # noqa: PERF203
-                        session.delete(j)
+                        ...
-                    progress.update(task_id, advance=1)
+                raise ValueError
-                session.commit()
+
+            with Progress(
+                TextColumn('[progress.description]{task.description}'),
+                BarColumn(),
+                MofNCompleteColumn(),
+                TaskProgressColumn(),
+                TimeRemainingColumn(),
+            ) as progress:
+                task_id = progress.add_task('[cyan]Updateing:', total=count)
+                for i in range(0, count, 100):
+                    for j in session.scalars(
+                        select(HistoricalData).where(HistoricalData.id > i).order_by(HistoricalData.id).limit(100)
+                    ):
+                        model = json_to_model(j.game_user)
+                        try:
+                            j.user_unique_identifier = model.unique_identifier
+                        except ValueError:
+                            session.delete(j)
+                        progress.update(task_id, advance=1)
+                    session.commit()
     with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
         batch_op.alter_column('user_unique_identifier', existing_type=sa.VARCHAR(length=32), nullable=False)
     logger.success('database upgrade success')

nonebot_plugin_tetris_stats/games/tetrio/api/cache.py

@@ -1,5 +1,5 @@
 from asyncio import Lock
-from datetime import datetime, timezone
+from datetime import datetime, timedelta, timezone
 from typing import ClassVar
 from weakref import WeakValueDictionary
 
@@ -9,12 +9,15 @@ from nonebot.log import logger
 from yarl import URL
 
 from ....config.config import config
+from ....utils.limit import limit
 from ....utils.request import Request
 from .schemas.base import FailedModel, SuccessModel
 
 UTC = timezone.utc
 
+
 request = Request(config.tetris.proxy.tetrio or config.tetris.proxy.main)
+request.request = limit(timedelta(seconds=1))(request.request)  # type: ignore[method-assign]
 
 
 class Cache:

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

@@ -50,19 +50,21 @@ class P(BaseModel):
         return Prisecter(f'{self.pri}:{self.sec}:{self.ter}')
 
 
+# fmt: off
 class ArCounts(BaseModel):
-    bronze: int | None = Field(default=None, alias='1')
+    bronze:   int | None = Field(default=None, alias='1')    # pyright: ignore [reportGeneralTypeIssues]
-    silver: int | None = Field(default=None, alias='2')
+    silver:   int | None = Field(default=None, alias='2')    # pyright: ignore [reportGeneralTypeIssues]
-    gold: int | None = Field(default=None, alias='3')
+    gold:     int | None = Field(default=None, alias='3')    # pyright: ignore [reportGeneralTypeIssues]
-    platinum: int | None = Field(default=None, alias='4')
+    platinum: int | None = Field(default=None, alias='4')    # pyright: ignore [reportGeneralTypeIssues]
-    diamond: int | None = Field(default=None, alias='5')
+    diamond:  int | None = Field(default=None, alias='5')    # pyright: ignore [reportGeneralTypeIssues]
-    issued: int | None = Field(default=None, alias='100')
+    issued:   int | None = Field(default=None, alias='100')  # pyright: ignore [reportGeneralTypeIssues]
-    top3: int | None = Field(default=None, alias='t3')
+    top3:     int | None = Field(default=None, alias='t3')
-    top5: int | None = Field(default=None, alias='t5')
+    top5:     int | None = Field(default=None, alias='t5')
-    top10: int | None = Field(default=None, alias='t10')
+    top10:    int | None = Field(default=None, alias='t10')
-    top25: int | None = Field(default=None, alias='t25')
+    top25:    int | None = Field(default=None, alias='t25')
-    top50: int | None = Field(default=None, alias='t50')
+    top50:    int | None = Field(default=None, alias='t50')
-    top100: int | None = Field(default=None, alias='t100')
+    top100:   int | None = Field(default=None, alias='t100')
+# fmt: on
 
 
 class Cache(BaseModel):

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/summaries/league.py

@@ -37,7 +37,7 @@ class PastInner(BaseModel):
 
 
 class Past(BaseModel):
-    first: PastInner | None = Field(default=None, alias='1')
+    first: PastInner | None = Field(default=None, alias='1')  # pyright: ignore [reportGeneralTypeIssues]
 
 
 class BaseData(BaseModel):

nonebot_plugin_tetris_stats/games/tetrio/list.py

@@ -36,6 +36,7 @@ async def _(
     limit: int | None = None,
     country: str | None = None,
 ):
+    country = country.upper() if country is not None else None
     async with trigger(
         session_persist_id=await get_session_persist_id(event_session),
         game_platform=GAME_TYPE,

nonebot_plugin_tetris_stats/games/tetrio/query.py

@@ -104,7 +104,7 @@ async def _(  # noqa: PLR0913
         session_persist_id=await get_session_persist_id(event_session),
         game_platform=GAME_TYPE,
         command_type='query',
-        command_args=[f'--default-template {template}'] if template is not None else [],
+        command_args=[f'--template {template}'] if template is not None else [],
     ):
         async with get_session() as session:
             bind = await query_bind_info(
@@ -131,7 +131,7 @@ async def _(user: NBUser, account: Player, event_session: EventSession, template
         session_persist_id=await get_session_persist_id(event_session),
         game_platform=GAME_TYPE,
         command_type='query',
-        command_args=[f'--default-template {template}'] if template is not None else [],
+        command_args=[f'--template {template}'] if template is not None else [],
     ):
         async with get_session() as session:
             if template is None:

nonebot_plugin_tetris_stats/games/tetrio/rank/__init__.py

@@ -13,7 +13,6 @@ from nonebot_plugin_orm import get_session
 from sqlalchemy import select
 
 from ....utils.exception import RequestError
-from ....utils.limit import limit
 from ....utils.retry import retry
 from .. import alc
 from .. import command as base_command
@@ -84,11 +83,11 @@ def find_special_player(
 @scheduler.scheduled_job('cron', hour='0,6,12,18', minute=0)
 async def get_tetra_league_data() -> None:
     x_session_id = uuid4()
-    limit_by = retry(max_attempts=10, exception_type=RequestError)(limit(timedelta(seconds=1))(by))
+    retry_by = retry(max_attempts=10, exception_type=RequestError)(by)
     prisecter = P(pri=9007199254740991, sec=9007199254740991, ter=9007199254740991)  # * from ch.tetr.io
     results: list[BySuccessModel] = []
     while True:
-        model = await limit_by('league', Parameter(after=prisecter.to_prisecter(), limit=100), x_session_id)
+        model = await retry_by('league', Parameter(after=prisecter.to_prisecter(), limit=100), x_session_id)
         prisecter = model.data.entries[-1].p
         results.append(model)
         if len(model.data.entries) < 100:  # 分页值 # noqa: PLR2004

nonebot_plugin_tetris_stats/games/tetrio/rank/all.py

@@ -32,12 +32,12 @@ command.add(
 
 
 @alc.assign('TETRIO.rank.all')
-async def _(event_session: EventSession, template: Template = 'v1'):
+async def _(event_session: EventSession, template: Template | None = None):
     async with trigger(
         session_persist_id=await get_session_persist_id(event_session),
         game_platform=GAME_TYPE,
         command_type='rank',
-        command_args=['--all'],
+        command_args=['--all'] + ([f'--template {template}'] if template is not None else []),
     ):
         async with get_session() as session:
             latest_data = (
@@ -62,7 +62,7 @@ async def _(event_session: EventSession, template: Template = 'v1'):
                 )
             ).one()
         match template:
-            case 'v1':
+            case 'v1' | None:
                 await UniMessage.image(raw=await make_image_v1(latest_data, compare_data)).finish()
             case 'v2':
                 await UniMessage.image(raw=await make_image_v2(latest_data, compare_data)).finish()

nonebot_plugin_tetris_stats/games/tos/api/player.py

@@ -59,13 +59,14 @@ class Player:
     async def get_info(self) -> UserInfoSuccess:
         """获取用户信息"""
         if self._user_info is None:
-            path = str(
+            if self.teaid is not None:
-                URL('getTeaIdInfo') % {'teaId': self.teaid}
+                path = 'getTeaIdInfo'
-                if self.teaid is not None
+                query = {'teaId': self.teaid}
-                else URL('getUsernameInfo') % {'username': cast(str, self.user_name)}
+            else:
-            )
+                path = 'getUsernameInfo'
+                query = {'username': cast(str, self.user_name)}
             raw_user_info = await request.failover_request(
-                [i / path for i in BASE_URL], failover_code=[502], failover_exc=(TimeoutException,)
+                [i / path % query for i in BASE_URL], failover_code=[502], failover_exc=(TimeoutException,)
             )
             user_info: UserInfo = type_validate_json(UserInfo, raw_user_info)  # type: ignore[arg-type]
             if not isinstance(user_info, UserInfoSuccess):

nonebot_plugin_tetris_stats/utils/limit.py

@@ -23,7 +23,7 @@ def limit(limit: timedelta) -> Callable[[Callable[P, Coroutine[Any, Any, T]]], C
             nonlocal last_call
             async with lock:
                 if (diff := (time() - last_call)) < limit_seconds:
-                    logger.debug(f'request limit {(limit_time:=limit_seconds-diff)}s')
+                    logger.debug(f'func: {func.__name__} trigger limit, wait {(limit_time:=limit_seconds-diff):.3f}s')
                     await sleep(limit_time)
             last_call = time()
             return await func(*args, **kwargs)

nonebot_plugin_tetris_stats/utils/typing.py

@@ -1,11 +1,11 @@
-from typing import Literal
+from typing import Literal, TypeAlias
 
-Number = float | int
+Number: TypeAlias = float | int
-GameType = Literal['IO', 'TOP', 'TOS']
+GameType: TypeAlias = Literal['IO', 'TOP', 'TOS']
-BaseCommandType = Literal['bind', 'query']
+BaseCommandType: TypeAlias = Literal['bind', 'query']
-TETRIOCommandType = BaseCommandType | Literal['rank', 'config', 'list', 'record']
+TETRIOCommandType: TypeAlias = BaseCommandType | Literal['rank', 'config', 'list', 'record']
-AllCommandType = BaseCommandType | TETRIOCommandType
+AllCommandType: TypeAlias = BaseCommandType | TETRIOCommandType
-Me = Literal[
+Me: TypeAlias = Literal[
     '我',
     '自己',
     '我等',

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = 'nonebot-plugin-tetris-stats'
-version = '1.5.2'
+version = "1.5.5"
 description = '一款基于 NoneBot2 的用于查询 Tetris 相关游戏数据的插件'
 authors = ['scdhh <wallfjjd@gmail.com>']
 readme = 'README.md'
@@ -21,20 +21,21 @@ nonebot-plugin-user = '>=0.2,<0.5'
 nonebot-plugin-userinfo = '^0.2.4'
 aiocache = '^0.12.2'
 aiofiles = '>=23.2.1,<25.0.0'
+arclet-alconna = '<2.0.0'
 async-lru = '^2.0.4'
 httpx = '^0.27.0'
 jinja2 = '^3.1.3'
 lxml = '^5.1.0'
-msgspec = "^0.18.6"
+msgspec = '^0.18.6'
 pandas = '>=1.4.3,<3.0.0'
-pillow = '^10.3.0'
+pillow = '>=10.3,<12.0'
 playwright = '^1.41.2'
 rich = '^13.7.1'
-yarl = "^1.9.4"
+yarl = '^1.9.4'
 zstandard = '>=0.22,<0.24'
 
 [tool.poetry.group.dev.dependencies]
-basedpyright = "^1.17.0"
+basedpyright = '^1.17.0'
 mypy = '>=1.9'
 pandas-stubs = '>=1.5.2,<3.0.0'
 ruff = '>=0.3.0'
@@ -51,11 +52,11 @@ nonebot-adapter-satori = '>=0.11.4,<0.13.0'
 nonebot-plugin-orm = { extras = ['default'], version = '>=0.3,<0.8' }
 
 [tool.poetry.group.debug.dependencies]
-matplotlib = "^3.9.2"
+matplotlib = '^3.9.2'
 memory-profiler = '^0.61.0'
 objprint = '^0.2.2'
-pyqt6 = "^6.7.1"
+pyqt6 = '^6.7.1'
-viztracer = '^0.16.2'
+viztracer = '>=0.16.2,<0.18.0'
 
 [build-system]
 requires = ['poetry-core>=1.0.0']
@@ -136,10 +137,10 @@ builtins-ignorelist = ['id']
 quote-style = 'single'
 
 [tool.basedpyright]
-pythonVersion = "3.10"
+pythonVersion = '3.10'
-pythonPlatform = "All"
+pythonPlatform = 'All'
 defineConstant = { PYDANTIC_V2 = true }
-typeCheckingMode = "standard"
+typeCheckingMode = 'standard'
 
 [tool.nonebot]
 plugins = ['nonebot_plugin_tetris_stats']