🔖 1.0.0.a1.post1

.github/dependabot.yml

@@ -7,6 +7,6 @@ version: 2
 updates:
   - package-ecosystem: "pip" # See documentation for possible values
     directory: "/" # Location of package manifests
-    target-branch: "main"
+    target-branch: "dev"
     schedule:
       interval: "daily"

.github/workflows/Release.yml

@@ -8,42 +8,18 @@ on:
 jobs:
   release:
     runs-on: ubuntu-latest
-    permissions:
-      id-token: write
-      contents: write
     steps:
       - uses: actions/checkout@v3
-
-      - name: Install poetry
-        run: pipx install poetry
-        shell: bash
-
       - uses: actions/setup-python@v4
         with:
           python-version: '3.11'
-          cache: "poetry"
+      - name: Install Poetry
-
-      - run: poetry install
         shell: bash
-
-      - name: Get Version
-        id: version
         run: |
-          echo "VERSION=$(poetry version -s)" >> $GITHUB_OUTPUT
+          pip install poetry
-          echo "TAG_VERSION=${GITHUB_REF#refs/tags/}" >> $GITHUB_OUTPUT
+      - name: Build
-          echo "TAG_NAME=${GITHUB_REF#refs/tags/}" >> $GITHUB_OUTPUT
+        shell: bash
-
+        run: |
-      - name: Check Version
+          poetry install
-        if: steps.version.outputs.VERSION != steps.version.outputs.TAG_VERSION
+          poetry env use python
-        run: exit 1
+          poetry publish --build -u ${{ secrets.USERNAME }} -p ${{ secrets.PASSWORD }} -n
-
-      - name: Build Package
-        run: poetry build
-
-      - name: Publish Package to PyPI
-        uses: pypa/gh-action-pypi-publish@release/v1
-
-      - name: Publish Package to GitHub Release
-        run: gh release create ${{ steps.version.outputs.TAG_NAME }} dist/*.tar.gz dist/*.whl -t "🔖 ${{ steps.version.outputs.TAG_NAME }}" --generate-notes
-        env:
-          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/TypeCheck.yml

@@ -1,27 +0,0 @@
-name: TypeCheck
-
-on:
-    push:
-
-jobs:
-    Mypy:
-        runs-on: ubuntu-latest
-        steps:
-            - uses: actions/checkout@v4
-
-            - name: Install poetry
-              run: pipx install poetry
-              shell: bash
-
-            - uses: actions/setup-python@v5
-              with:
-                  python-version: '3.10'
-                  cache: 'poetry'
-
-            - run: poetry install
-              shell: bash
-
-            - name: Run Mypy
-              shell: bash
-              run: |
-                  poetry run mypy ./nonebot_plugin_tetris_stats

.gitignore

@@ -5,20 +5,6 @@ Untitled*
 *copy*
 .vscode
 *dev*
-*_cache*
+*cache*
 *backup*
 *.pyc
-node_modules
-.prettier*
-package.json
-pnpm-lock.yaml
-*.drawio.svg
-package-lock.json
-*Zone.Identifier
-.env*
-bot.py
-TODO
-*.fish
-extracted_skin_mino_*
-sample_*
-TODO*

.pre-commit-config.yaml

@@ -1,22 +0,0 @@
-default_install_hook_types: [pre-commit, prepare-commit-msg]
-ci:
-    autofix_commit_msg: ':rotating_light: auto fix by pre-commit hooks'
-    autofix_prs: true
-    autoupdate_branch: main
-    autoupdate_schedule: weekly
-    autoupdate_commit_msg: ':arrow_up: auto update by pre-commit hooks'
-repos:
-    - repo: https://github.com/astral-sh/ruff-pre-commit
-      rev: v0.6.1
-      hooks:
-          - id: ruff
-            args: [--fix, --exit-non-zero-on-fix]
-            stages: [commit]
-          - id: ruff-format
-            stages: [commit]
-
-    - repo: https://github.com/nonebot/nonemoji
-      rev: v0.1.4
-      hooks:
-          - id: nonemoji
-            stages: [prepare-commit-msg]

README.md

@@ -87,13 +87,3 @@ pip install nonebot-plugin-tetris-stats
 ## 📝 开源
 
 本项目使用 [AGPL-3.0](https://github.com/shoucandanghehe/nonebot-plugin-tetris-stats/blob/main/LICENSE) 许可证开源
-
-## 🤓☝ 给个 star 吧
-
-<a href="https://star-history.com/#A-Minos/nonebot-plugin-tetris-stats&Date">
- <picture>
-   <source media="(prefers-color-scheme: dark)" srcset="https://api.star-history.com/svg?repos=A-Minos/nonebot-plugin-tetris-stats&type=Date&theme=dark" />
-   <source media="(prefers-color-scheme: light)" srcset="https://api.star-history.com/svg?repos=A-Minos/nonebot-plugin-tetris-stats&type=Date" />
-   <img alt="Star History Chart" src="https://api.star-history.com/svg?repos=A-Minos/nonebot-plugin-tetris-stats&type=Date" />
- </picture>
-</a>

nonebot_plugin_tetris_stats/__init__.py

@@ -1,39 +1,22 @@
 from nonebot import require
-from nonebot.plugin import PluginMetadata, inherit_supported_adapters
+from nonebot.plugin import PluginMetadata
 
-require_plugins = {
+require('nonebot_plugin_localstore')
-    'nonebot_plugin_alconna',
+require('nonebot_plugin_orm')
-    'nonebot_plugin_apscheduler',
+require('nonebot_plugin_alconna')
-    'nonebot_plugin_localstore',
+require('nonebot_plugin_apscheduler')
-    'nonebot_plugin_orm',
-    'nonebot_plugin_session_orm',
-    'nonebot_plugin_session',
-    'nonebot_plugin_user',
-    'nonebot_plugin_userinfo',
-}
 
-for i in require_plugins:
+from .config.config import migrations  # noqa: E402
-    require(i)
-
-from nonebot_plugin_alconna import namespace  # noqa: E402
-
-with namespace('tetris_stats') as ns:
-    ns.enable_message_cache = False
-
-from .config import migrations  # noqa: E402
-from .config.config import Config  # noqa: E402
 
 __plugin_meta__ = PluginMetadata(
     name='Tetris Stats',
     description='一个用于查询 Tetris 相关游戏玩家数据的插件',
-    usage='发送 tstats --help 查询使用方法',
+    usage='发送 {游戏名} --help 查询使用方法',
     type='application',
-    homepage='https://github.com/A-minos/nonebot-plugin-tetris-stats',
+    homepage='https://github.com/shoucandanghehe/nonebot-plugin-tetris-stats',
-    config=Config,
-    supported_adapters=inherit_supported_adapters(*require_plugins),
     extra={
         'orm_version_location': migrations,
     },
 )
 
-from . import games  # noqa: F401, E402
+from . import game_data_processor  # noqa: F401, E402

nonebot_plugin_tetris_stats/config/config.py

@@ -1,27 +1,14 @@
-from nonebot import get_plugin_config
+from pathlib import Path
-from nonebot_plugin_localstore import get_cache_dir, get_data_dir
-from pydantic import BaseModel, Field
 
-CACHE_PATH = get_cache_dir('nonebot_plugin_tetris_stats')
+from nonebot_plugin_localstore import get_cache_dir  # type: ignore[import-untyped]
-DATA_PATH = get_data_dir('nonebot_plugin_tetris_stats')
+from pydantic import BaseModel
 
+from . import migrations  # noqa: F401
 
-class Proxy(BaseModel):
+CACHE_PATH: Path = get_cache_dir('nonebot_plugin_tetris_stats')
-    main: str | None = None
-    github: str | None = None
-    tetrio: str | None = None
-    tos: str | None = None
-    top: str | None = None
-
-
-class ScopedConfig(BaseModel):
-    request_timeout: float = 30.0
-    screenshot_quality: float = 2
-    proxy: Proxy = Field(default_factory=Proxy)
 
 
 class Config(BaseModel):
-    tetris: ScopedConfig = Field(default_factory=ScopedConfig)
+    """配置类"""
 
-
+    db_url: str = 'sqlite://data/nonebot_plugin_tetris_stats/data.db'
-config = get_plugin_config(Config)

nonebot_plugin_tetris_stats/config/migrations/09d4bb60160d_rename_field.py

@@ -1,54 +0,0 @@
-"""Rename field
-
-迁移 ID: 09d4bb60160d
-父迁移: b9d65badc713
-创建时间: 2024-04-23 23:42:04.541672
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import sqlalchemy as sa
-from alembic import op
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = '09d4bb60160d'
-down_revision: str | Sequence[str] | None = 'b9d65badc713'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
-        batch_op.alter_column('create_time', new_column_name='update_time', existing_type=sa.DateTime())
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_iorank_create_time')
-    op.create_index(
-        batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_update_time'),
-        'nonebot_plugin_tetris_stats_iorank',
-        ['update_time'],
-        unique=False,
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
-        batch_op.alter_column('update_time', new_column_name='create_time')
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_update_time'))
-    op.create_index(
-        'ix_nonebot_plugin_tetris_stats_iorank_create_time',
-        'nonebot_plugin_tetris_stats_iorank',
-        ['create_time'],
-        unique=False,
-    )
-    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/0d50142b780f_add_field.py

@@ -1,46 +0,0 @@
-"""add field
-
-迁移 ID: 0d50142b780f
-父迁移: 09d4bb60160d
-创建时间: 2024-04-24 14:55:08.064098
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import sqlalchemy as sa
-from alembic import op
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = '0d50142b780f'
-down_revision: str | Sequence[str] | None = '09d4bb60160d'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
-        batch_op.add_column(sa.Column('file_hash', sa.String(length=128), nullable=True))
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_file_hash'), ['file_hash'], unique=False
-        )
-
-    # ### end Alembic commands ###
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_iorank_file_hash'))
-        batch_op.drop_column('file_hash')
-
-    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/3c25a5a8c050_refactor_historical.py

@@ -1,280 +0,0 @@
-"""Refactor Historical
-
-迁移 ID: 3c25a5a8c050
-父迁移: b7fbdafc339a
-创建时间: 2024-05-14 09:16:35.193001
-
-"""
-
-from __future__ import annotations
-
-from datetime import datetime
-from typing import TYPE_CHECKING, Any
-
-import sqlalchemy as sa
-from alembic import op
-from nonebot.log import logger
-from rich.progress import BarColumn, MofNCompleteColumn, Progress, TaskProgressColumn, TextColumn, TimeRemainingColumn
-from sqlalchemy import desc, select
-from sqlalchemy.dialects import sqlite
-from sqlalchemy.ext.automap import automap_base
-from sqlalchemy.orm import Session
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = '3c25a5a8c050'
-down_revision: str | Sequence[str] | None = 'b7fbdafc339a'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def migrate_old_data() -> None:
-    from json import dumps, loads
-
-    Base = automap_base()  # noqa: N806
-    Base.prepare(autoload_with=op.get_bind())
-    OldHistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
-    TETRIOHistoricalData = Base.classes.nonebot_plugin_tetris_stats_tetriohistoricaldata  # noqa: N806
-    TOSHistoricalData = Base.classes.nonebot_plugin_tetris_stats_toshistoricaldata  # noqa: N806
-    with (
-        Session(op.get_bind()) as session,
-        Progress(
-            TextColumn('[progress.description]{task.description}'),
-            BarColumn(),
-            MofNCompleteColumn(),
-            TaskProgressColumn(),
-            TimeRemainingColumn(),
-        ) as progress,
-    ):
-        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:
-            result = session.scalars(
-                select(OldHistoricalData)
-                .where(OldHistoricalData.id > pointer)
-                .order_by(OldHistoricalData.id)
-                .limit(100)
-            ).all()
-            for j in result:
-                processed_data: dict[str, Any] = loads(j.processed_data)
-                if j.game_platform == 'IO':
-                    if (data := processed_data.get('user_info')) is not None:
-                        session.add(
-                            TETRIOHistoricalData(
-                                user_unique_identifier=j.user_unique_identifier,
-                                api_type='User Info',
-                                data=dumps(data),
-                                update_time=datetime.fromisoformat(data['cache']['cached_at']),
-                            )
-                        )
-                    if (data := processed_data.get('user_records')) is not None:
-                        session.add(
-                            TETRIOHistoricalData(
-                                user_unique_identifier=j.user_unique_identifier,
-                                api_type='User Records',
-                                data=dumps(data),
-                                update_time=datetime.fromisoformat(data['cache']['cached_at']),
-                            )
-                        )
-                if j.game_platform == 'TOS' and not j.user_unique_identifier.isdigit():
-                    if (data := processed_data.get('user_info')) is not None:
-                        session.add(
-                            TOSHistoricalData(
-                                user_unique_identifier=j.user_unique_identifier,
-                                api_type='User Info',
-                                data=dumps(data),
-                                update_time=j.finish_time,
-                            )
-                        )
-                    if (data := processed_data.get('user_profile')) is not None:
-                        for v in data.values():
-                            session.add(
-                                TOSHistoricalData(
-                                    user_unique_identifier=j.user_unique_identifier,
-                                    api_type='User Profile',
-                                    data=dumps(v),
-                                    update_time=j.finish_time,
-                                )
-                            )
-                progress.update(task_id, advance=1)
-            session.commit()
-            pointer = result[-1].id
-    logger.success('Migrate successfully')
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table(
-        'nonebot_plugin_tetris_stats_tetriohistoricaldata',
-        sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('user_unique_identifier', sa.String(length=24), nullable=False),
-        sa.Column('api_type', sa.String(length=16), nullable=False),
-        sa.Column('data', sa.JSON(), nullable=False),
-        sa.Column('update_time', sa.DateTime(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_tetriohistoricaldata')),
-        info={'bind_key': 'nonebot_plugin_tetris_stats'},
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_tetriohistoricaldata', schema=None) as batch_op:
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_api_type'), ['api_type'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_update_time'), ['update_time'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_user_unique_identifier'),
-            ['user_unique_identifier'],
-            unique=False,
-        )
-
-    op.create_table(
-        'nonebot_plugin_tetris_stats_tophistoricaldata',
-        sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('user_unique_identifier', sa.String(length=24), nullable=False),
-        sa.Column('api_type', sa.String(length=16), nullable=False),
-        sa.Column('data', sa.JSON(), nullable=False),
-        sa.Column('update_time', sa.DateTime(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_tophistoricaldata')),
-        info={'bind_key': 'nonebot_plugin_tetris_stats'},
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_tophistoricaldata', schema=None) as batch_op:
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_api_type'), ['api_type'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_update_time'), ['update_time'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_user_unique_identifier'),
-            ['user_unique_identifier'],
-            unique=False,
-        )
-
-    op.create_table(
-        'nonebot_plugin_tetris_stats_toshistoricaldata',
-        sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('user_unique_identifier', sa.String(length=24), nullable=False),
-        sa.Column('api_type', sa.String(length=16), nullable=False),
-        sa.Column('data', sa.JSON(), nullable=False),
-        sa.Column('update_time', sa.DateTime(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_toshistoricaldata')),
-        info={'bind_key': 'nonebot_plugin_tetris_stats'},
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_toshistoricaldata', schema=None) as batch_op:
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_api_type'), ['api_type'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_update_time'), ['update_time'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_user_unique_identifier'),
-            ['user_unique_identifier'],
-            unique=False,
-        )
-
-    op.create_table(
-        'nonebot_plugin_tetris_stats_triggerhistoricaldata',
-        sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('trigger_time', sa.DateTime(), nullable=False),
-        sa.Column('session_persist_id', sa.Integer(), nullable=False),
-        sa.Column('game_platform', sa.String(length=32), nullable=False),
-        sa.Column('command_type', sa.String(length=16), nullable=False),
-        sa.Column('command_args', sa.JSON(), nullable=False),
-        sa.Column('finish_time', sa.DateTime(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_triggerhistoricaldata')),
-        info={'bind_key': 'nonebot_plugin_tetris_stats'},
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_triggerhistoricaldata', schema=None) as batch_op:
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_triggerhistoricaldata_command_type'),
-            ['command_type'],
-            unique=False,
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_triggerhistoricaldata_game_platform'),
-            ['game_platform'],
-            unique=False,
-        )
-
-    migrate_old_data()
-
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_command_type')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_source_account')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_source_type')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_user_unique_identifier')
-
-    op.drop_table('nonebot_plugin_tetris_stats_historicaldata')
-    # ### end Alembic commands ###
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table(
-        'nonebot_plugin_tetris_stats_historicaldata',
-        sa.Column('id', sa.INTEGER(), nullable=False),
-        sa.Column('trigger_time', sa.DATETIME(), nullable=False),
-        sa.Column('bot_platform', sa.VARCHAR(length=32), nullable=True),
-        sa.Column('bot_account', sa.VARCHAR(), nullable=True),
-        sa.Column('source_type', sa.VARCHAR(length=32), nullable=True),
-        sa.Column('source_account', sa.VARCHAR(), nullable=True),
-        sa.Column('message', sa.BLOB(), nullable=True),
-        sa.Column('game_platform', sa.VARCHAR(length=32), nullable=False),
-        sa.Column('command_type', sa.VARCHAR(length=16), nullable=False),
-        sa.Column('command_args', sqlite.JSON(), nullable=False),
-        sa.Column('game_user', sqlite.JSON(), nullable=False),
-        sa.Column('processed_data', sqlite.JSON(), nullable=False),
-        sa.Column('finish_time', sa.DATETIME(), nullable=False),
-        sa.Column('user_unique_identifier', sa.VARCHAR(length=32), nullable=False),
-        sa.PrimaryKeyConstraint('id', name='pk_nonebot_plugin_tetris_stats_historicaldata'),
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_user_unique_identifier',
-            ['user_unique_identifier'],
-            unique=False,
-        )
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_source_type', ['source_type'], unique=False
-        )
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_source_account', ['source_account'], unique=False
-        )
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_game_platform', ['game_platform'], unique=False
-        )
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_command_type', ['command_type'], unique=False
-        )
-
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_triggerhistoricaldata', schema=None) as batch_op:
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_triggerhistoricaldata_game_platform'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_triggerhistoricaldata_command_type'))
-
-    op.drop_table('nonebot_plugin_tetris_stats_triggerhistoricaldata')
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_toshistoricaldata', schema=None) as batch_op:
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_user_unique_identifier'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_update_time'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_toshistoricaldata_api_type'))
-
-    op.drop_table('nonebot_plugin_tetris_stats_toshistoricaldata')
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_tophistoricaldata', schema=None) as batch_op:
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_user_unique_identifier'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_update_time'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tophistoricaldata_api_type'))
-
-    op.drop_table('nonebot_plugin_tetris_stats_tophistoricaldata')
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_tetriohistoricaldata', schema=None) as batch_op:
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_user_unique_identifier'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_update_time'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_api_type'))
-
-    op.drop_table('nonebot_plugin_tetris_stats_tetriohistoricaldata')
-    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/6c3206f90cc3_add_field.py

@@ -1,70 +0,0 @@
-"""Add redundant platform field
-
-迁移 ID: 6c3206f90cc3
-父迁移: 9f6582279ce2
-创建时间: 2023-11-26 20:15:56.033892
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from alembic import op
-from sqlalchemy.ext.automap import automap_base
-from sqlalchemy.orm import Session
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = '6c3206f90cc3'
-down_revision: str | Sequence[str] | None = '9f6582279ce2'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    from json import dumps, loads
-
-    Base = automap_base()  # noqa: N806
-    connection = op.get_bind()
-    Base.prepare(autoload_with=connection)
-
-    HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
-
-    with Session(connection) as session:
-        for row in session.query(HistoricalData):
-            platform = row.game_platform
-            game_user = loads(row.game_user)
-            processed_data = loads(row.processed_data)
-            game_user['platform'] = platform
-            processed_data['platform'] = platform
-            row.game_user = dumps(game_user)
-            row.processed_data = dumps(processed_data)
-            session.add(row)
-        session.commit()
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    from json import dumps, loads
-
-    Base = automap_base()  # noqa: N806
-    connection = op.get_bind()
-    Base.prepare(autoload_with=connection)
-
-    HistoricalData = Base.classes.nonebot_plugin_tetris_stats_historicaldata  # noqa: N806
-
-    with Session(connection) as session:
-        for row in session.query(HistoricalData):
-            game_user = loads(row.game_user)
-            processed_data = loads(row.processed_data)
-            game_user.pop('platform', None)
-            processed_data.pop('platform', None)
-            row.game_user = dumps(game_user)
-            row.processed_data = dumps(processed_data)
-            session.add(row)
-        session.commit()

nonebot_plugin_tetris_stats/config/migrations/8a91210ce14d_correct_the_data_in_historicaldata.py

@@ -1,98 +0,0 @@
-"""Correct the data in HistoricalData
-
-迁移 ID: 8a91210ce14d
-父迁移: 0d50142b780f
-创建时间: 2024-05-06 08:16:38.487214
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from alembic import op
-from nonebot.log import logger
-from sqlalchemy import select
-from sqlalchemy.ext.automap import automap_base
-from sqlalchemy.orm import Session
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = '8a91210ce14d'
-down_revision: str | Sequence[str] | None = '0d50142b780f'
-branch_labels: str | Sequence[str] | None = None
-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
-    from rich.progress import (
-        BarColumn,
-        MofNCompleteColumn,
-        Progress,
-        TaskProgressColumn,
-        TextColumn,
-        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
-    if PYDANTIC_V2:
-
-        def model_to_json(value: BaseModel) -> str:
-            return value.model_dump_json(by_alias=True)
-    else:
-
-        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()
-        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.processed_data)
-                    j.processed_data = model_to_json(model)
-                    progress.update(task_id, advance=1)
-                session.commit()
-    logger.success('Corrected HistoricalData')
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return

nonebot_plugin_tetris_stats/config/migrations/9866f53ce44f_init_db.py

@@ -5,17 +5,13 @@
 创建时间: 2023-11-11 16:24:11.826667
 
 """
-
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
 
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
 revision: str = '9866f53ce44f'
 down_revision: str | Sequence[str] | None = None
 branch_labels: str | Sequence[str] | None = ('nonebot_plugin_tetris_stats',)

nonebot_plugin_tetris_stats/config/migrations/9cd1647db502_merge_old_db.py

@@ -5,12 +5,11 @@
 创建时间: 2023-11-11 16:51:30.718277
 
 """
-
 from __future__ import annotations
 
+from collections.abc import Sequence
 from pathlib import Path
 from shutil import copyfile
-from typing import TYPE_CHECKING
 
 from alembic import op
 from nonebot import get_driver
@@ -19,9 +18,6 @@ from sqlalchemy import Connection, create_engine, inspect, text
 from sqlalchemy.ext.automap import automap_base
 from sqlalchemy.orm import Session
 
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
 revision: str = '9cd1647db502'
 down_revision: str | Sequence[str] | None = '9866f53ce44f'
 branch_labels: str | Sequence[str] | None = None
@@ -37,10 +33,14 @@ def migrate_old_data(connection: Connection) -> None:
     Bind = Base.classes.nonebot_plugin_tetris_stats_bind  # noqa: N806
 
     def non_empty(obj: str) -> bool:
-        return bool(obj != '' and not obj.isspace())
+        if obj != '' and not obj.isspace():
+            return True
+        return False
 
     def is_int(obj: int | str) -> bool:
-        return bool(isinstance(obj, int) or obj.isdigit())
+        if isinstance(obj, int) or obj.isdigit():
+            return True
+        return False
 
     bind_list = [
         Bind(chat_platform='OneBot V11', chat_account=int(row.QQ), game_platform='IO', game_account=row.USER)
@@ -64,7 +64,7 @@ def upgrade(name: str = '') -> None:
     if name:
         return
     try:
-        db_path = Path(config.db_url)
+        db_path = Path(config.db_path)
     except AttributeError:
         db_path = Path('data/nonebot_plugin_tetris_stats/data.db')
     if db_path.exists() is False:
@@ -80,12 +80,11 @@ def upgrade(name: str = '') -> None:
             logger.success('nonebot_plugin_tetris_stats: 跳过迁移')
             return
         if 'IORANK' not in tables:
-            msg = 'nonebot_plugin_tetris_stats: 请先安装 0.4.4 版本完成迁移之后再升级'
+            logger.warning('nonebot_plugin_tetris_stats: 发现过早版本的数据, 请先更新到 0.4.4 版本')
-            logger.warning(msg)
+            raise RuntimeError('nonebot_plugin_tetris_stats: 请先安装 0.4.4 版本完成迁移之后再升级')
-            raise RuntimeError(msg)
         logger.info('nonebot_plugin_tetris_stats: 发现来自老版本的数据, 正在迁移...')
         migrate_old_data(connection)
-    db_path.unlink()
+        db_path.unlink()
 
 
 def downgrade(name: str = '') -> None:

nonebot_plugin_tetris_stats/config/migrations/9f6582279ce2_recreate_historical_data.py

@@ -1,114 +0,0 @@
-"""Recreate HistoricalData
-
-迁移 ID: 9f6582279ce2
-父迁移: 9cd1647db502
-创建时间: 2023-11-21 08:35:50.393246
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import sqlalchemy as sa
-from alembic import op
-from sqlalchemy.dialects import sqlite
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = '9f6582279ce2'
-down_revision: str | Sequence[str] | None = '9cd1647db502'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_command_type')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_source_account')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_historicaldata_source_type')
-
-    op.drop_table('nonebot_plugin_tetris_stats_historicaldata')
-
-    op.create_table(
-        'nonebot_plugin_tetris_stats_historicaldata',
-        sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('trigger_time', sa.DateTime(), nullable=False),
-        sa.Column('bot_platform', sa.String(length=32), nullable=True),
-        sa.Column('bot_account', sa.String(), nullable=True),
-        sa.Column('source_type', sa.String(length=32), nullable=True),
-        sa.Column('source_account', sa.String(), nullable=True),
-        sa.Column('message', sa.PickleType(), nullable=True),
-        sa.Column('game_platform', sa.String(length=32), nullable=False),
-        sa.Column('command_type', sa.String(length=16), nullable=False),
-        sa.Column('command_args', sa.JSON(), nullable=False),
-        sa.Column('game_user', sa.JSON(), nullable=False),
-        sa.Column('processed_data', sa.JSON(), nullable=False),
-        sa.Column('finish_time', sa.DateTime(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_historicaldata')),
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_command_type'), ['command_type'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform'), ['game_platform'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_account'), ['source_account'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_type'), ['source_type'], unique=False
-        )
-    # ### end Alembic commands ###
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_type'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_source_account'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_game_platform'))
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_command_type'))
-
-    op.drop_table('nonebot_plugin_tetris_stats_historicaldata')
-
-    op.create_table(
-        'nonebot_plugin_tetris_stats_historicaldata',
-        sa.Column('id', sa.INTEGER(), nullable=False),
-        sa.Column('trigger_time', sa.DATETIME(), nullable=False),
-        sa.Column('bot_platform', sa.VARCHAR(length=32), nullable=True),
-        sa.Column('bot_account', sa.VARCHAR(), nullable=True),
-        sa.Column('source_type', sa.VARCHAR(length=32), nullable=True),
-        sa.Column('source_account', sa.VARCHAR(), nullable=True),
-        sa.Column('message', sa.BLOB(), nullable=True),
-        sa.Column('game_platform', sa.VARCHAR(length=32), nullable=False),
-        sa.Column('command_type', sa.VARCHAR(length=16), nullable=False),
-        sa.Column('command_args', sqlite.JSON(), nullable=False),
-        sa.Column('game_user', sa.BLOB(), nullable=False),
-        sa.Column('processed_data', sa.BLOB(), nullable=False),
-        sa.Column('finish_time', sa.DATETIME(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name='pk_nonebot_plugin_tetris_stats_historicaldata'),
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_source_type', ['source_type'], unique=False
-        )
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_source_account', ['source_account'], unique=False
-        )
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_game_platform', ['game_platform'], unique=False
-        )
-        batch_op.create_index(
-            'ix_nonebot_plugin_tetris_stats_historicaldata_command_type', ['command_type'], unique=False
-        )
-
-    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/a1195e989cc6_add_tetrio_user_configuration.py

@@ -1,44 +0,0 @@
-"""Add TETRIO user configuration
-
-迁移 ID: a1195e989cc6
-父迁移: b15844837693
-创建时间: 2024-06-09 04:20:07.819194
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import sqlalchemy as sa
-from alembic import op
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = 'a1195e989cc6'
-down_revision: str | Sequence[str] | None = 'b15844837693'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table(
-        'nonebot_plugin_tetris_stats_tetriouserconfig',
-        sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('query_template', sa.String(length=2), nullable=False),
-        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_tetriouserconfig')),
-        info={'bind_key': 'nonebot_plugin_tetris_stats'},
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_table('nonebot_plugin_tetris_stats_tetriouserconfig')
-    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/b15844837693_migrate_to_nonobot_plugin_user.py

@@ -1,71 +0,0 @@
-"""Migrate to nonobot-plugin-user
-
-迁移 ID: b15844837693
-父迁移: 3c25a5a8c050
-创建时间: 2024-06-08 02:27:35.227596
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import sqlalchemy as sa
-from alembic import op
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = 'b15844837693'
-down_revision: str | Sequence[str] | None = '3c25a5a8c050'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_bind_chat_account')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_bind_chat_platform')
-
-    op.drop_table('nonebot_plugin_tetris_stats_bind')
-
-    op.create_table(
-        'nonebot_plugin_tetris_stats_bind',
-        sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('user_id', sa.Integer(), nullable=False),
-        sa.Column('game_platform', sa.String(length=32), nullable=False),
-        sa.Column('game_account', sa.String(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_bind')),
-        info={'bind_key': 'nonebot_plugin_tetris_stats'},
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
-        batch_op.create_index(batch_op.f('ix_nonebot_plugin_tetris_stats_bind_user_id'), ['user_id'], unique=False)
-    # ### end Alembic commands ###
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_bind_user_id'))
-
-    op.drop_table('nonebot_plugin_tetris_stats_bind')
-
-    op.create_table(
-        'nonebot_plugin_tetris_stats_bind',
-        sa.Column('id', sa.INTEGER(), nullable=False),
-        sa.Column('chat_platform', sa.VARCHAR(length=32), nullable=False),
-        sa.Column('chat_account', sa.VARCHAR(), nullable=False),
-        sa.Column('game_platform', sa.VARCHAR(length=32), nullable=False),
-        sa.Column('game_account', sa.VARCHAR(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name='pk_nonebot_plugin_tetris_stats_bind'),
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_bind', schema=None) as batch_op:
-        batch_op.create_index('ix_nonebot_plugin_tetris_stats_bind_chat_platform', ['chat_platform'], unique=False)
-        batch_op.create_index('ix_nonebot_plugin_tetris_stats_bind_chat_account', ['chat_account'], unique=False)
-
-    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/b7fbdafc339a_add_user_unique_identifier_field_to_.py

@@ -1,107 +0,0 @@
-"""Add user_unique_identifier field to HistoricalData
-
-迁移 ID: b7fbdafc339a
-父迁移: 8a91210ce14d
-创建时间: 2024-05-07 16:55:29.527215
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import sqlalchemy as sa
-from alembic import op
-from nonebot.log import logger
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = 'b7fbdafc339a'
-down_revision: str | Sequence[str] | None = '8a91210ce14d'
-branch_labels: str | Sequence[str] | None = None
-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 (
-        BarColumn,
-        MofNCompleteColumn,
-        Progress,
-        TaskProgressColumn,
-        TextColumn,
-        TimeRemainingColumn,
-    )
-    from sqlalchemy import select
-    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(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_user_unique_identifier'),
-            ['user_unique_identifier'],
-            unique=False,
-        )
-    Base = automap_base()  # noqa: N806
-    connection = op.get_bind()
-    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(
-            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')
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_historicaldata', schema=None) as batch_op:
-        batch_op.drop_index(batch_op.f('ix_nonebot_plugin_tetris_stats_historicaldata_user_unique_identifier'))
-        batch_op.drop_column('user_unique_identifier')
-
-    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/b9d65badc713_del_old_tos_bind_data.py

@@ -1,42 +0,0 @@
-"""Del old TOS bind data
-
-迁移 ID: b9d65badc713
-父迁移: 6c3206f90cc3
-创建时间: 2023-12-30 00:27:40.991704
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from alembic import op
-from sqlalchemy.ext.automap import automap_base
-from sqlalchemy.orm import Session
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = 'b9d65badc713'
-down_revision: str | Sequence[str] | None = '6c3206f90cc3'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-
-    Base = automap_base()  # noqa: N806
-    connection = op.get_bind()
-    Base.prepare(autoload_with=connection)
-
-    Bind = Base.classes.nonebot_plugin_tetris_stats_bind  # noqa: N806
-    with Session(connection) as session:
-        session.query(Bind).filter(Bind.game_platform == 'TOS').delete()
-        session.commit()
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return

nonebot_plugin_tetris_stats/config/migrations/cfeab6961dce_extend_api_type_field_length.py

@@ -1,50 +0,0 @@
-"""Extend api_type field length
-
-迁移 ID: cfeab6961dce
-父迁移: f5b4a6d1325b
-创建时间: 2024-08-09 14:20:59.789030
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import sqlalchemy as sa
-from alembic import op
-from nonebot.log import logger
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = 'cfeab6961dce'
-down_revision: str | Sequence[str] | None = 'f5b4a6d1325b'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_tetriohistoricaldata', schema=None) as batch_op:
-        batch_op.alter_column(
-            'api_type', existing_type=sa.VARCHAR(length=16), type_=sa.String(length=32), existing_nullable=False
-        )
-
-    # ### end Alembic commands ###
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    # ### commands auto generated by Alembic - please adjust! ###
-    logger.warning('新数据可能不支持降级!')
-    logger.warning('请确认数据库内数据可以迁移到旧版本!')
-    input('如果确认可以迁移, 请按回车键继续!')
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_tetriohistoricaldata', schema=None) as batch_op:
-        batch_op.alter_column(
-            'api_type', existing_type=sa.String(length=32), type_=sa.VARCHAR(length=16), existing_nullable=False
-        )
-
-    # ### end Alembic commands ###

nonebot_plugin_tetris_stats/config/migrations/f5b4a6d1325b_tetr_io_new_season.py

@@ -1,91 +0,0 @@
-"""TETR.IO new season
-
-迁移 ID: f5b4a6d1325b
-父迁移: a1195e989cc6
-创建时间: 2024-08-01 20:44:48.644912
-
-"""
-
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import sqlalchemy as sa
-from alembic import op
-
-if TYPE_CHECKING:
-    from collections.abc import Sequence
-
-revision: str = 'f5b4a6d1325b'
-down_revision: str | Sequence[str] | None = 'a1195e989cc6'
-branch_labels: str | Sequence[str] | None = None
-depends_on: str | Sequence[str] | None = None
-
-
-def upgrade(name: str = '') -> None:
-    if name:
-        return
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_iorank_file_hash')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_iorank_rank')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_iorank_update_time')
-
-    op.drop_table('nonebot_plugin_tetris_stats_iorank')
-
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_tetriohistoricaldata', schema=None) as batch_op:
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_api_type')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_update_time')
-        batch_op.drop_index('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_user_unique_identifier')
-
-    op.drop_table('nonebot_plugin_tetris_stats_tetriohistoricaldata')
-
-    op.create_table(
-        'nonebot_plugin_tetris_stats_tetriohistoricaldata',
-        sa.Column('id', sa.Integer(), nullable=False),
-        sa.Column('user_unique_identifier', sa.String(length=24), nullable=False),
-        sa.Column('api_type', sa.String(length=16), nullable=False),
-        sa.Column('data', sa.JSON(), nullable=False),
-        sa.Column('update_time', sa.DateTime(), nullable=False),
-        sa.PrimaryKeyConstraint('id', name=op.f('pk_nonebot_plugin_tetris_stats_tetriohistoricaldata')),
-        info={'bind_key': 'nonebot_plugin_tetris_stats'},
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_tetriohistoricaldata', schema=None) as batch_op:
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_api_type'), ['api_type'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_update_time'), ['update_time'], unique=False
-        )
-        batch_op.create_index(
-            batch_op.f('ix_nonebot_plugin_tetris_stats_tetriohistoricaldata_user_unique_identifier'),
-            ['user_unique_identifier'],
-            unique=False,
-        )
-
-
-def downgrade(name: str = '') -> None:
-    if name:
-        return
-    op.create_table(
-        'nonebot_plugin_tetris_stats_iorank',
-        sa.Column('id', sa.INTEGER(), nullable=False),
-        sa.Column('rank', sa.VARCHAR(length=2), nullable=False),
-        sa.Column('tr_line', sa.FLOAT(), nullable=False),
-        sa.Column('player_count', sa.INTEGER(), nullable=False),
-        sa.Column('low_pps', sa.JSON(), nullable=False),
-        sa.Column('low_apm', sa.JSON(), nullable=False),
-        sa.Column('low_vs', sa.JSON(), nullable=False),
-        sa.Column('avg_pps', sa.FLOAT(), nullable=False),
-        sa.Column('avg_apm', sa.FLOAT(), nullable=False),
-        sa.Column('avg_vs', sa.FLOAT(), nullable=False),
-        sa.Column('high_pps', sa.JSON(), nullable=False),
-        sa.Column('high_apm', sa.JSON(), nullable=False),
-        sa.Column('high_vs', sa.JSON(), nullable=False),
-        sa.Column('update_time', sa.DATETIME(), nullable=False),
-        sa.Column('file_hash', sa.VARCHAR(length=128), nullable=True),
-        sa.PrimaryKeyConstraint('id', name='pk_nonebot_plugin_tetris_stats_iorank'),
-    )
-    with op.batch_alter_table('nonebot_plugin_tetris_stats_iorank', schema=None) as batch_op:
-        batch_op.create_index('ix_nonebot_plugin_tetris_stats_iorank_update_time', ['update_time'], unique=False)
-        batch_op.create_index('ix_nonebot_plugin_tetris_stats_iorank_rank', ['rank'], unique=False)
-        batch_op.create_index('ix_nonebot_plugin_tetris_stats_iorank_file_hash', ['file_hash'], unique=False)

nonebot_plugin_tetris_stats/db/__init__.py

@@ -1,135 +1,50 @@
-from asyncio import Lock
+from nonebot_plugin_orm import AsyncSession
-from collections.abc import AsyncGenerator
-from contextlib import asynccontextmanager
-from datetime import datetime, timezone
-from enum import Enum, auto
-from typing import TYPE_CHECKING, Literal, TypeVar, overload
-
-from nonebot.exception import FinishedException
-from nonebot.log import logger
-from nonebot_plugin_orm import AsyncSession, get_session
-from nonebot_plugin_user import User
 from sqlalchemy import select
 
-from ..utils.typing import AllCommandType, BaseCommandType, GameType, TETRIOCommandType
+from ..utils.typing import GameType
-from .models import Bind, TriggerHistoricalData
+from .models import Bind
-
-UTC = timezone.utc
-
-if TYPE_CHECKING:
-    from ..games.tetrio.api.models import TETRIOHistoricalData
-    from ..games.top.api.models import TOPHistoricalData
-    from ..games.tos.api.models import TOSHistoricalData
-
-
-class BindStatus(Enum):
-    SUCCESS = auto()
-    UPDATE = auto()
 
 
 async def query_bind_info(
     session: AsyncSession,
-    user: User,
+    chat_platform: str,
+    chat_account: str,
     game_platform: GameType,
 ) -> Bind | None:
     return (
-        await session.scalars(select(Bind).where(Bind.user_id == user.id).where(Bind.game_platform == game_platform))
+        await session.scalars(
+            select(Bind)
+            .where(Bind.chat_platform == chat_platform)
+            .where(Bind.chat_account == chat_account)
+            .where(Bind.game_platform == game_platform)
+        )
     ).one_or_none()
 
 
 async def create_or_update_bind(
     session: AsyncSession,
-    user: User,
+    chat_platform: str,
+    chat_account: str,
     game_platform: GameType,
     game_account: str,
-) -> BindStatus:
+) -> str:
     bind = await query_bind_info(
         session=session,
-        user=user,
+        chat_platform=chat_platform,
+        chat_account=chat_account,
         game_platform=game_platform,
     )
     if bind is None:
         bind = Bind(
-            user_id=user.id,
+            chat_platform=chat_platform,
+            chat_account=chat_account,
             game_platform=game_platform,
             game_account=game_account,
         )
         session.add(bind)
-        message = BindStatus.SUCCESS
+        message = '绑定成功'
     else:
         bind.game_account = game_account
-        message = BindStatus.UPDATE
+        message = '更新绑定成功'
     await session.commit()
     return message
-
-
-T = TypeVar('T', 'TETRIOHistoricalData', 'TOPHistoricalData', 'TOSHistoricalData')
-
-lock = Lock()
-
-
-async def anti_duplicate_add(model: T) -> None:
-    async with lock, get_session() as session:
-        result = (
-            await session.scalars(
-                select(cls := model.__class__)
-                .where(cls.update_time == model.update_time)
-                .where(cls.user_unique_identifier == model.user_unique_identifier)
-                .where(cls.api_type == model.api_type)
-            )
-        ).all()
-        if result:
-            for i in result:
-                if i.data == model.data:
-                    logger.debug('Anti duplicate successfully')
-                    return
-        session.add(model)
-        await session.commit()
-
-
-@asynccontextmanager
-@overload
-async def trigger(
-    session_persist_id: int,
-    game_platform: Literal['IO'],
-    command_type: TETRIOCommandType,
-    command_args: list[str],
-) -> AsyncGenerator:
-    yield
-
-
-@asynccontextmanager
-@overload
-async def trigger(
-    session_persist_id: int,
-    game_platform: GameType,
-    command_type: BaseCommandType,
-    command_args: list[str],
-) -> AsyncGenerator:
-    yield
-
-
-@asynccontextmanager
-async def trigger(
-    session_persist_id: int,
-    game_platform: GameType,
-    command_type: AllCommandType,
-    command_args: list[str],
-) -> AsyncGenerator:
-    trigger_time = datetime.now(UTC)
-    try:
-        yield
-    except FinishedException:
-        async with get_session() as session:
-            session.add(
-                TriggerHistoricalData(
-                    trigger_time=trigger_time,
-                    session_persist_id=session_persist_id,
-                    game_platform=game_platform,
-                    command_type=command_type,
-                    command_args=command_args,
-                    finish_time=datetime.now(UTC),
-                )
-            )
-            await session.commit()
-        raise

nonebot_plugin_tetris_stats/db/models.py

@@ -1,81 +1,33 @@
-from collections.abc import Callable, Sequence
 from datetime import datetime
-from typing import Any
 
-from nonebot.compat import PYDANTIC_V2, type_validate_json
+from nonebot.adapters import Message
 from nonebot_plugin_orm import Model
-from pydantic import BaseModel, ValidationError
+from sqlalchemy import JSON, DateTime, PickleType, String
-from sqlalchemy import JSON, DateTime, Dialect, String, TypeDecorator
 from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
-from typing_extensions import override
 
-from ..utils.typing import AllCommandType, GameType
+from ..game_data_processor import ProcessedData, User
-
+from ..utils.typing import CommandType, GameType
-
-class PydanticType(TypeDecorator):
-    impl = JSON
-
-    @override
-    def __init__(
-        self,
-        get_model: Sequence[Callable[[], Sequence[type[BaseModel]]]],
-        models: set[type[BaseModel]],
-        *args: Any,
-        **kwargs: Any,
-    ):
-        self.get_model = get_model
-        self._models = models
-        super().__init__(*args, **kwargs)
-
-    if PYDANTIC_V2:
-
-        @override
-        def process_bind_param(self, value: Any | None, dialect: Dialect) -> str:
-            # 将 Pydantic 模型实例转换为 JSON
-            if isinstance(value, tuple(self.models)):
-                return value.model_dump_json(by_alias=True)  # type: ignore[union-attr]
-            raise TypeError
-    else:
-
-        @override
-        def process_bind_param(self, value: Any | None, dialect: Dialect) -> str:
-            # 将 Pydantic 模型实例转换为 JSON
-            if isinstance(value, tuple(self.models)):
-                return value.json(by_alias=True)  # type: ignore[union-attr]
-            raise TypeError
-
-    @override
-    def process_result_value(self, value: Any | None, dialect: Dialect) -> BaseModel:
-        # 将 JSON 转换回 Pydantic 模型实例
-        if isinstance(value, str | bytes):
-            for i in self.models:
-                try:
-                    return type_validate_json(i, value)
-                except ValidationError:  # noqa: PERF203
-                    ...
-        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)
-    user_id: Mapped[int] = mapped_column(index=True)
+    chat_platform: Mapped[str] = mapped_column(String(32), index=True)
+    chat_account: Mapped[str] = mapped_column(index=True)
     game_platform: Mapped[GameType] = mapped_column(String(32))
     game_account: Mapped[str]
 
 
-class TriggerHistoricalData(MappedAsDataclass, Model):
+class HistoricalData(MappedAsDataclass, Model):
     id: Mapped[int] = mapped_column(init=False, primary_key=True)
     trigger_time: Mapped[datetime] = mapped_column(DateTime)
-    session_persist_id: Mapped[int]
+    bot_platform: Mapped[str | None] = mapped_column(String(32))
-    game_platform: Mapped[GameType] = mapped_column(String(32), index=True)
+    bot_account: Mapped[str | None]
-    command_type: Mapped[AllCommandType] = mapped_column(String(16), index=True)
+    source_type: Mapped[str | None] = mapped_column(String(32), index=True)
-    command_args: Mapped[list[str]] = mapped_column(JSON)
+    source_account: Mapped[str | None] = mapped_column(index=True)
-    finish_time: Mapped[datetime] = mapped_column(DateTime)
+    message: Mapped[Message | None] = mapped_column(PickleType)
+    game_platform: Mapped[GameType] = mapped_column(String(32), index=True, init=False)
+    command_type: Mapped[CommandType] = mapped_column(String(16), index=True, init=False)
+    command_args: Mapped[list[str]] = mapped_column(JSON, init=False)
+    game_user: Mapped[User] = mapped_column(PickleType, init=False)
+    processed_data: Mapped[ProcessedData] = mapped_column(PickleType, init=False)
+    finish_time: Mapped[datetime] = mapped_column(DateTime, init=False)

nonebot_plugin_tetris_stats/game_data_processor/__init__.py

@@ -0,0 +1,82 @@
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
+from datetime import UTC, datetime
+
+from ..utils.typing import CommandType, GameType
+
+
+@dataclass
+class User:
+    """游戏用户"""
+
+
+@dataclass
+class RawResponse:
+    """原始请求数据"""
+
+
+@dataclass
+class ProcessedData:
+    """处理/验证后的数据"""
+
+
+from ..utils.recorder import Recorder  # noqa: E402 避免循环导入
+
+
+class Processor(ABC):
+    event_id: int
+    command_type: CommandType
+    command_args: list[str]
+    user: User
+    raw_response: RawResponse
+    processed_data: ProcessedData
+
+    @abstractmethod
+    def __init__(
+        self,
+        event_id: int,
+        user: User,
+        command_args: list[str],
+    ) -> None:
+        self.event_id = event_id
+        self.user = user
+        self.command_args = command_args
+
+    @property
+    @abstractmethod
+    def game_platform(self) -> GameType:
+        """游戏平台"""
+        raise NotImplementedError
+
+    @abstractmethod
+    async def handle_bind(self, platform: str, account: str) -> str:
+        """处理绑定消息"""
+        raise NotImplementedError
+
+    @abstractmethod
+    async def handle_query(self) -> str:
+        """处理查询消息"""
+        raise NotImplementedError
+
+    @abstractmethod
+    async def generate_message(self) -> str:
+        """生成消息"""
+        raise NotImplementedError
+
+    def __del__(self) -> None:
+        finish_time = datetime.now(tz=UTC)
+        historical_data = Recorder.get_historical_data(self.event_id)
+        historical_data.game_platform = self.game_platform
+        historical_data.command_type = self.command_type
+        historical_data.command_args = self.command_args
+        historical_data.game_user = self.user
+        historical_data.processed_data = self.processed_data
+        historical_data.finish_time = finish_time
+        Recorder.update_historical_data(self.event_id, historical_data)
+
+
+from . import (  # noqa: F401, E402
+    io_data_processor,
+    top_data_processor,
+    tos_data_processor,
+)

nonebot_plugin_tetris_stats/game_data_processor/constant.py

@@ -0,0 +1,2 @@
+BIND_COMMAND: list[str] = ['绑定', 'bind']
+QUERY_COMMAND: list[str] = ['查', '查询', 'query', 'stats']

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/__init__.py

@@ -0,0 +1,187 @@
+from datetime import timedelta
+
+from arclet.alconna import Alconna, Arg, ArgFlag, Args, CommandMeta, Option
+from nonebot.adapters import Bot, Event
+from nonebot.matcher import Matcher
+from nonebot_plugin_alconna import AlcMatches, At, on_alconna
+from nonebot_plugin_orm import get_session
+from sqlalchemy import select
+
+from ...db import query_bind_info
+from ...utils.exception import MessageFormatError, NeedCatchError
+from ...utils.metrics import get_metrics
+from ...utils.platform import get_platform
+from ...utils.typing import Me
+from ..constant import BIND_COMMAND, QUERY_COMMAND
+from .constant import GAME_TYPE
+from .model import IORank
+from .processor import Processor, User, check_rank_data, identify_user_info
+from .typing import Rank
+
+alc = on_alconna(
+    Alconna(
+        'io',
+        Option(
+            BIND_COMMAND[0],
+            Args(
+                Arg(
+                    'account',
+                    identify_user_info,
+                    notice='IO 用户名 / ID',
+                    flags=[ArgFlag.HIDDEN],
+                )
+            ),
+            alias=BIND_COMMAND[1:],
+            compact=True,
+            dest='bind',
+            help_text='绑定 IO 账号',
+        ),
+        Option(
+            QUERY_COMMAND[0],
+            Args(
+                Arg(
+                    'target',
+                    At | Me,
+                    notice='@想要查询的人 | 自己',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+                Arg(
+                    'account',
+                    identify_user_info,
+                    notice='IO 用户名 / ID',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+            ),
+            alias=QUERY_COMMAND[1:],
+            compact=True,
+            dest='query',
+            help_text='查询 IO 游戏信息',
+        ),
+        Option(
+            'rank',
+            Args(Arg('rank', Rank, notice='IO 段位')),
+            alias={'Rank', 'RANK', '段位'},
+            compact=True,
+            dest='rank',
+            help_text='查询 IO 段位信息',
+        ),
+        meta=CommandMeta(
+            description='查询 TETR.IO 的信息',
+            example='io绑定scdhh\nio查我\niorankx',
+            compact=True,
+            fuzzy_match=True,
+        ),
+    ),
+    skip_for_unmatch=False,
+    auto_send_output=True,
+    aliases={'IO'},
+)
+
+alc.shortcut('fkosk', {'command': 'io查', 'args': ['我']})
+
+
+@alc.assign('bind')
+async def _(bot: Bot, event: Event, matcher: Matcher, account: User):
+    proc = Processor(
+        event_id=id(event),
+        user=account,
+        command_args=[],
+    )
+    try:
+        await matcher.finish(await proc.handle_bind(platform=get_platform(bot), account=event.get_user_id()))
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.assign('query')
+async def _(bot: Bot, event: Event, matcher: Matcher, target: At | Me):
+    async with get_session() as session:
+        bind = await query_bind_info(
+            session=session,
+            chat_platform=get_platform(bot),
+            chat_account=(target.target if isinstance(target, At) else event.get_user_id()),
+            game_platform=GAME_TYPE,
+        )
+    if bind is None:
+        await matcher.finish('未查询到绑定信息')
+    message = '* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n'
+    proc = Processor(
+        event_id=id(event),
+        user=User(ID=bind.game_account),
+        command_args=[],
+    )
+    try:
+        await matcher.finish(message + await proc.handle_query())
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.assign('query')
+async def _(event: Event, matcher: Matcher, account: User):
+    proc = Processor(
+        event_id=id(event),
+        user=account,
+        command_args=[],
+    )
+    try:
+        await matcher.finish(await proc.handle_query())
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.assign('rank')
+async def _(event: Event, matcher: Matcher, rank: Rank):
+    if rank == 'z':
+        await matcher.finish('暂不支持查询未知段位')
+    try:
+        await check_rank_data()
+    except NeedCatchError as e:
+        await matcher.finish(str(f'段位信息获取失败\n{e}'))
+    async with get_session() as session:
+        data = (
+            await session.scalars(select(IORank).where(IORank.rank == rank).order_by(IORank.id.desc()).limit(5))
+        ).all()
+    latest_data = data[0]
+    message = f'{rank.upper()} 段 分数线 {latest_data.tr_line:.2f} TR, {latest_data.player_count} 名玩家\n'
+    if len(data) > 1:
+        message += f'对比 {(latest_data.create_time-data[-1].create_time).total_seconds()/3600:.2f} 小时前趋势: {f"↑{difference:.2f}" if (difference:=latest_data.tr_line-data[-1].tr_line) > 0 else f"↓{-difference:.2f}" if difference < 0 else "→"}'
+    else:
+        message += '暂无对比数据'
+    avg = get_metrics(pps=latest_data.avg_pps, apm=latest_data.avg_apm, vs=latest_data.avg_vs)
+    low_pps = get_metrics(pps=latest_data.low_pps[1])
+    low_vs = get_metrics(vs=latest_data.low_vs[1])
+    max_pps = get_metrics(pps=latest_data.high_pps[1])
+    max_vs = get_metrics(vs=latest_data.high_vs[1])
+    message += (
+        '\n'
+        '平均数据:\n'
+        f"L'PM: {avg.lpm} ( {avg.pps} pps )\n"
+        f'APM: {avg.apm} ( x{avg.apl} )\n'
+        f'ADPM: {avg.adpm} ( x{avg.adpl} ) ( {avg.vs}vs )\n'
+        '\n'
+        '最低数据:\n'
+        f"L'PM: {low_pps.lpm} ( {low_pps.pps} pps ) By: {latest_data.low_pps[0]['name'].upper()}\n"
+        f'APM: {latest_data.low_apm[1]} By: {latest_data.low_apm[0]["name"].upper()}\n'
+        f'ADPM: {low_vs.adpm} ( {low_vs.vs}vs ) By: {latest_data.low_vs[0]["name"].upper()}\n'
+        '\n'
+        '最高数据:\n'
+        f"L'PM: {max_pps.lpm} ( {max_pps.pps} pps ) By: {latest_data.high_pps[0]['name'].upper()}\n"
+        f'APM: {latest_data.high_apm[1]} By: {latest_data.high_apm[0]["name"].upper()}\n'
+        f'ADPM: {max_vs.adpm} ( {max_vs.vs}vs ) By: {latest_data.high_vs[0]["name"].upper()}\n'
+        '\n'
+        f'数据更新时间: {(latest_data.create_time+timedelta(hours=8)).strftime("%Y-%m-%d %H:%M:%S")}'
+    )
+    await matcher.finish(message)
+
+
+@alc.handle()
+async def _(matcher: Matcher, account: MessageFormatError):
+    await matcher.finish(str(account))
+
+
+@alc.handle()
+async def _(matcher: Matcher, matches: AlcMatches):
+    if matches.head_matched:
+        await matcher.finish(
+            f'{matches.error_info!r}\n' if matches.error_info is not None else '' + '输入"io --help"查看帮助'
+        )

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/constant.py

@@ -0,0 +1,24 @@
+from ...utils.typing import GameType
+from .typing import Rank
+
+GAME_TYPE: GameType = 'IO'
+BASE_URL = 'https://ch.tetr.io/api/'
+RANK_PERCENTILE: dict[Rank, float] = {
+    'x': 1,
+    'u': 5,
+    'ss': 11,
+    's+': 17,
+    's': 23,
+    's-': 30,
+    'a+': 38,
+    'a': 46,
+    'a-': 54,
+    'b+': 62,
+    'b': 70,
+    'b-': 78,
+    'c+': 84,
+    'c': 90,
+    'c-': 95,
+    'd+': 97.5,
+    'd': 100,
+}

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/model.py

@@ -0,0 +1,29 @@
+from datetime import UTC, datetime
+
+from nonebot_plugin_orm import Model
+from sqlalchemy import JSON, DateTime, String
+from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
+
+from .typing import Rank
+
+
+class IORank(MappedAsDataclass, Model):
+    id: Mapped[int] = mapped_column(init=False, primary_key=True)
+    rank: Mapped[Rank] = mapped_column(String(2), index=True)
+    tr_line: Mapped[float]
+    player_count: Mapped[int]
+    low_pps: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    low_apm: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    low_vs: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    avg_pps: Mapped[float]
+    avg_apm: Mapped[float]
+    avg_vs: Mapped[float]
+    high_pps: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    high_apm: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    high_vs: Mapped[tuple[dict[str, str], float]] = mapped_column(JSON)
+    create_time: Mapped[datetime] = mapped_column(
+        DateTime,
+        default=lambda: datetime.now(tz=UTC),
+        index=True,
+        init=False,
+    )

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/processor.py

@@ -0,0 +1,254 @@
+from collections import defaultdict
+from collections.abc import Callable
+from dataclasses import asdict, dataclass
+from datetime import UTC, datetime, timedelta
+from math import floor
+from re import match
+from statistics import mean
+
+from nonebot import get_driver
+from nonebot_plugin_apscheduler import scheduler  # type: ignore[import-untyped]
+from nonebot_plugin_orm import get_session
+from pydantic import parse_raw_as
+from sqlalchemy import select
+
+from ...db import create_or_update_bind
+from ...utils.exception import MessageFormatError, RequestError, WhatTheFuckError
+from ...utils.request import Request, splice_url
+from ...utils.typing import GameType
+from .. import ProcessedData as ProcessedDataMeta
+from .. import Processor as ProcessorMeta
+from .. import RawResponse as RawResponseMeta
+from .. import User as UserMeta
+from .constant import BASE_URL, GAME_TYPE, RANK_PERCENTILE
+from .model import IORank
+from .schemas.league_all import FailedModel as LeagueAllFailed
+from .schemas.league_all import LeagueAll
+from .schemas.league_all import ValidUser as LeagueAllUser
+from .schemas.user_info import FailedModel as InfoFailed
+from .schemas.user_info import (
+    NeverPlayedLeague,
+    NeverRatedLeague,
+    UserInfo,
+)
+from .schemas.user_info import SuccessModel as InfoSuccess
+from .schemas.user_records import FailedModel as RecordsFailed
+from .schemas.user_records import SoloRecord, UserRecords
+from .schemas.user_records import SuccessModel as RecordsSuccess
+from .typing import Rank
+
+driver = get_driver()
+
+
+@dataclass
+class User(UserMeta):
+    ID: str | None = None
+    name: str | None = None
+
+
+@dataclass
+class RawResponse(RawResponseMeta):
+    user_info: bytes | None = None
+    user_records: bytes | None = None
+
+
+@dataclass
+class ProcessedData(ProcessedDataMeta):
+    user_info: InfoSuccess | None = None
+    user_records: RecordsSuccess | None = None
+
+
+def identify_user_info(info: str) -> User | MessageFormatError:
+    if match(r'^[a-f0-9]{24}$', info):
+        return User(ID=info)
+    if match(r'^[a-zA-Z0-9_-]{3,16}$', info):
+        return User(name=info.lower())
+    return MessageFormatError('用户名/ID不合法')
+
+
+class Processor(ProcessorMeta):
+    user: User
+    raw_response: RawResponse
+    processed_data: ProcessedData
+
+    def __init__(self, event_id: int, user: User, command_args: list[str]) -> None:
+        super().__init__(event_id, user, command_args)
+        self.raw_response = RawResponse()
+        self.processed_data = ProcessedData()
+
+    @property
+    def game_platform(self) -> GameType:
+        return GAME_TYPE
+
+    async def handle_bind(self, platform: str, account: str) -> str:
+        """处理绑定消息"""
+        self.command_type = 'bind'
+        await self.get_user()
+        if self.user.ID is None:
+            raise  # FIXME: 不知道怎么才能把这类型给变过来了
+        async with get_session() as session:
+            return await create_or_update_bind(
+                session=session,
+                chat_platform=platform,
+                chat_account=account,
+                game_platform=GAME_TYPE,
+                game_account=self.user.ID,
+            )
+
+    async def handle_query(self) -> str:
+        """处理查询消息"""
+        self.command_type = 'query'
+        await self.get_user()
+        return await self.generate_message()
+
+    async def get_user(self) -> None:
+        """
+        用于获取 UserName 和 UserID 的函数
+        """
+        if self.user.name is None:
+            self.user.name = (await self.get_user_info()).data.user.username
+        if self.user.ID is None:
+            self.user.ID = (await self.get_user_info()).data.user.id
+
+    async def get_user_info(self) -> InfoSuccess:
+        """获取用户数据"""
+        if self.processed_data.user_info is None:
+            self.raw_response.user_info = await Request.request(
+                splice_url([BASE_URL, 'users/', f'{self.user.ID or self.user.name}'])
+            )
+            user_info: UserInfo = parse_raw_as(UserInfo, self.raw_response.user_info)  # type: ignore[arg-type]
+            if isinstance(user_info, InfoFailed):
+                raise RequestError(f'用户信息请求错误:\n{user_info.error}')
+            self.processed_data.user_info = user_info
+        return self.processed_data.user_info
+
+    async def get_user_records(self) -> RecordsSuccess:
+        """获取Solo数据"""
+        if self.processed_data.user_records is None:
+            self.raw_response.user_records = await Request.request(
+                splice_url(
+                    [
+                        BASE_URL,
+                        'users/',
+                        f'{self.user.ID or self.user.name}/',
+                        'records',
+                    ]
+                )
+            )
+            user_records: UserRecords = parse_raw_as(
+                UserRecords,  # type: ignore[arg-type]
+                self.raw_response.user_records,
+            )
+            if isinstance(user_records, RecordsFailed):
+                raise RequestError(f'用户Solo数据请求错误:\n{user_records.error}')
+            self.processed_data.user_records = user_records
+        return self.processed_data.user_records
+
+    async def generate_message(self) -> str:
+        """生成消息"""
+        user_info = await self.get_user_info()
+        user_name = user_info.data.user.username.upper()
+        league = user_info.data.user.league
+        ret_message = ''
+        if isinstance(league, NeverPlayedLeague):
+            ret_message += f'用户 {user_name} 没有排位统计数据'
+        else:
+            if isinstance(league, NeverRatedLeague):
+                ret_message += f'用户 {user_name} 暂未完成定级赛, 最近十场的数据:'
+            elif league.rank == 'z':
+                ret_message += f'用户 {user_name} 暂无段位, {round(league.rating,2)} TR'
+            else:
+                ret_message += (
+                    f'{league.rank.upper()} 段用户 {user_name} {round(league.rating,2)} TR (#{league.standing})'
+                )
+                ret_message += f', 段位分 {round(league.glicko,2)}±{round(league.rd,2)}, 最近十场的数据:'
+            lpm = league.pps * 24
+            ret_message += f"\nL'PM: {round(lpm, 2)} ( {league.pps} pps )"
+            ret_message += f'\nAPM: {league.apm} ( x{round(league.apm/(league.pps*24),2)} )'
+            if league.vs is not None:
+                adpm = league.vs * 0.6
+                ret_message += f'\nADPM: {round(adpm,2)} ( x{round(adpm/lpm,2)} ) ( {league.vs}vs )'
+        user_records = await self.get_user_records()
+        sprint = user_records.data.records.sprint
+        if sprint.record is not None:
+            if not isinstance(sprint.record, SoloRecord):
+                raise WhatTheFuckError('40L记录不是单人记录')
+            ret_message += f'\n40L: {round(sprint.record.endcontext.final_time/1000,2)}s'
+            ret_message += f' ( #{sprint.rank} )' if sprint.rank is not None else ''
+        blitz = user_records.data.records.blitz
+        if blitz.record is not None:
+            if not isinstance(blitz.record, SoloRecord):
+                raise WhatTheFuckError('Blitz记录不是单人记录')
+            ret_message += f'\nBlitz: {blitz.record.endcontext.score}'
+            ret_message += f' ( #{blitz.rank} )' if blitz.rank is not None else ''
+        return ret_message
+
+
+@scheduler.scheduled_job('cron', hour='0,6,12,18', minute=0)
+async def get_io_rank_data() -> None:
+    league_all: LeagueAll = parse_raw_as(
+        LeagueAll,  # type: ignore[arg-type]
+        await Request.request(splice_url([BASE_URL, 'users/lists/league/all'])),
+    )
+    if isinstance(league_all, LeagueAllFailed):
+        raise RequestError(f'用户Solo数据请求错误:\n{league_all.error}')
+
+    def pps(user: LeagueAllUser) -> float:
+        return user.league.pps
+
+    def apm(user: LeagueAllUser) -> float:
+        return user.league.apm
+
+    def vs(user: LeagueAllUser) -> float:
+        return user.league.vs
+
+    def _min(users: list[LeagueAllUser], field: Callable[[LeagueAllUser], float]) -> LeagueAllUser:
+        return min(users, key=field)
+
+    def _max(users: list[LeagueAllUser], field: Callable[[LeagueAllUser], float]) -> LeagueAllUser:
+        return max(users, key=field)
+
+    def build_extremes_data(
+        users: list[LeagueAllUser],
+        field: Callable[[LeagueAllUser], float],
+        sort: Callable[[list[LeagueAllUser], Callable[[LeagueAllUser], float]], LeagueAllUser],
+    ) -> tuple[dict[str, str], float]:
+        user = sort(users, field)
+        return asdict(User(ID=user.id, name=user.username)), field(user)
+
+    users = [i for i in league_all.data.users if isinstance(i, LeagueAllUser)]
+    rank_to_users: defaultdict[Rank, list[LeagueAllUser]] = defaultdict(list)
+    for i in users:
+        rank_to_users[i.league.rank].append(i)
+    rank_info: list[IORank] = []
+    for rank, percentile in RANK_PERCENTILE.items():
+        offset = floor((percentile / 100) * len(users)) - 1
+        tr_line = users[offset].league.rating
+        rank_users = rank_to_users[rank]
+        rank_info.append(
+            IORank(
+                rank=rank,
+                tr_line=tr_line,
+                player_count=len(rank_users),
+                low_pps=(build_extremes_data(rank_users, pps, _min)),
+                low_apm=(build_extremes_data(rank_users, apm, _min)),
+                low_vs=(build_extremes_data(rank_users, vs, _min)),
+                avg_pps=mean({i.league.pps for i in rank_users}),
+                avg_apm=mean({i.league.apm for i in rank_users}),
+                avg_vs=mean({i.league.vs for i in rank_users}),
+                high_pps=(build_extremes_data(rank_users, pps, _max)),
+                high_apm=(build_extremes_data(rank_users, apm, _max)),
+                high_vs=(build_extremes_data(rank_users, vs, _max)),
+            )
+        )
+    async with get_session() as session:
+        session.add_all(rank_info)
+        await session.commit()
+
+
+@driver.on_startup
+async def check_rank_data() -> None:
+    async with get_session() as session:
+        latest_time = await session.scalar(select(IORank.create_time).order_by(IORank.id.desc()).limit(1))
+        if latest_time is None or datetime.now(tz=UTC) - latest_time.replace(tzinfo=UTC) > timedelta(hours=6):
+            await get_io_rank_data()

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/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/game_data_processor/io_data_processor/schemas/league_all.py

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

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/schemas/user_info.py

@@ -0,0 +1,125 @@
+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 SuccessModel(BaseSuccessModel):
+    class Data(BaseModel):
+        class User(BaseModel):
+            class Badge(BaseModel):
+                id: str
+                label: str
+                ts: datetime | None
+
+            class NeverPlayedLeague(BaseModel):
+                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
+                pps: None
+                vs: None
+                decaying: bool
+
+            class NeverRatedLeague(BaseModel):
+                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
+                decaying: bool
+
+            class RatedLeague(BaseModel):
+                gamesplayed: int
+                gameswon: int
+                rating: float
+                rank: Rank
+                bestrank: Rank
+                standing: int
+                standing_local: int
+                next_rank: Rank | None
+                prev_rank: Rank | None
+                next_at: int
+                prev_at: int
+                percentile: float
+                percentile_rank: str
+                glicko: float
+                rd: float
+                apm: float
+                pps: float
+                vs: float | None
+                decaying: bool
+
+            class Connections(BaseModel):
+                class Discord(BaseModel):
+                    id: str
+                    username: str
+
+                discord: Discord | None
+
+            class Distinguishment(BaseModel):
+                type: str  # noqa: A003
+
+            id: str = Field(..., alias='_id')
+            username: str
+            role: Literal['anon', 'user', 'bot', 'halfmod', 'mod', 'admin', 'sysop', 'banned']
+            ts: datetime | None
+            botmaster: str | None
+            badges: list[Badge]
+            xp: float
+            gamesplayed: int
+            gameswon: int
+            gametime: float
+            country: str | None
+            badstanding: bool | None
+            supporter: bool | None  # osk说是必有, 但实际上不是 fk osk
+            supporter_tier: int
+            verified: bool
+            league: NeverPlayedLeague | NeverRatedLeague | RatedLeague
+            avatar_revision: int | 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
+            """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
+            connections: Connections
+            friend_count: int
+            distinguishment: Distinguishment | None
+
+        user: User
+
+    data: Data
+
+
+NeverPlayedLeague = SuccessModel.Data.User.NeverPlayedLeague
+NeverRatedLeague = SuccessModel.Data.User.NeverRatedLeague
+RatedLeague = SuccessModel.Data.User.RatedLeague
+UserInfo = SuccessModel | FailedModel

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/schemas/user_records.py

@@ -0,0 +1,125 @@
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+from .base import FailedModel
+from .base import SuccessModel as BaseSuccessModel
+
+
+class EndContext(BaseModel):
+    class Time(BaseModel):
+        start: int
+        zero: bool
+        locked: bool
+        prev: int
+        frameoffset: int
+
+    class Clears(BaseModel):
+        singles: int
+        doubles: int
+        triples: int
+        quads: int
+        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
+        cleared: int
+
+    class Finesse(BaseModel):
+        combo: int
+        faults: int
+        perfectpieces: int
+
+    seed: int
+    lines: int
+    level_lines: int
+    level_lines_needed: int
+    inputs: int
+    holds: int
+    time: Time
+    score: int
+    zenlevel: int
+    zenprogress: int
+    level: int
+    combo: int
+    currentcombopower: int  # WTF
+    topcombo: int
+    btb: int
+    topbtb: int
+    currentbtbchainpower: int | None  # WTF * 2 40l 里有 但是 blitz 没有
+    tspins: int
+    piecesplaced: int
+    clears: Clears
+    garbage: Garbage
+    kills: int
+    finesse: Finesse
+    final_time: float = Field(..., alias='finalTime')
+    gametype: str
+
+
+class BaseModeRecord(BaseModel):
+    class SoloRecord(BaseModel):
+        class User(BaseModel):
+            id: str = Field(..., alias='_id')
+            username: str
+
+        id: str = Field(..., alias='_id')
+        stream: str
+        replayid: str
+        user: User
+        ts: datetime
+        ismulti: bool | None
+        endcontext: EndContext
+
+    class MultiRecord(BaseModel):
+        class User(BaseModel):
+            id: str = Field(..., alias='_id')
+            username: str
+
+        id: str = Field(..., alias='_id')
+        stream: str
+        replayid: str
+        user: User
+        ts: datetime
+        ismulti: bool | None
+        endcontext: list[EndContext]
+
+    record: SoloRecord | MultiRecord | None
+    rank: int | None
+
+
+class SuccessModel(BaseSuccessModel):
+    class Data(BaseModel):
+        class Records(BaseModel):
+            class Sprint(BaseModeRecord):
+                ...
+
+            class Blitz(BaseModeRecord):
+                ...
+
+            sprint: Sprint = Field(..., alias='40l')
+            blitz: Blitz
+
+        class Zen(BaseModel):
+            level: int
+            score: int
+
+        records: Records
+        zen: Zen
+
+    data: Data
+
+
+SoloRecord = BaseModeRecord.SoloRecord
+MultiRecord = BaseModeRecord.MultiRecord
+UserRecords = SuccessModel | FailedModel

nonebot_plugin_tetris_stats/game_data_processor/io_data_processor/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',  # 未定级
+]

nonebot_plugin_tetris_stats/game_data_processor/top_data_processor/__init__.py

@@ -0,0 +1,126 @@
+from arclet.alconna import Alconna, Arg, ArgFlag, Args, CommandMeta, Option
+from nonebot.adapters import Bot, Event
+from nonebot.matcher import Matcher
+from nonebot_plugin_alconna import AlcMatches, At, on_alconna
+from nonebot_plugin_orm import get_session
+
+from ...db import query_bind_info
+from ...utils.exception import MessageFormatError, NeedCatchError
+from ...utils.platform import get_platform
+from ...utils.typing import Me
+from ..constant import BIND_COMMAND, QUERY_COMMAND
+from .constant import GAME_TYPE
+from .processor import Processor, User, identify_user_info
+
+alc = on_alconna(
+    Alconna(
+        'top',
+        Option(
+            BIND_COMMAND[0],
+            Args(
+                Arg(
+                    'account',
+                    identify_user_info,
+                    notice='TOP 用户名',
+                    flags=[ArgFlag.HIDDEN],
+                )
+            ),
+            alias=BIND_COMMAND[1:],
+            compact=True,
+            dest='bind',
+            help_text='绑定 TOP 账号',
+        ),
+        Option(
+            QUERY_COMMAND[0],
+            Args(
+                Arg(
+                    'target',
+                    At | Me,
+                    notice='@想要查询的人 | 自己',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+                Arg(
+                    'account',
+                    identify_user_info | Me | At,
+                    notice='TOP 用户名',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+            ),
+            alias=QUERY_COMMAND[1:],
+            compact=True,
+            dest='query',
+            help_text='查询 TOP 游戏信息',
+        ),
+        meta=CommandMeta(
+            description='查询 TetrisOnline波兰服 的信息',
+            example='top绑定scdhh\ntop查我',
+            compact=True,
+            fuzzy_match=True,
+        ),
+    ),
+    skip_for_unmatch=False,
+    auto_send_output=True,
+    aliases={'TOP'},
+)
+
+
+@alc.assign('bind')
+async def _(bot: Bot, event: Event, matcher: Matcher, account: User):
+    proc = Processor(
+        event_id=id(event),
+        user=account,
+        command_args=[],
+    )
+    try:
+        await matcher.finish(await proc.handle_bind(platform=get_platform(bot), account=event.get_user_id()))
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.assign('query')
+async def _(bot: Bot, event: Event, matcher: Matcher, target: At | Me):
+    async with get_session() as session:
+        bind = await query_bind_info(
+            session=session,
+            chat_platform=get_platform(bot),
+            chat_account=(target.target if isinstance(target, At) else event.get_user_id()),
+            game_platform=GAME_TYPE,
+        )
+    if bind is None:
+        await matcher.finish('未查询到绑定信息')
+    message = '* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n'
+    proc = Processor(
+        event_id=id(event),
+        user=User(name=bind.game_account),
+        command_args=[],
+    )
+    try:
+        await matcher.finish(message + await proc.handle_query())
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.assign('query')
+async def _(event: Event, matcher: Matcher, account: User):
+    proc = Processor(
+        event_id=id(event),
+        user=account,
+        command_args=[],
+    )
+    try:
+        await matcher.finish(await proc.handle_query())
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.handle()
+async def _(matcher: Matcher, account: MessageFormatError):
+    await matcher.finish(str(account))
+
+
+@alc.handle()
+async def _(matcher: Matcher, matches: AlcMatches):
+    if matches.head_matched:
+        await matcher.finish(
+            f'{matches.error_info!r}\n' if matches.error_info is not None else '' + '输入"top --help"查看帮助'
+        )

nonebot_plugin_tetris_stats/game_data_processor/top_data_processor/constant.py

@@ -0,0 +1,4 @@
+from ...utils.typing import GameType
+
+GAME_TYPE: GameType = 'TOP'
+BASE_URL = 'http://tetrisonline.pl/top/'

nonebot_plugin_tetris_stats/game_data_processor/top_data_processor/processor.py

@@ -0,0 +1,142 @@
+from contextlib import suppress
+from dataclasses import dataclass
+from io import StringIO
+from re import match
+from typing import NoReturn
+from urllib.parse import urlencode
+
+from lxml import etree
+from nonebot_plugin_orm import get_session
+from pandas import read_html
+
+from ...db import create_or_update_bind
+from ...utils.exception import MessageFormatError, RequestError
+from ...utils.request import Request, splice_url
+from ...utils.typing import GameType
+from .. import ProcessedData as ProcessedDataMeta
+from .. import Processor as ProcessorMeta
+from .. import RawResponse as RawResponseMeta
+from .. import User as UserMeta
+from .constant import BASE_URL, GAME_TYPE
+
+
+@dataclass
+class User(UserMeta):
+    name: str
+
+
+@dataclass
+class RawResponse(RawResponseMeta):
+    user_profile: bytes | None = None
+
+
+@dataclass
+class ProcessedData(ProcessedDataMeta):
+    user_profile: str | None = None
+
+
+@dataclass
+class Data:
+    lpm: float
+    apm: float
+
+
+@dataclass
+class GameData:
+    day: Data | None
+    total: Data | None
+
+
+def identify_user_info(info: str) -> User | MessageFormatError:
+    if match(r'^[a-zA-Z0-9_]{1,16}$', info):
+        return User(name=info)
+    return MessageFormatError('用户名不合法')
+
+
+class Processor(ProcessorMeta):
+    user: User
+    raw_response: RawResponse
+    processed_data: ProcessedData
+
+    def __init__(self, event_id: int, user: User, command_args: list[str]) -> None:
+        super().__init__(event_id, user, command_args)
+        self.raw_response = RawResponse()
+        self.processed_data = ProcessedData()
+
+    @property
+    def game_platform(self) -> GameType:
+        return GAME_TYPE
+
+    async def handle_bind(self, platform: str, account: str) -> str:
+        """处理绑定消息"""
+        self.command_type = 'bind'
+        await self.check_user()
+        async with get_session() as session:
+            return await create_or_update_bind(
+                session=session,
+                chat_platform=platform,
+                chat_account=account,
+                game_platform=GAME_TYPE,
+                game_account=self.user.name,
+            )
+
+    async def handle_query(self) -> str:
+        """处理查询消息"""
+        self.command_type = 'query'
+        await self.check_user()
+        return await self.generate_message()
+
+    async def get_user_profile(self) -> str:
+        """获取用户信息"""
+        if self.processed_data.user_profile is None:
+            url = splice_url([BASE_URL, 'profile.php', f'?{urlencode({"user":self.user.name})}'])
+            self.raw_response.user_profile = await Request.request(url, is_json=False)
+            self.processed_data.user_profile = self.raw_response.user_profile.decode()
+        return self.processed_data.user_profile
+
+    async def check_user(self) -> None | NoReturn:
+        if 'user not found!' in await self.get_user_profile():
+            raise RequestError('用户不存在!')
+        return None
+
+    async def get_user_name(self) -> str:
+        """获取用户名"""
+        data = etree.HTML(await self.get_user_profile()).xpath('//div[@class="mycontent"]/h1/text()')
+        return data[0].replace("'s profile", '')
+
+    async def get_game_data(self) -> GameData:
+        """获取游戏统计数据"""
+        html = etree.HTML(await self.get_user_profile())
+        day = None
+        with suppress(ValueError):
+            day = Data(
+                lpm=float(str(html.xpath('//div[@class="mycontent"]/text()[3]')[0]).replace('lpm:', '').strip()),
+                apm=float(str(html.xpath('//div[@class="mycontent"]/text()[4]')[0]).replace('apm:', '').strip()),
+            )
+        table = StringIO(
+            etree.tostring(
+                html.xpath('//div[@class="mycontent"]/table[@class="mytable"]')[0],
+                encoding='utf-8',
+            ).decode()
+        )
+        dataframe = read_html(table, encoding='utf-8', header=0)[0]
+        total = Data(lpm=dataframe['lpm'].mean(), apm=dataframe['apm'].mean()) if len(dataframe) != 0 else None
+        return GameData(day=day, total=total)
+
+    async def generate_message(self) -> str:
+        """生成消息"""
+        game_data = await self.get_game_data()
+        message = ''
+        if game_data.day is not None:
+            message += f'用户 {self.user.name} 24小时内统计数据为: '
+            message += f"\nL'PM: {round(game_data.day.lpm,2)} ( {round(game_data.day.lpm/24,2)} pps )"
+            message += f'\nAPM: {round(game_data.day.apm,2)} ( x{round(game_data.day.apm/game_data.day.lpm,2)} )'
+        else:
+            message += f'用户 {self.user.name} 暂无24小时内统计数据'
+        if game_data.total is not None:
+            message += '\n历史统计数据为: '
+            message += f"\nL'PM: {round(game_data.total.lpm,2)} ( {round(game_data.total.lpm/24,2)} pps )"
+            message += f'\nAPM: {round(game_data.total.apm,2)} ( x{round(game_data.total.apm/game_data.total.lpm,2)} )'
+        else:
+            message += '\n暂无历史统计数据'
+        return message

nonebot_plugin_tetris_stats/game_data_processor/tos_data_processor/__init__.py

@@ -0,0 +1,151 @@
+from arclet.alconna import Alconna, Arg, ArgFlag, Args, CommandMeta, Option
+from nonebot.adapters import Bot, Event
+from nonebot.matcher import Matcher
+from nonebot_plugin_alconna import AlcMatches, At, on_alconna
+from nonebot_plugin_orm import get_session
+
+from ...db import query_bind_info
+from ...utils.exception import MessageFormatError, NeedCatchError
+from ...utils.platform import get_platform
+from ...utils.typing import Me
+from ..constant import BIND_COMMAND, QUERY_COMMAND
+from .constant import GAME_TYPE
+from .processor import Processor, User, identify_user_info
+
+alc = on_alconna(
+    Alconna(
+        '茶服',
+        Option(
+            BIND_COMMAND[0],
+            Args(
+                Arg(
+                    'account',
+                    identify_user_info,
+                    notice='茶服 用户名 / TeaID',
+                    flags=[ArgFlag.HIDDEN],
+                )
+            ),
+            alias=BIND_COMMAND[1:],
+            compact=True,
+            dest='bind',
+            help_text='绑定 茶服 账号',
+        ),
+        Option(
+            QUERY_COMMAND[0],
+            Args(
+                Arg(
+                    'target',
+                    At | Me,
+                    notice='@想要查询的人 | 自己',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+                Arg(
+                    'account',
+                    identify_user_info,
+                    notice='茶服 用户名 / TeaID',
+                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
+                ),
+                # 如果放在一个 Union Args 里, 验证顺序不能保证, 可能出错
+            ),
+            alias=QUERY_COMMAND[1:],
+            compact=True,
+            dest='query',
+            help_text='查询 茶服 游戏信息',
+        ),
+        meta=CommandMeta(
+            description='查询 TetrisOnline茶服 的信息',
+            example='茶服查我',
+            compact=True,
+            fuzzy_match=True,
+        ),
+    ),
+    skip_for_unmatch=False,
+    auto_send_output=True,
+    aliases={'tos', 'TOS'},
+)
+
+try:
+    from nonebot.adapters.onebot.v11 import GROUP, MessageEvent
+    from nonebot.adapters.onebot.v11 import Bot as OB11Bot
+
+    @alc.assign('bind')
+    async def _(event: MessageEvent, matcher: Matcher):
+        await matcher.finish('QQ 平台无需绑定')
+
+    @alc.assign('query')
+    async def _(bot: OB11Bot, event: MessageEvent, matcher: Matcher, target: At | Me):
+        if event.is_tome() and await GROUP(bot, event):
+            await matcher.finish('不能查询bot的信息')
+        proc = Processor(
+            event_id=id(event),
+            user=User(teaid=target.target if isinstance(target, At) else event.get_user_id()),
+            command_args=[],
+        )
+        try:
+            await matcher.finish(await proc.handle_query())
+        except NeedCatchError as e:
+            await matcher.finish(str(e))
+except ImportError:
+    pass
+
+
+@alc.assign('bind')
+async def _(bot: Bot, event: Event, matcher: Matcher, account: User):
+    proc = Processor(
+        event_id=id(event),
+        user=account,
+        command_args=[],
+    )
+    try:
+        await matcher.finish(await proc.handle_bind(platform=get_platform(bot), account=event.get_user_id()))
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.assign('query')
+async def _(bot: Bot, event: Event, matcher: Matcher, target: At | Me):
+    async with get_session() as session:
+        bind = await query_bind_info(
+            session=session,
+            chat_platform=get_platform(bot),
+            chat_account=(target.target if isinstance(target, At) else event.get_user_id()),
+            game_platform=GAME_TYPE,
+        )
+    if bind is None:
+        await matcher.finish('未查询到绑定信息')
+    message = '* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n'
+    proc = Processor(
+        event_id=id(event),
+        user=User(name=bind.game_account),
+        command_args=[],
+    )
+    try:
+        await matcher.finish(message + await proc.handle_query())
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.assign('query')
+async def _(event: Event, matcher: Matcher, account: User):
+    proc = Processor(
+        event_id=id(event),
+        user=account,
+        command_args=[],
+    )
+    try:
+        await matcher.finish(await proc.handle_query())
+    except NeedCatchError as e:
+        await matcher.finish(str(e))
+
+
+@alc.handle()
+async def _(matcher: Matcher, account: MessageFormatError):
+    await matcher.finish(str(account))
+
+
+@alc.handle()
+async def _(matcher: Matcher, matches: AlcMatches):
+    if matches.head_matched:
+        await matcher.finish(
+            f'{matches.error_info!r}\n' if matches.error_info is not None else '' + '输入"茶服 --help"查看帮助'
+        )

nonebot_plugin_tetris_stats/game_data_processor/tos_data_processor/constant.py

@@ -0,0 +1,4 @@
+from ...utils.typing import GameType
+
+GAME_TYPE: GameType = 'TOS'
+BASE_URL = 'https://teatube.cn:8888/'

nonebot_plugin_tetris_stats/game_data_processor/tos_data_processor/processor.py

@@ -0,0 +1,221 @@
+from dataclasses import dataclass
+from re import match
+from typing import Any
+from urllib.parse import urlencode
+
+from nonebot_plugin_orm import get_session
+from pydantic import parse_raw_as
+
+from ...db import create_or_update_bind
+from ...utils.exception import MessageFormatError, RequestError
+from ...utils.request import Request, splice_url
+from ...utils.typing import GameType
+from .. import ProcessedData as ProcessedDataMeta
+from .. import Processor as ProcessorMeta
+from .. import RawResponse as RawResponseMeta
+from .. import User as UserMeta
+from .constant import BASE_URL, GAME_TYPE
+from .schemas.user_info import SuccessModel as InfoSuccess
+from .schemas.user_info import UserInfo
+from .schemas.user_profile import UserProfile
+
+
+@dataclass
+class User(UserMeta):
+    teaid: str | None = None
+    name: str | None = None
+
+
+@dataclass
+class RawResponse(RawResponseMeta):
+    user_profile: dict[frozenset[tuple[str, Any]], bytes]
+    user_info: bytes | None = None
+
+
+@dataclass
+class ProcessedData(ProcessedDataMeta):
+    user_profile: dict[frozenset[tuple[str, Any]], UserProfile]
+    user_info: InfoSuccess | None = None
+
+
+@dataclass
+class GameData:
+    num: int
+    pps: float
+    lpm: float
+    apm: float
+    adpm: float
+    apl: float
+    adpl: float
+    vs: float
+
+
+def identify_user_info(info: str) -> User | MessageFormatError:
+    if (
+        match(
+            r'^(?!\.)(?!com[0-9]$)(?!con$)(?!lpt[0-9]$)(?!nul$)(?!prn$)[^\-][^\+][^\|\*\?\\\s\!:<>/$"]*[^\.\|\*\?\\\s\!:<>/$"]+$',
+            info,
+        )
+        and info.isdigit() is False
+        and 2 <= len(info) <= 18  # noqa: PLR2004
+    ):
+        return User(name=info)
+    if info.isdigit():
+        return User(teaid=info)
+    return MessageFormatError('用户名/QQ号不合法')
+
+
+class Processor(ProcessorMeta):
+    user: User
+    raw_response: RawResponse
+    processed_data: ProcessedData
+
+    def __init__(self, event_id: int, user: User, command_args: list[str]) -> None:
+        super().__init__(event_id, user, command_args)
+        self.raw_response = RawResponse(user_profile={})
+        self.processed_data = ProcessedData(user_profile={})
+
+    @property
+    def game_platform(self) -> GameType:
+        return GAME_TYPE
+
+    async def handle_bind(self, platform: str, account: str) -> str:
+        """处理绑定消息"""
+        self.command_type = 'bind'
+        await self.get_user()
+        if self.user.name is None:
+            raise  # FIXME: 不知道怎么才能把这类型给变过来了
+        async with get_session() as session:
+            return await create_or_update_bind(
+                session=session,
+                chat_platform=platform,
+                chat_account=account,
+                game_platform=GAME_TYPE,
+                game_account=self.user.name,
+            )
+
+    async def handle_query(self) -> str:
+        """处理查询消息"""
+        self.command_type = 'query'
+        await self.get_user()
+        return await self.generate_message()
+
+    async def get_user(self) -> None:
+        """
+        用于获取 UserName 和 UserID 的函数
+        """
+        if self.user.name is None:
+            self.user.name = (await self.get_user_info()).data.name
+        if self.user.teaid is None:
+            self.user.teaid = (await self.get_user_info()).data.teaid
+
+    async def get_user_info(self) -> InfoSuccess:
+        """获取用户信息"""
+        if self.processed_data.user_info is None:
+            if self.user.teaid is not None:
+                url = splice_url(
+                    [
+                        BASE_URL,
+                        'getTeaIdInfo',
+                        f'?{urlencode({"teaId":self.user.teaid})}',
+                    ]
+                )
+            else:
+                url = splice_url(
+                    [
+                        BASE_URL,
+                        'getUsernameInfo',
+                        f'?{urlencode({"username":self.user.name})}',
+                    ]
+                )
+            self.raw_response.user_info = await Request.request(url)
+            user_info: UserInfo = parse_raw_as(UserInfo, self.raw_response.user_info)  # type: ignore[arg-type]
+            if not isinstance(user_info, InfoSuccess):
+                raise RequestError(f'用户信息请求错误:\n{user_info.error}')
+            self.processed_data.user_info = user_info
+        return self.processed_data.user_info
+
+    async def get_user_profile(self, other_parameter: dict[str, Any] | None = None) -> UserProfile:
+        """获取用户数据"""
+        if other_parameter is None:
+            other_parameter = {}
+        fset = frozenset(other_parameter.items())
+        if self.processed_data.user_profile.get(fset) is None:
+            self.raw_response.user_profile[fset] = await Request.request(
+                splice_url(
+                    [
+                        BASE_URL,
+                        'getProfile',
+                        f'?{urlencode({"id":self.user.teaid or self.user.name},**other_parameter)}',
+                    ]
+                )
+            )
+            self.processed_data.user_profile[fset] = UserProfile.parse_raw(self.raw_response.user_profile[fset])
+        return self.processed_data.user_profile[fset]
+
+    async def get_game_data(self) -> GameData | None:
+        """获取游戏数据"""
+        user_profile = await self.get_user_profile()
+        if user_profile.data == []:
+            return None
+        weighted_total_lpm = weighted_total_apm = weighted_total_adpm = 0.0
+        total_time = 0.0
+        num = 0
+        for i in user_profile.data:
+            # 排除单人局和时间为0的游戏
+            # 茶: 不计算没挖掘的局, 即使apm和lpm也如此
+            if i.num_players == 1 or i.time == 0 or i.dig is None:
+                continue
+            # 加权计算
+            time = i.time / 1000
+            lpm = 24 * (i.pieces / time)
+            apm = (i.attack / time) * 60
+            adpm = ((i.attack + i.dig) / time) * 60
+            weighted_total_lpm += lpm * time
+            weighted_total_apm += apm * time
+            weighted_total_adpm += adpm * time
+            total_time += time
+            num += 1
+            if num == 50:  # noqa: PLR2004  # TODO: 将查询局数作为可选命令参数
+                break
+        if num == 0:
+            return None
+        # TODO: 如果有效局数不满50, 没有无dig信息的局, 且userData['data']内有50个局, 则继续往前获取信息
+        lpm = weighted_total_lpm / total_time
+        apm = weighted_total_apm / total_time
+        adpm = weighted_total_adpm / total_time
+        return GameData(
+            num=num,
+            pps=round(lpm / 24, 2),
+            lpm=round(lpm, 2),
+            apm=round(apm, 2),
+            adpm=round(adpm, 2),
+            apl=round((apm / lpm), 2),
+            adpl=round((adpm / lpm), 2),
+            vs=round((adpm / 60 * 100), 2),
+        )
+
+    async def generate_message(self) -> str:
+        """生成消息"""
+        user_info = (await self.get_user_info()).data
+        message = f'用户 {user_info.name} ({user_info.teaid}) '
+        if user_info.ranked_games == '0':
+            message += '暂无段位统计数据'
+        else:
+            message += f', 段位分 {round(float(user_info.rating_now),2)}±{round(float(user_info.rd_now),2)} ({round(float(user_info.vol_now),2)}) '
+        game_data = await self.get_game_data()
+        if game_data is None:
+            message += ', 暂无游戏数据'
+        else:
+            message += f', 最近 {game_data.num} 局数据'
+            message += f"\nL'PM: {game_data.lpm} ( {game_data.pps} pps )"
+            message += f'\nAPM: {game_data.apm} ( x{game_data.apl} )'
+            message += f'\nADPM: {game_data.adpm} ( x{game_data.adpl} ) ( {game_data.vs}vs )'
+        message += (
+            f'\n40L: {float(user_info.pb_sprint)/1000:.2f}s'
+            if user_info.pb_sprint != 2147483647  # noqa: PLR2004
+            else ''
+        )
+        message += f'\nMarathon: {user_info.pb_marathon}' if user_info.pb_marathon != 0 else ''
+        message += f'\nChallenge: {user_info.pb_challenge}' if user_info.pb_challenge != 0 else ''
+        return message

nonebot_plugin_tetris_stats/game_data_processor/tos_data_processor/schemas/user_info.py

@@ -0,0 +1,86 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+
+class SuccessModel(BaseModel):
+    class Data(BaseModel):
+        class PeriodMatch(BaseModel):
+            name: str
+            teaid: str = Field(..., alias='teaId')
+            rating: str
+            rd: str
+            start_time: datetime = Field(..., alias='startTime')
+            end_time: datetime = Field(..., alias='endTime')
+            win: str
+            lose: str
+            score: str
+
+        class UserDataTotalItem(BaseModel):
+            time_map: str = Field(..., alias='timeMap')
+            pieces_map: str = Field(..., alias='piecesMap')
+            clear_lines_map: str = Field(..., alias='clearLinesMap')
+            attacks_map: str = Field(..., alias='attacksMap')
+            dig_map: str = Field(..., alias='digMap')
+            send_map: str = Field(..., alias='sendMap')
+            rise_map: str = Field(..., alias='riseMap')
+            offset_map: str = Field(..., alias='offsetMap')
+            receive_map: str = Field(..., alias='receiveMap')
+            games_map: str = Field(..., alias='gamesMap')
+            tetris_map: str = Field(..., alias='tetrisMap')
+            combo_map: str = Field(..., alias='comboMap')
+            tspin_map: str = Field(..., alias='tspinMap')
+            b2b_map: str = Field(..., alias='b2bMap')
+            perfect_clear_map: str = Field(..., alias='perfectClearMap')
+            time_no_map: str = Field(..., alias='timeNoMap')
+            pieces_no_map: str = Field(..., alias='piecesNoMap')
+            clear_lines_no_map: str = Field(..., alias='clearLinesNoMap')
+            attacks_no_map: str = Field(..., alias='attacksNoMap')
+            dig_no_map: str = Field(..., alias='digNoMap')
+            send_no_map: str = Field(..., alias='sendNoMap')
+            rise_no_map: str = Field(..., alias='riseNoMap')
+            offset_no_map: str = Field(..., alias='offsetNoMap')
+            receive_no_map: str = Field(..., alias='receiveNoMap')
+            games_no_map: str = Field(..., alias='gamesNoMap')
+            tetris_no_map: str = Field(..., alias='tetrisNoMap')
+            combo_no_map: str = Field(..., alias='comboNoMap')
+            tspin_no_map: str = Field(..., alias='tspinNoMap')
+            b2b_no_map: str = Field(..., alias='b2bNoMap')
+            perfect_clear_no_map: str = Field(..., alias='perfectClearNoMap')
+
+        teaid: str = Field(..., alias='teaId')
+        name: str
+        total_exp: str = Field(..., alias='totalExp')
+        ranking: str
+        ranked_games: str = Field(..., alias='rankedGames')
+        rating_now: str = Field(..., alias='ratingNow')
+        rd_now: str = Field(..., alias='rdNow')
+        vol_now: str = Field(..., alias='volNow')
+        rating_last: str = Field(..., alias='ratingLast')
+        rd_last: str = Field(..., alias='rdLast')
+        vol_last: str = Field(..., alias='volLast')
+        period_matches: list[PeriodMatch] = Field(..., alias='periodMatches')
+        user_data_total: list[UserDataTotalItem] = Field(..., alias='userDataTotal')
+        ranking_items: str = Field(..., alias='rankingItems')
+        ranking_game_items: str = Field(..., alias='rankingGameItems')
+        training_level: str = Field(..., alias='trainingLevel')
+        training_wins: str = Field(..., alias='trainingWins')
+        pb_sprint: str = Field(..., alias='PBSprint')
+        pb_marathon: str = Field(..., alias='PBMarathon')
+        pb_challenge: str = Field(..., alias='PBChallenge')
+        register_date: datetime = Field(..., alias='registerDate')
+        last_login_date: datetime = Field(..., alias='lastLoginDate')
+
+    code: int
+    success: Literal[True]
+    data: Data
+
+
+class FailedModel(BaseModel):
+    code: int
+    success: Literal[False]
+    error: str
+
+
+UserInfo = SuccessModel | FailedModel

nonebot_plugin_tetris_stats/game_data_processor/tos_data_processor/schemas/user_profile.py

@@ -0,0 +1,33 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel
+
+
+class UserProfile(BaseModel):
+    class Data(BaseModel):
+        idmultiplayergameresult: int
+        iduser: str
+        teaid: str
+        time: int
+        clear_lines: int
+        attack: int
+        send: int
+        offset: int
+        receive: int
+        rise: int
+        dig: int
+        pieces: int
+        max_combo: int
+        pc_count: int
+        place: int
+        num_players: int
+        fumen_code: Literal['0', '1']  # wtf
+        rule_set: str
+        garbage: str
+        idmultiplayergame: int
+        datetime: datetime
+
+    code: int
+    success: bool
+    data: list[Data]

nonebot_plugin_tetris_stats/games/__init__.py

@@ -1,55 +0,0 @@
-from collections.abc import Callable
-
-from nonebot.adapters import Bot
-from nonebot.matcher import Matcher
-from nonebot.message import run_postprocessor
-from nonebot.typing import T_Handler
-from nonebot_plugin_alconna import AlcMatches, Alconna, At, CommandMeta, on_alconna
-
-from .. import ns
-from ..utils.exception import MessageFormatError, NeedCatchError
-
-command: Alconna = Alconna(
-    ['tetris-stats', 'tstats'],
-    namespace=ns,
-    meta=CommandMeta(
-        description='俄罗斯方块相关游戏数据查询',
-        fuzzy_match=True,
-    ),
-)
-
-alc = on_alconna(
-    command=command,
-    skip_for_unmatch=False,
-    auto_send_output=True,
-    use_origin=True,
-)
-
-
-def add_block_handlers(handler: Callable[[T_Handler], T_Handler]) -> None:
-    @handler
-    async def _(bot: Bot, matcher: Matcher, target: At):
-        if isinstance(target, At) and target.target == bot.self_id:
-            await matcher.finish('不能查询bot的信息')
-
-
-from . import tetrio, top, tos  # noqa: F401, E402
-
-
-@alc.handle()
-async def _(matcher: Matcher, account: MessageFormatError):
-    await matcher.finish(str(account))
-
-
-@alc.handle()
-async def _(matcher: Matcher, matches: AlcMatches):
-    if matches.head_matched and matches.options != {} or matches.main_args == {}:
-        await matcher.finish(
-            (f'{matches.error_info!r}\n' if matches.error_info is not None else '')
-            + f'输入"{matches.header_result} --help"查看帮助'
-        )
-
-
-@run_postprocessor
-async def _(matcher: Matcher, exception: NeedCatchError):
-    await matcher.send(str(exception))

nonebot_plugin_tetris_stats/games/constant.py

@@ -1 +0,0 @@
-CANT_VERIFY_MESSAGE = '* 由于无法验证绑定信息, 不能保证查询到的用户为本人\n'

nonebot_plugin_tetris_stats/games/schemas.py

@@ -1,24 +0,0 @@
-from abc import ABC, abstractmethod
-from typing import Generic, TypeVar
-
-from pydantic import BaseModel
-
-from ..utils.typing import GameType
-
-T = TypeVar('T', bound=GameType)
-
-
-class BaseUser(BaseModel, ABC, Generic[T]):
-    """游戏用户"""
-
-    platform: T
-
-    def __eq__(self, other: object) -> bool:
-        if isinstance(other, BaseUser):
-            return self.unique_identifier == other.unique_identifier
-        return False
-
-    @property
-    @abstractmethod
-    def unique_identifier(self) -> str:
-        raise NotImplementedError

nonebot_plugin_tetris_stats/games/tetrio/__init__.py

@@ -1,36 +0,0 @@
-from nonebot_plugin_alconna import Subcommand
-
-from ...utils.exception import MessageFormatError
-from .. import alc
-from .. import command as main_command
-from .api import Player
-from .constant import USER_ID, USER_NAME
-
-
-def get_player(user_id_or_name: str) -> Player | MessageFormatError:
-    if USER_ID.match(user_id_or_name):
-        return Player(user_id=user_id_or_name, trust=True)
-    if USER_NAME.match(user_id_or_name):
-        return Player(user_name=user_id_or_name, trust=True)
-    return MessageFormatError('用户名/ID不合法')
-
-
-command = Subcommand(
-    'TETR.IO',
-    alias=['TETRIO', 'tetr.io', 'tetrio', 'io'],
-    dest='TETRIO',
-    help_text='TETR.IO 游戏相关指令',
-)
-
-
-from . import bind, config, query, record  # noqa: E402
-
-main_command.add(command)
-
-__all__ = [
-    'alc',
-    'bind',
-    'config',
-    'query',
-    'record',
-]

nonebot_plugin_tetris_stats/games/tetrio/api/__init__.py

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

nonebot_plugin_tetris_stats/games/tetrio/api/cache.py

@@ -1,39 +0,0 @@
-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 yarl import URL
-
-from ....config.config import config
-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)
-
-
-class Cache:
-    cache = ACache(ACache.MEMORY)
-    task: ClassVar[WeakValueDictionary[URL, Lock]] = WeakValueDictionary()
-
-    @classmethod
-    async def get(cls, url: URL) -> 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

@@ -1,18 +0,0 @@
-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
-from .typing import Records, Summaries
-
-
-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', Records, Summaries]] = mapped_column(String(32), 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

@@ -1,229 +0,0 @@
-from enum import Enum
-from types import MappingProxyType
-from typing import Literal, NamedTuple, cast, overload
-
-from async_lru import alru_cache
-from nonebot.compat import type_validate_json
-
-from ....db import anti_duplicate_add
-from ....utils.exception import RequestError
-from ..constant import BASE_URL, USER_ID, USER_NAME
-from .cache import Cache
-from .models import TETRIOHistoricalData
-from .schemas.base import FailedModel
-from .schemas.records.solo import Solo as SoloRecord
-from .schemas.records.solo import SoloSuccessModel as RecordsSoloSuccessModel
-from .schemas.summaries import (
-    AchievementsSuccessModel,
-    SummariesModel,
-    ZenithSuccessModel,
-    ZenSuccessModel,
-)
-from .schemas.summaries import (
-    SoloSuccessModel as SummariesSoloSuccessModel,
-)
-from .schemas.summaries.base import User as SummariesUser
-from .schemas.summaries.league import LeagueSuccessModel
-from .schemas.user import User
-from .schemas.user_info import UserInfo, UserInfoSuccess
-from .typing import Records, Summaries
-
-
-class RecordModeType(str, Enum):
-    Sprint = '40l'
-    Blitz = 'blitz'
-
-
-class RecordType(str, Enum):
-    Top = 'top'
-    Recent = 'recent'
-    Progression = 'progression'
-
-
-class RecordKey(NamedTuple):
-    mode_type: RecordModeType
-    record_type: RecordType
-
-    def to_records(self) -> Records:
-        return cast(Records, f'{self.mode_type.value}_{self.record_type.value}')
-
-
-class Player:
-    __SUMMARIES_MAPPING: MappingProxyType[Summaries, type[SummariesModel]] = MappingProxyType(
-        {
-            '40l': SummariesSoloSuccessModel,
-            'blitz': SummariesSoloSuccessModel,
-            'zenith': ZenithSuccessModel,
-            'zenithex': ZenithSuccessModel,
-            'league': LeagueSuccessModel,
-            'zen': ZenSuccessModel,
-            'achievements': AchievementsSuccessModel,
-        }
-    )
-
-    @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._summaries: dict[Summaries, SummariesModel] = {}
-        self._records: dict[RecordKey, RecordsSoloSuccessModel] = {}
-
-    @property
-    def _request_user_parameter(self) -> str:
-        return self.user_id or cast(str, self.user_name).lower()
-
-    @property
-    async def user(self) -> User:
-        if self.__user is not None:
-            return self.__user
-        if (user := (await self._get_local_summaries_user())) is not None:
-            self.__user = User(
-                ID=user.id,
-                name=user.username,
-            )
-        else:
-            user_info = await self.get_info()
-            self.__user = User(
-                ID=user_info.data.id,
-                name=user_info.data.username,
-            )
-        self.user_id = self.__user.ID
-        self.user_name = self.__user.name
-        return self.__user
-
-    async def get_info(self) -> UserInfoSuccess:
-        """Get User Info"""
-        if self._user_info is None:
-            raw_user_info = await Cache.get(BASE_URL / 'users' / 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(
-                    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
-
-    @overload
-    async def get_summaries(self, summaries_type: Literal['40l', 'blitz']) -> SummariesSoloSuccessModel: ...
-    @overload
-    async def get_summaries(self, summaries_type: Literal['zenith', 'zenithex']) -> ZenithSuccessModel: ...
-    @overload
-    async def get_summaries(self, summaries_type: Literal['zen']) -> ZenSuccessModel: ...
-    @overload
-    async def get_summaries(self, summaries_type: Literal['league']) -> LeagueSuccessModel: ...
-    @overload
-    async def get_summaries(self, summaries_type: Literal['achievements']) -> AchievementsSuccessModel: ...
-
-    async def get_summaries(self, summaries_type: Summaries) -> SummariesModel:
-        if summaries_type not in self._summaries:
-            raw_summaries = await Cache.get(
-                BASE_URL / 'users' / self._request_user_parameter / 'summaries' / summaries_type
-            )
-            summaries: SummariesModel | FailedModel = type_validate_json(
-                self.__SUMMARIES_MAPPING[summaries_type] | FailedModel,  # type: ignore[arg-type]
-                raw_summaries,
-            )
-            if isinstance(summaries, FailedModel):
-                msg = f'用户Summaries数据请求错误:\n{summaries.error}'
-                raise RequestError(msg)
-            self._summaries[summaries_type] = summaries
-            await anti_duplicate_add(
-                TETRIOHistoricalData(
-                    user_unique_identifier=(await self.user).unique_identifier,
-                    api_type=summaries_type,
-                    data=summaries,
-                    update_time=summaries.cache.cached_at,
-                ),
-            )
-        return self._summaries[summaries_type]
-
-    @property
-    async def sprint(self) -> SummariesSoloSuccessModel:
-        return await self.get_summaries('40l')
-
-    @property
-    async def blitz(self) -> SummariesSoloSuccessModel:
-        return await self.get_summaries('blitz')
-
-    @property
-    async def zen(self) -> ZenSuccessModel:
-        return await self.get_summaries('zen')
-
-    @property
-    async def league(self) -> LeagueSuccessModel:
-        return await self.get_summaries('league')
-
-    async def _get_local_summaries_user(self) -> SummariesUser | None:
-        allow_summaries: set[Literal['40l', 'blitz', 'zenith', 'zenithex']] = {
-            '40l',
-            'blitz',
-            'zenith',
-            'zenithex',
-        }
-        if has_summaries := (allow_summaries & self._summaries.keys()):
-            for i in has_summaries:
-                if (record := (await self.get_summaries(i)).data.record) is not None:
-                    return record.user
-        return None
-
-    @property
-    @alru_cache
-    async def avatar_revision(self) -> int | None:
-        if self._user_info is not None:
-            return self._user_info.data.avatar_revision
-        if (user := (await self._get_local_summaries_user())) is not None:
-            return user.avatar_revision
-        return (await self.get_info()).data.avatar_revision
-
-    @property
-    @alru_cache
-    async def banner_revision(self) -> int | None:
-        if self._user_info is not None:
-            return self._user_info.data.banner_revision
-        if (user := (await self._get_local_summaries_user())) is not None:
-            return user.banner_revision
-        return (await self.get_info()).data.banner_revision
-
-    async def get_records(self, mode_type: RecordModeType, records_type: RecordType) -> RecordsSoloSuccessModel:
-        if (record_key := RecordKey(mode_type, records_type)) not in self._records:
-            raw_records = await Cache.get(
-                BASE_URL / 'users' / self._request_user_parameter / 'records' / mode_type / records_type,
-            )
-            records: RecordsSoloSuccessModel | FailedModel = type_validate_json(SoloRecord, raw_records)  # type: ignore[arg-type]
-            if isinstance(records, FailedModel):
-                msg = f'用户Summaries数据请求错误:\n{records.error}'
-                raise RequestError(msg)
-            self._records[record_key] = records
-            await anti_duplicate_add(
-                TETRIOHistoricalData(
-                    user_unique_identifier=(await self.user).unique_identifier,
-                    api_type=record_key.to_records(),
-                    data=records,
-                    update_time=records.cache.cached_at,
-                ),
-            )
-        return self._records[record_key]

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

@@ -1,61 +0,0 @@
-from datetime import datetime
-from typing import Literal
-
-from pydantic import BaseModel
-
-
-class AggregateStats(BaseModel):
-    apm: float
-    pps: float
-    vsscore: float
-
-
-class Finesse(BaseModel):
-    combo: int
-    faults: int
-    perfectpieces: int
-
-
-class Clears(BaseModel):
-    singles: int
-    doubles: int
-    triples: int
-    quads: int
-    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
-    cleared: int
-
-
-class P(BaseModel):  # what is P
-    pri: float
-    sec: float
-    ter: float
-
-
-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/base/solo.py

@@ -1,65 +0,0 @@
-from datetime import datetime
-from typing import Literal
-
-from pydantic import BaseModel, Field
-
-from ..base import P
-from . import AggregateStats, Clears, Finesse, Garbage
-
-
-class Time(BaseModel):
-    start: int
-    zero: bool
-    locked: bool
-    prev: int
-    frameoffset: int
-
-
-class Stats(BaseModel):
-    seed: int | None = None  # ?: 不知道是之后都没有了还是还会有
-    lines: int
-    level_lines: int
-    level_lines_needed: int
-    inputs: int
-    holds: int = 0
-    time: Time | None = None  # ?: 不知道是之后都没有了还是还会有
-    score: int
-    zenlevel: int
-    zenprogress: int
-    level: int
-    combo: int
-    currentcombopower: int | None = None
-    topcombo: int
-    btb: int
-    topbtb: int
-    currentbtbchainpower: int | None = None
-    tspins: int
-    piecesplaced: int
-    clears: Clears
-    garbage: Garbage
-    kills: int
-    finesse: Finesse
-    finaltime: float
-
-
-class Results(BaseModel):
-    aggregatestats: AggregateStats
-    stats: Stats
-    gameoverreason: str
-
-
-class Record(BaseModel):
-    id: str = Field(..., alias='_id')
-    replayid: str
-    stub: bool
-    gamemode: Literal['40l', 'blitz']
-    pb: bool
-    oncepb: bool
-    ts: datetime
-    revolution: None
-    otherusers: list
-    leaderboards: list[str]
-    results: Results
-    extras: dict
-    disputed: bool
-    p: P

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/leaderboards/ar.py

@@ -1,27 +0,0 @@
-from pydantic import BaseModel, Field
-
-from ..base import SuccessModel
-from .base import Entry as BaseEntry
-
-
-class ArCounts(BaseModel):
-    bronze: int | None = Field(None, alias='1')
-    silver: int | None = Field(None, alias='2')
-    gold: int | None = Field(None, alias='3')
-    platinum: int | None = Field(None, alias='4')
-    diamond: int | None = Field(None, alias='5')
-    issued: int | None = Field(None, alias='100')
-    top10: int | None = Field(None, alias='t10')
-
-
-class Entry(BaseEntry):
-    ar: int
-    ar_counts: ArCounts
-
-
-class Data(BaseModel):
-    entries: list[Entry]
-
-
-class ArSuccessModel(SuccessModel):
-    data: Data

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

@@ -1,30 +0,0 @@
-from datetime import datetime
-
-from pydantic import BaseModel, Field
-
-from ...typing import Rank
-from ..base import P
-
-
-class League(BaseModel):
-    gamesplayed: int
-    gameswon: int
-    rating: int
-    rank: Rank
-    decaying: bool
-
-
-class Entry(BaseModel):
-    id: str = Field(..., alias='_id')
-    username: str
-    role: str
-    xp: float
-    league: League
-    supporter: bool | None = None
-    verified: bool
-    country: str | None = None
-    ts: datetime
-    gamesplayed: int
-    gameswon: int
-    gametime: float
-    p: P

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/leaderboards/solo.py

@@ -1,12 +0,0 @@
-from pydantic import BaseModel
-
-from ..base import SuccessModel
-from ..summaries.solo import Record
-
-
-class Data(BaseModel):
-    entries: list[Record]
-
-
-class SoloSuccessModel(SuccessModel):
-    data: Data

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/leaderboards/xp.py

@@ -1,12 +0,0 @@
-from pydantic import BaseModel
-
-from ..base import SuccessModel
-from .base import Entry
-
-
-class Data(BaseModel):
-    entries: list[Entry]
-
-
-class XpSuccessModel(SuccessModel):
-    data: Data

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/leaderboards/zenith.py

@@ -1,12 +0,0 @@
-from pydantic import BaseModel
-
-from ..base import SuccessModel
-from ..summaries.zenith import Record
-
-
-class Data(BaseModel):
-    entries: list[Record]
-
-
-class ZenithSuccessModel(SuccessModel):
-    data: Data

nonebot_plugin_tetris_stats/games/tetrio/api/schemas/records/solo.py

@@ -1,17 +0,0 @@
-from typing import TypeAlias
-
-from pydantic import BaseModel
-
-from ..base import FailedModel, SuccessModel
-from ..base.solo import Record
-
-
-class Data(BaseModel):
-    entries: list[Record]
-
-
-class SoloSuccessModel(SuccessModel):
-    data: Data
-
-
-Solo: TypeAlias = SoloSuccessModel | FailedModel

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

@@ -1,21 +0,0 @@
-from .achievements import Achievements, AchievementsSuccessModel
-from .league import LeagueSuccessModel
-from .solo import Solo, SoloSuccessModel
-from .zen import Zen, ZenSuccessModel
-from .zenith import Zenith, ZenithEx, ZenithSuccessModel
-
-SummariesModel = AchievementsSuccessModel | SoloSuccessModel | ZenSuccessModel | LeagueSuccessModel | ZenithSuccessModel
-
-__all__ = [
-    'Achievements',
-    'AchievementsSuccessModel',
-    'LeagueSuccessModel',
-    'Solo',
-    'SoloSuccessModel',
-    'SummariesModel',
-    'Zen',
-    'Zenith',
-    'ZenithEx',
-    'ZenithSuccessModel',
-    'ZenSuccessModel',
-]

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

@@ -1,29 +0,0 @@
-from typing import TypeAlias
-
-from pydantic import BaseModel
-
-from ..base import FailedModel, SuccessModel
-
-
-class Achievement(BaseModel):
-    # 这**都是些啥
-    k: int
-    o: int
-    rt: int
-    vt: int
-    min: int
-    deci: int
-    name: str
-    object: str
-    category: str
-    hidden: bool
-    desc: str
-    n: str
-    stub: bool
-
-
-class AchievementsSuccessModel(SuccessModel):
-    data: list[Achievement]
-
-
-Achievements: TypeAlias = AchievementsSuccessModel | FailedModel

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

@@ -1,11 +0,0 @@
-from pydantic import BaseModel
-
-
-class User(BaseModel):
-    id: str
-    username: str
-    avatar_revision: int | None
-    banner_revision: int | None
-    country: str | None
-    verified: int | None = None
-    supporter: int

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

@@ -1,102 +0,0 @@
-from typing import Literal
-
-from pydantic import BaseModel, Field
-
-from ...typing import Rank, S1Rank, S1ValidRank
-from ..base import SuccessModel
-
-
-class PastInner(BaseModel):
-    season: str
-    username: str
-    country: str | None = None
-    placement: int | None = None
-    gamesplayed: int
-    gameswon: int
-    glicko: float
-    gxe: float
-    tr: float
-    rd: float
-    rank: S1Rank
-    bestrank: S1ValidRank
-    ranked: bool
-    apm: float
-    pps: float
-    vs: float
-
-
-class Past(BaseModel):
-    first: PastInner | None = Field(default=None, alias='1')
-
-
-class BaseData(BaseModel):
-    decaying: bool
-    past: Past
-
-
-class NeverPlayedData(BaseData):
-    gamesplayed: Literal[0]
-    gameswon: Literal[0]
-    glicko: Literal[-1]
-    rd: Literal[-1]
-    gxe: Literal[-1]
-    tr: Literal[-1]
-    rank: Literal['z']
-    apm: None = None
-    pps: None = None
-    vs: None = None
-    standing: Literal[-1]
-    standing_local: Literal[-1]
-    prev_rank: None
-    prev_at: Literal[-1]
-    next_rank: None
-    next_at: Literal[-1]
-    percentile: Literal[-1]
-    percentile_rank: Literal['z']
-
-
-class NeverRatedData(BaseData):
-    gamesplayed: Literal[1, 2, 3, 4, 5, 6, 7, 8, 9]
-    gameswon: int
-    glicko: Literal[-1]
-    rd: Literal[-1]
-    gxe: Literal[-1]
-    tr: Literal[-1]
-    apm: float
-    pps: float
-    vs: float
-    rank: Literal['z']
-    standing: Literal[-1]
-    standing_local: Literal[-1]
-    prev_rank: None
-    prev_at: Literal[-1]
-    next_rank: None
-    next_at: Literal[-1]
-    percentile: Literal[-1]
-    percentile_rank: Literal['z']
-
-
-class RatedData(BaseData):
-    gamesplayed: int
-    gameswon: int
-    glicko: float
-    rd: float
-    gxe: float
-    tr: float
-    rank: Rank
-    bestrank: Rank
-    standing: int
-    apm: float
-    pps: float
-    vs: float
-    standing_local: int
-    prev_rank: Rank | None = None
-    prev_at: int
-    next_rank: Rank | None = None
-    next_at: int
-    percentile: float
-    percentile_rank: str
-
-
-class LeagueSuccessModel(SuccessModel):
-    data: NeverPlayedData | NeverRatedData | RatedData

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

@@ -1,24 +0,0 @@
-from typing import TypeAlias
-
-from pydantic import BaseModel
-
-from ..base import FailedModel, SuccessModel
-from ..base.solo import Record as BaseRecord
-from .base import User
-
-
-class Record(BaseRecord):
-    user: User
-
-
-class Data(BaseModel):
-    record: Record | None
-    rank: int
-    rank_local: int
-
-
-class SoloSuccessModel(SuccessModel):
-    data: Data
-
-
-Solo: TypeAlias = SoloSuccessModel | FailedModel

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

@@ -1,17 +0,0 @@
-from typing import TypeAlias
-
-from pydantic import BaseModel
-
-from ..base import FailedModel, SuccessModel
-
-
-class Data(BaseModel):
-    level: int
-    score: int
-
-
-class ZenSuccessModel(SuccessModel):
-    data: Data
-
-
-Zen: TypeAlias = ZenSuccessModel | FailedModel

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

@@ -1,116 +0,0 @@
-from datetime import datetime
-from typing import Literal, TypeAlias
-
-from pydantic import BaseModel, Field
-
-from ..base import AggregateStats, FailedModel, Finesse, P, SuccessModel
-from ..base import Clears as BaseClears
-from ..base import Garbage as BaseGarbage
-from .base import User
-
-
-class Clears(BaseClears):
-    pentas: int
-    minitspintriples: int
-    minitspinquads: int
-    tspinpentas: int
-
-
-class Garbage(BaseGarbage):
-    sent_nomult: int
-    maxspike: int
-    maxspike_nomult: int
-
-
-class _Zenith(BaseModel):
-    altitude: float
-    rank: float
-    peakrank: float
-    avgrankpts: float
-    floor: int
-    targetingfactor: float
-    targetinggrace: float
-    totalbonus: float
-    revives: int
-    revives_total: int = Field(..., alias='revivesTotal')
-    speedrun: bool
-    speedrun_seen: bool
-    splits: list[int]
-
-
-class Stats(BaseModel):
-    lines: int
-    level_lines: int
-    level_lines_needed: int
-    inputs: int
-    holds: int
-    score: int
-    zenlevel: int
-    zenprogress: int
-    level: int
-    combo: int
-    topcombo: int
-    combopower: int
-    btb: int
-    topbtb: int
-    btbpower: int
-    tspins: int
-    piecesplaced: int
-    clears: Clears
-    garbage: Garbage
-    kills: int
-    finesse: Finesse
-    zenith: _Zenith
-    finaltime: float
-
-
-class Results(BaseModel):
-    aggregatestats: AggregateStats
-    stats: Stats
-    gameoverreason: str
-
-
-class ExtrasZenith(BaseModel):
-    mods: list[str]
-
-
-class Extras(BaseModel):
-    zenith: ExtrasZenith
-
-
-class Record(BaseModel):
-    id: str = Field(..., alias='_id')
-    replayid: str
-    stub: bool
-    gamemode: Literal['zenith', 'zenithex']
-    pb: bool
-    oncepb: bool
-    ts: datetime
-    revolution: None
-    user: User
-    otherusers: list
-    leaderboards: list[str]
-    results: Results
-    extras: Extras
-    disputed: bool
-    p: P
-
-
-class Best(BaseModel):
-    record: None  # WTF
-    rank: int
-
-
-class Data(BaseModel):
-    record: Record | None
-    rank: int
-    rank_local: int
-    best: Best
-
-
-class ZenithSuccessModel(SuccessModel):
-    data: Data
-
-
-Zenith: TypeAlias = ZenithSuccessModel | FailedModel
-ZenithEx: TypeAlias = ZenithSuccessModel | FailedModel

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

@@ -1,59 +0,0 @@
-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 | None = None
-    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

@@ -1,18 +0,0 @@
-from typing import Literal
-
-from typing_extensions import override
-
-from ....schemas import BaseUser
-from ...constant import GAME_TYPE
-
-
-class User(BaseUser[Literal['IO']]):
-    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

@@ -1,66 +0,0 @@
-from datetime import datetime
-from typing import Literal
-
-from pydantic import BaseModel, Field
-
-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 Discord(BaseModel):
-    id: str
-    username: str
-
-
-class Connections(BaseModel):
-    discord: Discord | None = None
-
-
-class Distinguishment(BaseModel):
-    type: str
-
-
-class Data(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 | None = None
-    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 UserInfoSuccess(BaseSuccessModel):
-    data: Data
-
-
-UserInfo = UserInfoSuccess | FailedModel

nonebot_plugin_tetris_stats/games/tetrio/api/tetra_league.py

@@ -1,55 +0,0 @@
-from typing import Literal, NamedTuple, overload
-
-from msgspec import Struct, to_builtins
-from nonebot.compat import type_validate_json
-
-from ....utils.exception import RequestError
-from ..constant import BASE_URL
-from .cache import Cache
-from .schemas.base import FailedModel
-from .schemas.tetra_league import TetraLeague, TetraLeagueSuccess
-
-
-class Parameter(Struct, omit_defaults=True):
-    after: float | None = None
-    before: float | None = None
-    limit: int | None = None
-    country: str | None = None
-
-
-async def leaderboard(parameter: Parameter | None = None) -> TetraLeagueSuccess:
-    league: TetraLeague = type_validate_json(
-        TetraLeague,  # type: ignore[arg-type]
-        (await Cache.get(BASE_URL / 'users/lists/league' % to_builtins(parameter))),
-    )
-    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(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

@@ -1,45 +0,0 @@
-from typing import Literal
-
-S1ValidRank = Literal[
-    'x+',
-    'x',
-    'u',
-    'ss',
-    's+',
-    's',
-    's-',
-    'a+',
-    'a',
-    'a-',
-    'b+',
-    'b',
-    'b-',
-    'c+',
-    'c',
-    'c-',
-    'd+',
-    'd',
-]
-S1Rank = S1ValidRank | Literal['z']
-
-ValidRank = Literal['x+'] | S1ValidRank
-Rank = ValidRank | Literal['z']  # 未定级
-
-Summaries = Literal[
-    '40l',
-    'blitz',
-    'zenith',
-    'zenithex',
-    'league',
-    'zen',
-    'achievements',
-]
-
-Records = Literal[
-    '40l_top',
-    '40l_recent',
-    '40l_progression',
-    'blitz_top',
-    'blitz_recent',
-    'blitz_progression',
-]

nonebot_plugin_tetris_stats/games/tetrio/bind.py

@@ -1,86 +0,0 @@
-from hashlib import md5
-
-from arclet.alconna import Arg, ArgFlag
-from nonebot_plugin_alconna import Args, Subcommand
-from nonebot_plugin_alconna.uniseg import UniMessage
-from nonebot_plugin_orm import get_session
-from nonebot_plugin_session import EventSession
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import User
-from nonebot_plugin_userinfo import BotUserInfo, UserInfo
-from yarl import URL
-
-from ...db import BindStatus, create_or_update_bind, trigger
-from ...utils.host import HostPage, get_self_netloc
-from ...utils.image import get_avatar
-from ...utils.render import Bind, render
-from ...utils.render.schemas.base import Avatar, People
-from ...utils.screenshot import screenshot
-from . import alc, command, get_player
-from .api import Player
-from .constant import GAME_TYPE
-
-command.add(
-    Subcommand(
-        'bind',
-        Args(
-            Arg(
-                'account',
-                get_player,
-                notice='TETR.IO 用户名 / ID',
-                flags=[ArgFlag.HIDDEN],
-            )
-        ),
-        help_text='绑定 TETR.IO 账号',
-    )
-)
-
-alc.shortcut(
-    '(?i:io)(?i:绑定|绑|bind)',
-    command='tstats TETR.IO bind',
-    humanized='io绑定',
-)
-
-
-@alc.assign('TETRIO.bind')
-async def _(nb_user: User, account: Player, event_session: EventSession, bot_info: UserInfo = BotUserInfo()):  # noqa: B008
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='bind',
-        command_args=[],
-    ):
-        user = await account.user
-        async with get_session() as session:
-            bind_status = await create_or_update_bind(
-                session=session,
-                user=nb_user,
-                game_platform=GAME_TYPE,
-                game_account=user.unique_identifier,
-            )
-        if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
-            netloc = get_self_netloc()
-            async with HostPage(
-                await render(
-                    'v1/binding',
-                    Bind(
-                        platform='TETR.IO',
-                        status='unknown',
-                        user=People(
-                            avatar=str(
-                                URL(f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}')
-                                % {'revision': avatar_revision}
-                            )
-                            if (avatar_revision := (await account.avatar_revision)) is not None and avatar_revision != 0
-                            else Avatar(type='identicon', hash=md5(user.ID.encode()).hexdigest()),  # noqa: S324
-                            name=user.name.upper(),
-                        ),
-                        bot=People(
-                            avatar=await get_avatar(bot_info, 'Data URI', '../../static/logo/logo.svg'),
-                            name=bot_info.user_name,
-                        ),
-                        command='io查我',
-                    ),
-                )
-            ) as page_hash:
-                await UniMessage.image(raw=await screenshot(f'http://{netloc}/host/{page_hash}.html')).finish()

nonebot_plugin_tetris_stats/games/tetrio/config.py

@@ -1,51 +0,0 @@
-from arclet.alconna import Arg
-from nonebot_plugin_alconna import Option, Subcommand
-from nonebot_plugin_alconna.uniseg import UniMessage
-from nonebot_plugin_orm import async_scoped_session
-from nonebot_plugin_session import EventSession
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import User
-from sqlalchemy import select
-
-from ...db import trigger
-from . import alc, command
-from .constant import GAME_TYPE
-from .models import TETRIOUserConfig
-from .typing import Template
-
-command.add(
-    Subcommand(
-        'config',
-        Option(
-            '--default-template',
-            Arg('template', Template, notice='模板版本'),
-            alias=['-DT', 'DefaultTemplate'],
-            help_text='设置默认查询模板',
-        ),
-        help_text='TETR.IO 查询个性化配置',
-    ),
-)
-
-alc.shortcut(
-    '(?i:io)(?i:配置|配|config)',
-    command='tstats TETR.IO config',
-    humanized='io配置',
-)
-
-
-@alc.assign('TETRIO.config')
-async def _(user: User, session: async_scoped_session, event_session: EventSession, template: Template):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='config',
-        command_args=[f'--default-template {template}'],
-    ):
-        config = (await session.scalars(select(TETRIOUserConfig).where(TETRIOUserConfig.id == user.id))).one_or_none()
-        if config is None:
-            config = TETRIOUserConfig(id=user.id, query_template=template)
-            session.add(config)
-        else:
-            config.query_template = template
-        await session.commit()
-        await UniMessage('配置成功').finish()

nonebot_plugin_tetris_stats/games/tetrio/constant.py

@@ -1,36 +0,0 @@
-from re import compile
-from typing import Literal
-
-from yarl import URL
-
-from .api.typing import ValidRank
-
-GAME_TYPE: Literal['IO'] = 'IO'
-
-BASE_URL = URL('https://ch.tetr.io/api/')
-
-RANK_PERCENTILE: dict[ValidRank, float] = {
-    'x': 1,
-    'u': 5,
-    'ss': 11,
-    's+': 17,
-    's': 23,
-    's-': 30,
-    'a+': 38,
-    'a': 46,
-    'a-': 54,
-    'b+': 62,
-    'b': 70,
-    'b-': 78,
-    'c+': 84,
-    'c': 90,
-    'c-': 95,
-    'd+': 97.5,
-    'd': 100,
-}
-
-TR_MIN = 0
-TR_MAX = 25000
-
-USER_ID = compile(r'^[a-f0-9]{24}$')
-USER_NAME = compile(r'^[a-zA-Z0-9_-]{3,16}$')

nonebot_plugin_tetris_stats/games/tetrio/models.py

@@ -1,10 +0,0 @@
-from nonebot_plugin_orm import Model
-from sqlalchemy import String
-from sqlalchemy.orm import Mapped, MappedAsDataclass, mapped_column
-
-from .typing import Template
-
-
-class TETRIOUserConfig(MappedAsDataclass, Model):
-    id: Mapped[int] = mapped_column(primary_key=True)
-    query_template: Mapped[Template] = mapped_column(String(2))

nonebot_plugin_tetris_stats/games/tetrio/query.py

@@ -1,275 +0,0 @@
-from asyncio import gather
-from datetime import datetime, timedelta, timezone
-from hashlib import md5
-from typing import TYPE_CHECKING, TypeVar
-
-from arclet.alconna import Arg, ArgFlag
-from nonebot import get_driver
-from nonebot.adapters import Event
-from nonebot.matcher import Matcher
-from nonebot_plugin_alconna import Args, At, Option, Subcommand
-from nonebot_plugin_alconna.uniseg import UniMessage
-from nonebot_plugin_orm import get_session
-from nonebot_plugin_session import EventSession
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import User as NBUser
-from nonebot_plugin_user import get_user
-from sqlalchemy import select
-from yarl import URL
-
-from ...db import query_bind_info, trigger
-from ...utils.host import HostPage, get_self_netloc
-from ...utils.metrics import get_metrics
-from ...utils.render import render
-from ...utils.render.schemas.base import Avatar
-from ...utils.render.schemas.tetrio.user.info_v2 import (
-    Badge,
-    Blitz,
-    Sprint,
-    Statistic,
-    TetraLeague,
-    TetraLeagueStatistic,
-    Zen,
-)
-from ...utils.render.schemas.tetrio.user.info_v2 import Info as V2TemplateInfo
-from ...utils.render.schemas.tetrio.user.info_v2 import User as V2TemplateUser
-from ...utils.screenshot import screenshot
-from ...utils.typing import Me
-from .. import add_block_handlers, alc
-from ..constant import CANT_VERIFY_MESSAGE
-from . import command, get_player
-from .api import Player
-from .api.schemas.summaries.league import LeagueSuccessModel, NeverPlayedData, NeverRatedData
-from .constant import GAME_TYPE
-from .models import TETRIOUserConfig
-from .typing import Template
-
-if TYPE_CHECKING:
-    from .api.schemas.summaries import SoloSuccessModel, ZenSuccessModel
-    from .api.schemas.user import User
-    from .api.schemas.user_info import UserInfoSuccess
-
-UTC = timezone.utc
-
-driver = get_driver()
-
-command.add(
-    Subcommand(
-        'query',
-        Args(
-            Arg(
-                'target',
-                At | Me,
-                notice='@想要查询的人 / 自己',
-                flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
-            ),
-            Arg(
-                'account',
-                get_player,
-                notice='TETR.IO 用户名 / ID',
-                flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
-            ),
-        ),
-        Option(
-            '--template',
-            Arg('template', Template),
-            alias=['-T'],
-            help_text='要使用的查询模板',
-        ),
-        help_text='查询 TETR.IO 游戏信息',
-    ),
-)
-
-alc.shortcut(
-    '(?i:io)(?i:查询|查|query|stats)',
-    command='tstats TETR.IO query',
-    humanized='io查',
-)
-alc.shortcut(
-    'fkosk',
-    command='tstats TETR.IO query',
-    arguments=['我'],
-    fuzzy=False,
-    humanized='An Easter egg!',
-)
-
-add_block_handlers(alc.assign('TETRIO.query'))
-
-
-@alc.assign('TETRIO.query')
-async def _(  # noqa: PLR0913
-    user: NBUser,
-    event: Event,
-    matcher: Matcher,
-    target: At | Me,
-    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='query',
-        command_args=[f'--default-template {template}'] if template is not None else [],
-    ):
-        async with get_session() as session:
-            bind = await query_bind_info(
-                session=session,
-                user=await get_user(
-                    event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
-                ),
-                game_platform=GAME_TYPE,
-            )
-            if template is None:
-                template = await session.scalar(
-                    select(TETRIOUserConfig.query_template).where(TETRIOUserConfig.id == user.id)
-                )
-        if bind is None:
-            await matcher.finish('未查询到绑定信息')
-        message = UniMessage(CANT_VERIFY_MESSAGE)
-        player = Player(user_id=bind.game_account, trust=True)
-        await (message + UniMessage.image(raw=await make_query_image_v2(player))).finish()
-
-
-@alc.assign('TETRIO.query')
-async def _(user: NBUser, account: Player, 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='query',
-        command_args=[f'--default-template {template}'] if template is not None else [],
-    ):
-        async with get_session() as session:
-            if template is None:
-                template = await session.scalar(
-                    select(TETRIOUserConfig.query_template).where(TETRIOUserConfig.id == user.id)
-                )
-        await (UniMessage.image(raw=await make_query_image_v2(account))).finish()
-
-
-N = TypeVar('N', int, float)
-
-
-def handling_special_value(value: N) -> N | None:
-    return value if value != -1 else None
-
-
-async def make_query_image_v2(player: Player) -> bytes:
-    user: User
-    user_info: UserInfoSuccess
-    league: LeagueSuccessModel
-    sprint: SoloSuccessModel
-    blitz: SoloSuccessModel
-    zen: ZenSuccessModel
-    avatar_revision: int | None
-    banner_revision: int | None
-    # TODO)) 有没有什么办法能让这类型推导成功)
-    user, user_info, league, sprint, blitz, zen, avatar_revision, banner_revision = await gather(  # type: ignore[assignment]
-        player.user,
-        player.get_info(),
-        player.league,
-        player.sprint,
-        player.blitz,
-        player.zen,
-        player.avatar_revision,
-        player.banner_revision,
-    )
-
-    if sprint.data.record is not None:
-        duration = timedelta(milliseconds=sprint.data.record.results.stats.finaltime).total_seconds()
-        sprint_value = f'{duration:.3f}s' if duration < 60 else f'{duration // 60:.0f}m {duration % 60:.3f}s'  # noqa: PLR2004
-    else:
-        sprint_value = 'N/A'
-
-    play_time: str | None
-    if (game_time := handling_special_value(user_info.data.gametime)) is not None:
-        if game_time // 3600 > 0:
-            play_time = f'{game_time//3600:.0f}h {game_time % 3600 // 60:.0f}m {game_time % 60:.0f}s'
-        elif game_time // 60 > 0:
-            play_time = f'{game_time//60:.0f}m {game_time % 60:.0f}s'
-        else:
-            play_time = f'{game_time:.0f}s'
-    else:
-        play_time = game_time
-    netloc = get_self_netloc()
-    async with HostPage(
-        await render(
-            'v2/tetrio/user/info',
-            V2TemplateInfo(
-                user=V2TemplateUser(
-                    id=user.ID,
-                    name=user.name.upper(),
-                    bio=user_info.data.bio,
-                    banner=str(
-                        URL(f'http://{netloc}/host/resource/tetrio/banners/{user.ID}') % {'revision': banner_revision}
-                    )
-                    if banner_revision is not None and banner_revision != 0
-                    else None,
-                    avatar=str(
-                        URL(f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}') % {'revision': avatar_revision}
-                    )
-                    if avatar_revision is not None and avatar_revision != 0
-                    else Avatar(
-                        type='identicon',
-                        hash=md5(user.ID.encode()).hexdigest(),  # noqa: S324
-                    ),
-                    badges=[
-                        Badge(
-                            id=i.id,
-                            description=i.label,
-                            group=i.group,
-                            receive_at=i.ts if isinstance(i.ts, datetime) else None,
-                        )
-                        for i in user_info.data.badges
-                    ],
-                    country=user_info.data.country,
-                    role=user_info.data.role,
-                    xp=user_info.data.xp,
-                    friend_count=user_info.data.friend_count,
-                    supporter_tier=user_info.data.supporter_tier,
-                    bad_standing=user_info.data.badstanding or False,
-                    verified=user_info.data.verified or False,
-                    playtime=play_time,
-                    join_at=user_info.data.ts,
-                ),
-                tetra_league=TetraLeague(
-                    rank=league.data.rank,
-                    highest_rank='z' if isinstance(league.data, NeverRatedData) else league.data.bestrank,
-                    tr=round(league.data.tr, 2),
-                    glicko=round(league.data.glicko, 2),
-                    rd=round(league.data.rd, 2),
-                    global_rank=league.data.standing,
-                    country_rank=league.data.standing_local,
-                    pps=(metrics := get_metrics(pps=league.data.pps, apm=league.data.apm, vs=league.data.vs)).pps,
-                    apm=metrics.apm,
-                    apl=metrics.apl,
-                    vs=metrics.vs,
-                    adpl=metrics.adpl,
-                    statistic=TetraLeagueStatistic(total=league.data.gamesplayed, wins=league.data.gameswon),
-                    decaying=league.data.decaying,
-                    history=None,
-                )
-                if not isinstance(league.data, NeverPlayedData)
-                else None,
-                statistic=Statistic(
-                    total=handling_special_value(user_info.data.gamesplayed),
-                    wins=handling_special_value(user_info.data.gameswon),
-                ),
-                sprint=Sprint(
-                    time=sprint_value,
-                    global_rank=sprint.data.rank,
-                    play_at=sprint.data.record.ts,
-                )
-                if sprint.data.record is not None
-                else None,
-                blitz=Blitz(
-                    score=blitz.data.record.results.stats.score,
-                    global_rank=blitz.data.rank,
-                    play_at=blitz.data.record.ts,
-                )
-                if blitz.data.record is not None
-                else None,
-                zen=Zen(level=zen.data.level, score=zen.data.score),
-            ),
-        ),
-    ) as page_hash:
-        return await screenshot(f'http://{netloc}/host/{page_hash}.html')

nonebot_plugin_tetris_stats/games/tetrio/record/__init__.py

@@ -1,33 +0,0 @@
-from arclet.alconna import Arg, ArgFlag
-from nonebot_plugin_alconna import Args, At, Subcommand
-
-from ....utils.typing import Me
-from .. import command as base_command
-from .. import get_player
-
-command = Subcommand(
-    'record',
-    Args(
-        Arg(
-            'target',
-            At | Me,
-            notice='@想要查询的人 / 自己',
-            flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
-        ),
-        Arg(
-            'account',
-            get_player,
-            notice='TETR.IO 用户名 / ID',
-            flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
-        ),
-    ),
-)
-
-from . import blitz, sprint  # noqa: E402
-
-base_command.add(command)
-
-__all__ = [
-    'blitz',
-    'sprint',
-]

nonebot_plugin_tetris_stats/games/tetrio/record/blitz.py

@@ -1,150 +0,0 @@
-from asyncio import gather
-from datetime import timedelta
-from hashlib import md5
-
-from nonebot.adapters import Event
-from nonebot.matcher import Matcher
-from nonebot_plugin_alconna import At, Option
-from nonebot_plugin_alconna.uniseg import UniMessage
-from nonebot_plugin_orm import get_session
-from nonebot_plugin_session import EventSession
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import get_user
-from yarl import URL
-
-from ....db import query_bind_info, trigger
-from ....utils.exception import RecordNotFoundError
-from ....utils.host import HostPage, get_self_netloc
-from ....utils.metrics import get_metrics
-from ....utils.render import render
-from ....utils.render.schemas.base import Avatar
-from ....utils.render.schemas.tetrio.record.base import Finesse, Max, Mini, Tspins, User
-from ....utils.render.schemas.tetrio.record.blitz import Record, Statistic
-from ....utils.screenshot import screenshot
-from ....utils.typing import Me
-from ...constant import CANT_VERIFY_MESSAGE
-from .. import alc
-from ..api.player import Player
-from ..constant import GAME_TYPE
-from . import command
-
-command.add(Option('--blitz', dest='blitz'))
-
-alc.shortcut(
-    '(?i:io)(?i:记录|record)(?i:blitz)',
-    command='tstats TETR.IO record --blitz',
-    humanized='io记录blitz',
-)
-
-
-@alc.assign('TETRIO.record.blitz')
-async def _(
-    event: Event,
-    matcher: Matcher,
-    target: At | Me,
-    event_session: EventSession,
-):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='record',
-        command_args=['--blitz'],
-    ):
-        async with get_session() as session:
-            bind = await query_bind_info(
-                session=session,
-                user=await get_user(
-                    event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
-                ),
-                game_platform=GAME_TYPE,
-            )
-        if bind is None:
-            await matcher.finish('未查询到绑定信息')
-        message = UniMessage(CANT_VERIFY_MESSAGE)
-        player = Player(user_id=bind.game_account, trust=True)
-        await (message + UniMessage.image(raw=await make_blitz_image(player))).finish()
-
-
-@alc.assign('TETRIO.record.blitz')
-async def _(account: Player, event_session: EventSession):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='record',
-        command_args=['--blitz'],
-    ):
-        await UniMessage.image(raw=await make_blitz_image(account)).finish()
-
-
-async def make_blitz_image(player: Player) -> bytes:
-    user, blitz = await gather(player.user, player.blitz)
-    if blitz.data.record is None:
-        msg = f'未找到用户 {user.name.upper()} 的 Blitz 记录'
-        raise RecordNotFoundError(msg)
-    stats = blitz.data.record.results.stats
-    clears = stats.clears
-    duration = timedelta(milliseconds=stats.finaltime).total_seconds()
-    metrics = get_metrics(pps=stats.piecesplaced / duration)
-    netloc = get_self_netloc()
-    async with HostPage(
-        page=await render(
-            'v2/tetrio/record/blitz',
-            Record(
-                type='best',
-                user=User(
-                    id=user.ID,
-                    name=user.name.upper(),
-                    avatar=str(
-                        URL(f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}') % {'revision': avatar_revision}
-                    )
-                    if (avatar_revision := (await player.avatar_revision)) is not None and avatar_revision != 0
-                    else Avatar(
-                        type='identicon',
-                        hash=md5(user.ID.encode()).hexdigest(),  # noqa: S324
-                    ),
-                ),
-                replay_id=blitz.data.record.replayid,
-                rank=blitz.data.rank,
-                personal_rank=1,
-                statistic=Statistic(
-                    keys=stats.inputs,
-                    kpp=round(stats.inputs / stats.piecesplaced, 2),
-                    kps=round(stats.inputs / duration, 2),
-                    max=Max(
-                        combo=max((0, stats.topcombo - 1)),
-                        btb=max((0, stats.topbtb - 1)),
-                    ),
-                    pieces=stats.piecesplaced,
-                    pps=metrics.pps,
-                    lines=stats.lines,
-                    lpm=metrics.lpm,
-                    holds=stats.holds,
-                    score=stats.score,
-                    spp=round(stats.score / stats.piecesplaced, 2),
-                    single=clears.singles,
-                    double=clears.doubles,
-                    triple=clears.triples,
-                    quad=clears.quads,
-                    tspins=Tspins(
-                        total=clears.realtspins,
-                        single=clears.tspinsingles,
-                        double=clears.tspindoubles,
-                        triple=clears.tspintriples,
-                        mini=Mini(
-                            total=clears.minitspins,
-                            single=clears.minitspinsingles,
-                            double=clears.minitspindoubles,
-                        ),
-                    ),
-                    all_clear=clears.allclear,
-                    finesse=Finesse(
-                        faults=stats.finesse.faults,
-                        accuracy=round(stats.finesse.perfectpieces / stats.piecesplaced * 100, 2),
-                    ),
-                    level=stats.level,
-                ),
-                play_at=blitz.data.record.ts,
-            ),
-        )
-    ) as page_hash:
-        return await screenshot(f'http://{netloc}/host/{page_hash}.html')

nonebot_plugin_tetris_stats/games/tetrio/record/sprint.py

@@ -1,150 +0,0 @@
-from asyncio import gather
-from datetime import timedelta
-from hashlib import md5
-
-from nonebot.adapters import Event
-from nonebot.matcher import Matcher
-from nonebot_plugin_alconna import At, Option
-from nonebot_plugin_alconna.uniseg import UniMessage
-from nonebot_plugin_orm import get_session
-from nonebot_plugin_session import EventSession
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import get_user
-from yarl import URL
-
-from ....db import query_bind_info, trigger
-from ....utils.exception import RecordNotFoundError
-from ....utils.host import HostPage, get_self_netloc
-from ....utils.metrics import get_metrics
-from ....utils.render import render
-from ....utils.render.schemas.base import Avatar
-from ....utils.render.schemas.tetrio.record.base import Finesse, Max, Mini, Statistic, Tspins, User
-from ....utils.render.schemas.tetrio.record.sprint import Record
-from ....utils.screenshot import screenshot
-from ....utils.typing import Me
-from ...constant import CANT_VERIFY_MESSAGE
-from .. import alc
-from ..api.player import Player
-from ..constant import GAME_TYPE
-from . import command
-
-command.add(Option('--40l', dest='sprint'))
-
-alc.shortcut(
-    '(?i:io)(?i:记录|record)(?i:40l)',
-    command='tstats TETR.IO record --40l',
-    humanized='io记录40l',
-)
-
-
-@alc.assign('TETRIO.record.sprint')
-async def _(
-    event: Event,
-    matcher: Matcher,
-    target: At | Me,
-    event_session: EventSession,
-):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='record',
-        command_args=['--40l'],
-    ):
-        async with get_session() as session:
-            bind = await query_bind_info(
-                session=session,
-                user=await get_user(
-                    event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
-                ),
-                game_platform=GAME_TYPE,
-            )
-        if bind is None:
-            await matcher.finish('未查询到绑定信息')
-        message = UniMessage(CANT_VERIFY_MESSAGE)
-        player = Player(user_id=bind.game_account, trust=True)
-        await (message + UniMessage.image(raw=await make_sprint_image(player))).finish()
-
-
-@alc.assign('TETRIO.record.sprint')
-async def _(account: Player, event_session: EventSession):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='record',
-        command_args=['--40l'],
-    ):
-        await UniMessage.image(raw=await make_sprint_image(account)).finish()
-
-
-async def make_sprint_image(player: Player) -> bytes:
-    user, sprint = await gather(player.user, player.sprint)
-    if sprint.data.record is None:
-        msg = f'未找到用户 {user.name.upper()} 的 40L 记录'
-        raise RecordNotFoundError(msg)
-    stats = sprint.data.record.results.stats
-    clears = stats.clears
-    duration = timedelta(milliseconds=stats.finaltime).total_seconds()
-    sprint_value = f'{duration:.3f}s' if duration < 60 else f'{duration // 60:.0f}m {duration % 60:.3f}s'  # noqa: PLR2004
-    metrics = get_metrics(pps=stats.piecesplaced / duration)
-    netloc = get_self_netloc()
-    async with HostPage(
-        page=await render(
-            'v2/tetrio/record/40l',
-            Record(
-                type='best',
-                user=User(
-                    id=user.ID,
-                    name=user.name.upper(),
-                    avatar=str(
-                        URL(f'http://{netloc}/host/resource/tetrio/avatars/{user.ID}') % {'revision': avatar_revision}
-                    )
-                    if (avatar_revision := (await player.avatar_revision)) is not None and avatar_revision != 0
-                    else Avatar(
-                        type='identicon',
-                        hash=md5(user.ID.encode()).hexdigest(),  # noqa: S324
-                    ),
-                ),
-                time=sprint_value,
-                replay_id=sprint.data.record.replayid,
-                rank=sprint.data.rank,
-                personal_rank=1,
-                statistic=Statistic(
-                    keys=stats.inputs,
-                    kpp=round(stats.inputs / stats.piecesplaced, 2),
-                    kps=round(stats.inputs / duration, 2),
-                    max=Max(
-                        combo=max((0, stats.topcombo - 1)),
-                        btb=max((0, stats.topbtb - 1)),
-                    ),
-                    pieces=stats.piecesplaced,
-                    pps=metrics.pps,
-                    lines=stats.lines,
-                    lpm=metrics.lpm,
-                    holds=stats.holds,
-                    score=stats.score,
-                    single=clears.singles,
-                    double=clears.doubles,
-                    triple=clears.triples,
-                    quad=clears.quads,
-                    tspins=Tspins(
-                        total=clears.realtspins,
-                        single=clears.tspinsingles,
-                        double=clears.tspindoubles,
-                        triple=clears.tspintriples,
-                        mini=Mini(
-                            total=clears.minitspins,
-                            single=clears.minitspinsingles,
-                            double=clears.minitspindoubles,
-                        ),
-                    ),
-                    all_clear=clears.allclear,
-                    finesse=Finesse(
-                        faults=stats.finesse.faults,
-                        accuracy=round(stats.finesse.perfectpieces / stats.piecesplaced * 100, 2),
-                    ),
-                ),
-                play_at=sprint.data.record.ts,
-            ),
-        )
-    ) as page_hash:
-        return await screenshot(f'http://{netloc}/host/{page_hash}.html')

nonebot_plugin_tetris_stats/games/tetrio/typing.py

@@ -1,3 +0,0 @@
-from typing import Literal
-
-Template = Literal['v1', 'v2']

nonebot_plugin_tetris_stats/games/top/__init__.py

@@ -1,59 +0,0 @@
-from arclet.alconna import Arg, ArgFlag
-from nonebot_plugin_alconna import Args, At, Subcommand
-
-from ...utils.exception import MessageFormatError
-from ...utils.typing import Me
-from .. import add_block_handlers, alc, command
-from .api import Player
-from .constant import USER_NAME
-
-
-def get_player(name: str) -> Player | MessageFormatError:
-    if USER_NAME.match(name):
-        return Player(user_name=name, trust=True)
-    return MessageFormatError('用户名/ID不合法')
-
-
-command.add(
-    Subcommand(
-        'TOP',
-        Subcommand(
-            'bind',
-            Args(
-                Arg(
-                    'account',
-                    get_player,
-                    notice='TOP 用户名 / ID',
-                    flags=[ArgFlag.HIDDEN],
-                )
-            ),
-            help_text='绑定 TOP 账号',
-        ),
-        Subcommand(
-            'query',
-            Args(
-                Arg(
-                    'target',
-                    At | Me,
-                    notice='@想要查询的人 / 自己',
-                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
-                ),
-                Arg(
-                    'account',
-                    get_player,
-                    notice='TOP 用户名',
-                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
-                ),
-            ),
-            help_text='查询 TOP 游戏信息',
-        ),
-        help_text='TOP 游戏相关指令',
-    )
-)
-
-alc.shortcut('(?i:top)(?i:绑定|绑|bind)', {'command': 'tstats TOP bind', 'humanized': 'top绑定'})
-alc.shortcut('(?i:top)(?i:查询|查|query|stats)', {'command': 'tstats TOP query', 'humanized': 'top查'})
-
-add_block_handlers(alc.assign('TOP.query'))
-
-from . import bind, query  # noqa: E402, F401

nonebot_plugin_tetris_stats/games/top/api/__init__.py

@@ -1,3 +0,0 @@
-from .player import Player
-
-__all__ = ['Player']

nonebot_plugin_tetris_stats/games/top/api/models.py

@@ -1,17 +0,0 @@
-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.user_profile import UserProfile
-
-
-class TOPHistoricalData(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 Profile']] = mapped_column(String(16), index=True)
-    data: Mapped[UserProfile] = mapped_column(PydanticType(get_model=[], models={UserProfile}))
-    update_time: Mapped[datetime] = mapped_column(DateTime, index=True)

nonebot_plugin_tetris_stats/games/top/api/player.py

@@ -1,72 +0,0 @@
-from contextlib import suppress
-from datetime import datetime, timezone
-from io import StringIO
-
-from lxml import etree
-from pandas import read_html
-
-from ....config.config import config
-from ....db import anti_duplicate_add
-from ....utils.request import Request
-from ..constant import BASE_URL, USER_NAME
-from .models import TOPHistoricalData
-from .schemas.user import User
-from .schemas.user_profile import Data, UserProfile
-
-UTC = timezone.utc
-
-request = Request(config.tetris.proxy.top or config.tetris.proxy.main)
-
-
-class Player:
-    def __init__(self, *, user_name: str, trust: bool = False) -> None:
-        self.user_name = user_name
-        if not trust and not USER_NAME.match(self.user_name):
-            msg = 'Invalid user name'
-            raise ValueError(msg)
-        self.__user: User | None = None
-        self._user_profile: UserProfile | None = None
-
-    @property
-    async def user(self) -> User:
-        if self.__user is None:
-            profile = await self.get_profile()
-            self.__user = User(user_name=profile.user_name)
-        return self.__user
-
-    async def get_profile(self) -> UserProfile:
-        """获取用户信息"""
-        if self._user_profile is None:
-            raw_user_profile = await request.request(BASE_URL / 'profile.php' % {'user': self.user_name}, is_json=False)
-            self._user_profile = self._parse_profile(raw_user_profile)
-            await anti_duplicate_add(
-                TOPHistoricalData(
-                    user_unique_identifier=(await self.user).unique_identifier,
-                    api_type='User Profile',
-                    data=self._user_profile,
-                    update_time=datetime.now(tz=UTC),
-                ),
-            )
-        return self._user_profile
-
-    @staticmethod
-    def _parse_profile(original_user_profile: bytes) -> UserProfile:
-        html = etree.HTML(original_user_profile)
-        user_name = html.xpath('//div[@class="mycontent"]/h1/text()')[0].replace("'s profile", '')
-        today = None
-        with suppress(ValueError):
-            today = Data(
-                lpm=float(str(html.xpath('//div[@class="mycontent"]/text()[3]')[0]).replace('lpm:', '').strip()),
-                apm=float(str(html.xpath('//div[@class="mycontent"]/text()[4]')[0]).replace('apm:', '').strip()),
-            )
-        table = StringIO(
-            etree.tostring(
-                html.xpath('//div[@class="mycontent"]/table[@class="mytable"]')[0],
-                encoding='utf-8',
-            ).decode()
-        )
-        dataframe = read_html(table, encoding='utf-8', header=0)[0]
-        total: list[Data] = []
-        for _, value in dataframe.iterrows():
-            total.append(Data(lpm=value['lpm'], apm=value['apm']))
-        return UserProfile(user_name=user_name, today=today, total=total or None)

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

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

nonebot_plugin_tetris_stats/games/top/api/schemas/user_profile.py

@@ -1,12 +0,0 @@
-from pydantic import BaseModel
-
-
-class Data(BaseModel):
-    lpm: float
-    apm: float
-
-
-class UserProfile(BaseModel):
-    user_name: str
-    today: Data | None
-    total: list[Data] | None

nonebot_plugin_tetris_stats/games/top/bind.py

@@ -1,62 +0,0 @@
-from nonebot_plugin_alconna.uniseg import UniMessage
-from nonebot_plugin_orm import get_session
-from nonebot_plugin_session import EventSession
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import User
-from nonebot_plugin_userinfo import BotUserInfo, EventUserInfo, UserInfo
-
-from ...db import BindStatus, create_or_update_bind, trigger
-from ...utils.host import HostPage, get_self_netloc
-from ...utils.image import get_avatar
-from ...utils.render import Bind, render
-from ...utils.render.schemas.base import People
-from ...utils.screenshot import screenshot
-from . import alc
-from .api import Player
-from .constant import GAME_TYPE
-
-
-@alc.assign('TOP.bind')
-async def _(
-    nb_user: User,
-    account: Player,
-    event_session: EventSession,
-    event_user_info: UserInfo = EventUserInfo(),  # noqa: B008
-    bot_info: UserInfo = BotUserInfo(),  # noqa: B008
-):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='bind',
-        command_args=[],
-    ):
-        user = await account.user
-        async with get_session() as session:
-            bind_status = await create_or_update_bind(
-                session=session,
-                user=nb_user,
-                game_platform=GAME_TYPE,
-                game_account=user.unique_identifier,
-            )
-        if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
-            async with HostPage(
-                await render(
-                    'v1/binding',
-                    Bind(
-                        platform=GAME_TYPE,
-                        status='unknown',
-                        user=People(
-                            avatar=await get_avatar(event_user_info, 'Data URI', None),
-                            name=user.user_name,
-                        ),
-                        bot=People(
-                            avatar=await get_avatar(bot_info, 'Data URI', '../../static/logo/logo.svg'),
-                            name=bot_info.user_name,
-                        ),
-                        command='top查我',
-                    ),
-                )
-            ) as page_hash:
-                await UniMessage.image(
-                    raw=await screenshot(f'http://{get_self_netloc()}/host/{page_hash}.html')
-                ).finish()

nonebot_plugin_tetris_stats/games/top/constant.py

@@ -1,10 +0,0 @@
-from re import compile
-from typing import Literal
-
-from yarl import URL
-
-GAME_TYPE: Literal['TOP'] = 'TOP'
-
-BASE_URL = URL('http://tetrisonline.pl/top/')
-
-USER_NAME = compile(r'^[a-zA-Z0-9_]{1,16}$')

nonebot_plugin_tetris_stats/games/top/query.py

@@ -1,114 +0,0 @@
-from nonebot.adapters import 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
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import get_user
-
-from ...db import query_bind_info, trigger
-from ...utils.exception import FallbackError
-from ...utils.host import HostPage, get_self_netloc
-from ...utils.metrics import TetrisMetricsBasicWithLPM, get_metrics
-from ...utils.render import render
-from ...utils.render.avatar import get_avatar
-from ...utils.render.schemas.base import People
-from ...utils.render.schemas.top_info import Data as InfoData
-from ...utils.render.schemas.top_info import Info
-from ...utils.screenshot import screenshot
-from ...utils.typing import Me
-from ..constant import CANT_VERIFY_MESSAGE
-from . import alc
-from .api import Player
-from .api.schemas.user_profile import Data, UserProfile
-from .constant import GAME_TYPE
-
-
-@alc.assign('TOP.query')
-async def _(event: Event, matcher: Matcher, target: At | Me, event_session: EventSession):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='query',
-        command_args=[],
-    ):
-        async with get_session() as session:
-            bind = await query_bind_info(
-                session=session,
-                user=await get_user(
-                    event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
-                ),
-                game_platform=GAME_TYPE,
-            )
-        if bind is None:
-            await matcher.finish('未查询到绑定信息')
-        await (
-            UniMessage(CANT_VERIFY_MESSAGE)
-            + await make_query_result(await Player(user_name=bind.game_account, trust=True).get_profile())
-        ).finish()
-
-
-@alc.assign('TOP.query')
-async def _(account: Player, event_session: EventSession):
-    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_result(await account.get_profile())).finish()
-
-
-def get_avg_metrics(data: list[Data]) -> TetrisMetricsBasicWithLPM:
-    total_lpm = total_apm = 0.0
-    for value in data:
-        total_lpm += value.lpm
-        total_apm += value.apm
-    num = len(data)
-    return get_metrics(lpm=total_lpm / num, apm=total_apm / num)
-
-
-async def make_query_image(profile: UserProfile) -> bytes:
-    if profile.today is None or profile.total is None:
-        raise FallbackError
-    today = get_metrics(lpm=profile.today.lpm, apm=profile.today.apm)
-    history = get_avg_metrics(profile.total)
-    async with HostPage(
-        await render(
-            'v1/top/info',
-            Info(
-                user=People(avatar=get_avatar(profile.user_name), name=profile.user_name),
-                today=InfoData(pps=today.pps, lpm=today.lpm, apm=today.apm, apl=today.apl),
-                history=InfoData(pps=history.pps, lpm=history.lpm, apm=history.apm, apl=history.apl),
-            ),
-        )
-    ) as page_hash:
-        return await screenshot(f'http://{get_self_netloc()}/host/{page_hash}.html')
-
-
-def make_query_text(profile: UserProfile) -> UniMessage:
-    message = ''
-    if profile.today is not None:
-        today = get_metrics(lpm=profile.today.lpm, apm=profile.today.apm)
-        message += f'用户 {profile.user_name} 24小时内统计数据为: '
-        message += f"\nL'PM: {today.lpm} ( {today.pps} pps )"
-        message += f'\nAPM: {today.apm} ( x{today.apl} )'
-    else:
-        message += f'用户 {profile.user_name} 暂无24小时内统计数据'
-    if profile.total is not None:
-        total = get_avg_metrics(profile.total)
-        message += '\n历史统计数据为: '
-        message += f"\nL'PM: {total.lpm} ( {total.pps} pps )"
-        message += f'\nAPM: {total.apm} ( x{total.apl} )'
-    else:
-        message += '\n暂无历史统计数据'
-    return UniMessage(message)
-
-
-async def make_query_result(profile: UserProfile) -> UniMessage:
-    try:
-        return UniMessage.image(raw=await make_query_image(profile))
-    except FallbackError:
-        ...
-    return make_query_text(profile)

nonebot_plugin_tetris_stats/games/tos/__init__.py

@@ -1,64 +0,0 @@
-from arclet.alconna import Arg, ArgFlag
-from nonebot_plugin_alconna import Args, At, Subcommand
-
-from ...utils.exception import MessageFormatError
-from ...utils.typing import Me
-from .. import add_block_handlers, alc, command
-from .api import Player
-from .constant import USER_NAME
-
-
-def get_player(teaid_or_name: str) -> Player | MessageFormatError:
-    if (
-        teaid_or_name.startswith(('onebot-', 'qqguild-', 'kook-', 'discord-'))
-        and teaid_or_name.split('-', maxsplit=1)[1].isdigit()
-    ):
-        return Player(teaid=teaid_or_name, trust=True)
-    if USER_NAME.match(teaid_or_name) and not teaid_or_name.isdigit() and 2 <= len(teaid_or_name) <= 18:  # noqa: PLR2004
-        return Player(user_name=teaid_or_name, trust=True)
-    return MessageFormatError('用户名/ID不合法')
-
-
-command.add(
-    Subcommand(
-        'TOS',
-        Subcommand(
-            'bind',
-            Args(
-                Arg(
-                    'account',
-                    get_player,
-                    notice='茶服 用户名 / ID',
-                    flags=[ArgFlag.HIDDEN],
-                )
-            ),
-            help_text='绑定 茶服 账号',
-        ),
-        Subcommand(
-            'query',
-            Args(
-                Arg(
-                    'target',
-                    At | Me,
-                    notice='@想要查询的人 / 自己',
-                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
-                ),
-                Arg(
-                    'account',
-                    get_player,
-                    notice='茶服 用户名 / TeaID',
-                    flags=[ArgFlag.HIDDEN, ArgFlag.OPTIONAL],
-                ),
-            ),
-            help_text='查询 茶服 游戏信息',
-        ),
-        help_text='茶服 游戏相关指令',
-    )
-)
-
-alc.shortcut('(?i:tos|茶服)(?i:绑定|绑|bind)', {'command': 'tstats TOS bind', 'humanized': '茶服绑定'})
-alc.shortcut('(?i:tos|茶服)(?i:查询|查|query|stats)', {'command': 'tstats TOS query', 'humanized': '茶服查'})
-
-add_block_handlers(alc.assign('TOS.query'))
-
-from . import bind, query  # noqa: E402, F401

nonebot_plugin_tetris_stats/games/tos/api/__init__.py

@@ -1,3 +0,0 @@
-from .player import Player
-
-__all__ = ['Player']

nonebot_plugin_tetris_stats/games/tos/api/models.py

@@ -1,20 +0,0 @@
-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.user_info import UserInfoSuccess
-from .schemas.user_profile import UserProfile
-
-
-class TOSHistoricalData(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 Profile']] = mapped_column(String(16), index=True)
-    data: Mapped[UserInfoSuccess | UserProfile] = mapped_column(
-        PydanticType(get_model=[], models={UserInfoSuccess, UserProfile})
-    )
-    update_time: Mapped[datetime] = mapped_column(DateTime, index=True)

nonebot_plugin_tetris_stats/games/tos/api/player.py

@@ -1,108 +0,0 @@
-from datetime import datetime, timezone
-from typing import cast, overload
-
-from httpx import TimeoutException
-from nonebot.compat import type_validate_json
-from yarl import URL
-
-from ....config.config import config
-from ....db import anti_duplicate_add
-from ....utils.exception import RequestError
-from ....utils.request import Request
-from ..constant import BASE_URL, USER_NAME
-from .models import TOSHistoricalData
-from .schemas.user import User
-from .schemas.user_info import UserInfo, UserInfoSuccess
-from .schemas.user_profile import UserProfile
-
-UTC = timezone.utc
-
-request = Request(config.tetris.proxy.tos or config.tetris.proxy.main)
-
-
-class Player:
-    @overload
-    def __init__(self, *, teaid: str, trust: bool = False): ...
-    @overload
-    def __init__(self, *, user_name: str, trust: bool = False): ...
-    def __init__(self, *, teaid: str | None = None, user_name: str | None = None, trust: bool = False):
-        self.teaid = teaid
-        self.user_name = user_name
-        if not trust:
-            if self.teaid is not None:
-                if (
-                    not self.teaid.startswith(('onebot-', 'qqguild-', 'kook-', 'discord-'))
-                    or not self.teaid.split('-', maxsplit=1)[1].isdigit()
-                ):
-                    msg = 'Invalid teaid'
-                    raise ValueError(msg)
-            elif self.user_name is not None:
-                if not USER_NAME.match(self.user_name) or self.user_name.isdigit() or 2 > len(self.user_name) > 18:  # noqa: PLR2004
-                    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_profile: dict[str, UserProfile] = {}
-
-    @property
-    async def user(self) -> User:
-        if self.__user is None:
-            user_info = await self.get_info()
-            self.__user = User(teaid=user_info.data.teaid, name=user_info.data.name)
-            self.teaid = user_info.data.teaid
-            self.user_name = user_info.data.name
-        return self.__user
-
-    async def get_info(self) -> UserInfoSuccess:
-        """获取用户信息"""
-        if self._user_info is None:
-            path = str(
-                URL('getTeaIdInfo') % {'teaId': self.teaid}
-                if self.teaid is not None
-                else URL('getUsernameInfo') % {'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,)
-            )
-            user_info: UserInfo = type_validate_json(UserInfo, raw_user_info)  # type: ignore[arg-type]
-            if not isinstance(user_info, UserInfoSuccess):
-                msg = f'用户信息请求错误:\n{user_info.error}'
-                raise RequestError(msg)
-            self._user_info = user_info
-            await anti_duplicate_add(
-                TOSHistoricalData(
-                    user_unique_identifier=(await self.user).unique_identifier,
-                    api_type='User Info',
-                    data=user_info,
-                    update_time=datetime.now(UTC),
-                ),
-            )
-        return self._user_info
-
-    async def get_profile(self, other_parameter: dict[str, str | bytes] | None = None) -> UserProfile:
-        """获取用户数据"""
-        if other_parameter is None:
-            other_parameter = {}
-        params = (URL('') % dict(sorted(other_parameter.items()))).human_repr()
-        if self._user_profile.get(params) is None:
-            raw_user_profile = await request.failover_request(
-                [
-                    i / 'getProfile' % {'id': self.teaid or cast(str, self.user_name), **other_parameter}
-                    for i in BASE_URL
-                ],
-                failover_code=[502],
-                failover_exc=(TimeoutException,),
-            )
-            self._user_profile[params] = type_validate_json(UserProfile, raw_user_profile)
-            await anti_duplicate_add(
-                TOSHistoricalData(
-                    user_unique_identifier=(await self.user).unique_identifier,
-                    api_type='User Profile',
-                    data=self._user_profile[params],
-                    update_time=datetime.now(UTC),
-                ),
-            )
-        return self._user_profile[params]

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

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

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

@@ -1,89 +0,0 @@
-from datetime import datetime
-from typing import Literal
-
-from pydantic import BaseModel, Field
-
-
-class PeriodMatch(BaseModel):
-    name: str
-    teaid: str = Field(..., alias='teaId')
-    rating: str
-    rd: str
-    start_time: datetime = Field(..., alias='startTime')
-    end_time: datetime = Field(..., alias='endTime')
-    win: str
-    lose: str
-    score: str
-
-
-class UserDataTotalItem(BaseModel):
-    time_map: str = Field(..., alias='timeMap')
-    pieces_map: str = Field(..., alias='piecesMap')
-    clear_lines_map: str = Field(..., alias='clearLinesMap')
-    attacks_map: str = Field(..., alias='attacksMap')
-    dig_map: str = Field(..., alias='digMap')
-    send_map: str = Field(..., alias='sendMap')
-    rise_map: str = Field(..., alias='riseMap')
-    offset_map: str = Field(..., alias='offsetMap')
-    receive_map: str = Field(..., alias='receiveMap')
-    games_map: str = Field(..., alias='gamesMap')
-    tetris_map: str = Field(..., alias='tetrisMap')
-    combo_map: str = Field(..., alias='comboMap')
-    tspin_map: str = Field(..., alias='tspinMap')
-    b2b_map: str = Field(..., alias='b2bMap')
-    perfect_clear_map: str = Field(..., alias='perfectClearMap')
-    time_no_map: str = Field(..., alias='timeNoMap')
-    pieces_no_map: str = Field(..., alias='piecesNoMap')
-    clear_lines_no_map: str = Field(..., alias='clearLinesNoMap')
-    attacks_no_map: str = Field(..., alias='attacksNoMap')
-    dig_no_map: str = Field(..., alias='digNoMap')
-    send_no_map: str = Field(..., alias='sendNoMap')
-    rise_no_map: str = Field(..., alias='riseNoMap')
-    offset_no_map: str = Field(..., alias='offsetNoMap')
-    receive_no_map: str = Field(..., alias='receiveNoMap')
-    games_no_map: str = Field(..., alias='gamesNoMap')
-    tetris_no_map: str = Field(..., alias='tetrisNoMap')
-    combo_no_map: str = Field(..., alias='comboNoMap')
-    tspin_no_map: str = Field(..., alias='tspinNoMap')
-    b2b_no_map: str = Field(..., alias='b2bNoMap')
-    perfect_clear_no_map: str = Field(..., alias='perfectClearNoMap')
-
-
-class Data(BaseModel):
-    teaid: str = Field(..., alias='teaId')
-    name: str
-    total_exp: str = Field(..., alias='totalExp')
-    ranking: str
-    ranked_games: str = Field(..., alias='rankedGames')
-    rating_now: str = Field(..., alias='ratingNow')
-    rd_now: str = Field(..., alias='rdNow')
-    vol_now: str = Field(..., alias='volNow')
-    rating_last: str = Field(..., alias='ratingLast')
-    rd_last: str = Field(..., alias='rdLast')
-    vol_last: str = Field(..., alias='volLast')
-    period_matches: list[PeriodMatch] = Field(..., alias='periodMatches')
-    user_data_total: list[UserDataTotalItem] = Field(..., alias='userDataTotal')
-    ranking_items: str = Field(..., alias='rankingItems')
-    ranking_game_items: str = Field(..., alias='rankingGameItems')
-    training_level: str = Field(..., alias='trainingLevel')
-    training_wins: str = Field(..., alias='trainingWins')
-    pb_sprint: str = Field(..., alias='PBSprint')
-    pb_marathon: str = Field(..., alias='PBMarathon')
-    pb_challenge: str = Field(..., alias='PBChallenge')
-    register_date: datetime = Field(..., alias='registerDate')
-    last_login_date: datetime = Field(..., alias='lastLoginDate')
-
-
-class UserInfoSuccess(BaseModel):
-    code: int
-    success: Literal[True]
-    data: Data
-
-
-class FailedModel(BaseModel):
-    code: int
-    success: Literal[False]
-    error: str
-
-
-UserInfo = UserInfoSuccess | FailedModel

nonebot_plugin_tetris_stats/games/tos/api/schemas/user_profile.py

@@ -1,34 +0,0 @@
-from datetime import datetime
-from typing import Literal
-
-from pydantic import BaseModel
-
-
-class Data(BaseModel):
-    idmultiplayergameresult: int
-    iduser: str
-    teaid: str
-    time: int
-    clear_lines: int
-    attack: int
-    send: int
-    offset: int
-    receive: int
-    rise: int
-    dig: int
-    pieces: int
-    max_combo: int
-    pc_count: int
-    place: int
-    num_players: int
-    fumen_code: Literal['0', '1']  # wtf
-    rule_set: str
-    garbage: str
-    idmultiplayergame: int
-    datetime: datetime
-
-
-class UserProfile(BaseModel):
-    code: int
-    success: bool
-    data: list[Data]

nonebot_plugin_tetris_stats/games/tos/bind.py

@@ -1,62 +0,0 @@
-from nonebot_plugin_alconna.uniseg import UniMessage
-from nonebot_plugin_orm import get_session
-from nonebot_plugin_session import EventSession
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import User
-from nonebot_plugin_userinfo import BotUserInfo, EventUserInfo, UserInfo
-
-from ...db import BindStatus, create_or_update_bind, trigger
-from ...utils.host import HostPage, get_self_netloc
-from ...utils.image import get_avatar
-from ...utils.render import Bind, render
-from ...utils.render.schemas.base import People
-from ...utils.screenshot import screenshot
-from . import alc
-from .api import Player
-from .constant import GAME_TYPE
-
-
-@alc.assign('TOS.bind')
-async def _(
-    nb_user: User,
-    account: Player,
-    event_session: EventSession,
-    event_user_info: UserInfo = EventUserInfo(),  # noqa: B008
-    bot_info: UserInfo = BotUserInfo(),  # noqa: B008
-):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='bind',
-        command_args=[],
-    ):
-        user = await account.user
-        async with get_session() as session:
-            bind_status = await create_or_update_bind(
-                session=session,
-                user=nb_user,
-                game_platform=GAME_TYPE,
-                game_account=user.unique_identifier,
-            )
-        user_info = await account.get_info()
-        if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
-            async with HostPage(
-                await render(
-                    'v1/binding',
-                    Bind(
-                        platform=GAME_TYPE,
-                        status='unknown',
-                        user=People(
-                            avatar=await get_avatar(event_user_info, 'Data URI', None), name=user_info.data.name
-                        ),
-                        bot=People(
-                            avatar=await get_avatar(bot_info, 'Data URI', '../../static/logo/logo.svg'),
-                            name=bot_info.user_remark or bot_info.user_displayname or bot_info.user_name,
-                        ),
-                        command='茶服查我',
-                    ),
-                )
-            ) as page_hash:
-                await UniMessage.image(
-                    raw=await screenshot(f'http://{get_self_netloc()}/host/{page_hash}.html')
-                ).finish()

nonebot_plugin_tetris_stats/games/tos/constant.py

@@ -1,15 +0,0 @@
-from re import compile
-from typing import Literal
-
-from yarl import URL
-
-GAME_TYPE: Literal['TOS'] = 'TOS'
-
-BASE_URL = {
-    URL('https://teatube.cn:8888/'),
-    URL('http://cafuuchino1.studio26f.org:19970'),
-}
-
-USER_NAME = compile(
-    r'^(?!\.)(?!com[0-9]$)(?!con$)(?!lpt[0-9]$)(?!nul$)(?!prn$)[^\-][^\+][^\|\*\?\\\s\!:<>/$"]*[^\.\|\*\?\\\s\!:<>/$"]+$'
-)

nonebot_plugin_tetris_stats/games/tos/query.py

@@ -1,272 +0,0 @@
-from asyncio import gather
-from datetime import timedelta
-from http import HTTPStatus
-from typing import Literal, NamedTuple
-
-from nonebot.adapters import 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
-from nonebot_plugin_session_orm import get_session_persist_id  # type: ignore[import-untyped]
-from nonebot_plugin_user import get_user
-from nonebot_plugin_userinfo import EventUserInfo, UserInfo
-
-from ...db import query_bind_info, trigger
-from ...utils.exception import RequestError
-from ...utils.host import HostPage, get_self_netloc
-from ...utils.image import get_avatar
-from ...utils.metrics import TetrisMetricsProWithLPMADPM, get_metrics
-from ...utils.render import render
-from ...utils.render.avatar import get_avatar as get_random_avatar
-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
-
-
-def add_special_handlers(
-    teaid_prefix: Literal['onebot-', 'kook-', 'discord-', 'qqguild-'], match_event: type[Event]
-) -> None:
-    @alc.assign('TOS.query')
-    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),
-                game_platform=GAME_TYPE,
-                command_type='query',
-                command_args=[],
-            ):
-                player = Player(
-                    teaid=f'{teaid_prefix}{target.target}'
-                    if isinstance(target, At)
-                    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:
-                        await UniMessage.image(
-                            raw=await make_query_image(
-                                user_info, game_data, None if isinstance(target, At) else event_user_info
-                            )
-                        ).finish()
-                    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:
-    from nonebot.adapters.onebot.v11 import MessageEvent as OB11MessageEvent
-
-    add_special_handlers('onebot-', OB11MessageEvent)
-except ImportError:
-    pass
-
-try:
-    from nonebot.adapters.qq.event import GuildMessageEvent as QQGuildMessageEvent
-    from nonebot.adapters.qq.event import QQMessageEvent
-
-    add_special_handlers('qqguild-', QQGuildMessageEvent)
-    add_special_handlers('onebot-', QQMessageEvent)
-except ImportError:
-    pass
-
-try:
-    from nonebot.adapters.kaiheila.event import MessageEvent as KookMessageEvent
-
-    add_special_handlers('kook-', KookMessageEvent)
-except ImportError:
-    pass
-
-try:
-    from nonebot.adapters.discord import MessageEvent as DiscordMessageEvent
-
-    add_special_handlers('discord-', DiscordMessageEvent)
-except ImportError:
-    pass
-
-
-@alc.assign('TOS.query')
-async def _(
-    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,
-        command_type='query',
-        command_args=[],
-    ):
-        async with get_session() as session:
-            bind = await query_bind_info(
-                session=session,
-                user=await get_user(
-                    event_session.platform, target.target if isinstance(target, At) else event.get_user_id()
-                ),
-                game_platform=GAME_TYPE,
-            )
-        if bind is None:
-            await matcher.finish('未查询到绑定信息')
-        message = CANT_VERIFY_MESSAGE
-        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, None if isinstance(target, At) else event_user_info
-                    )
-                )
-            ).finish()
-        await (message + make_query_text(user_info, game_data)).finish()
-
-
-@alc.assign('TOS.query')
-async def _(account: Player, event_session: EventSession):
-    async with trigger(
-        session_persist_id=await get_session_persist_id(event_session),
-        game_platform=GAME_TYPE,
-        command_type='query',
-        command_args=[],
-    ):
-        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, None)).finish()
-        await make_query_text(user_info, game_data).finish()
-
-
-class GameData(NamedTuple):
-    game_num: int
-    metrics: TetrisMetricsProWithLPMADPM
-    OR: Number
-    dspp: Number
-    ge: Number
-
-
-async def get_game_data(player: Player, query_num: int = 50) -> GameData | None:
-    """获取游戏数据"""
-    user_profile = await player.get_profile()
-    if user_profile.data == []:
-        return None
-    weighted_total_lpm = weighted_total_apm = weighted_total_adpm = total_time = 0.0
-    total_attack = total_dig = total_offset = total_pieses = total_receive = num = 0
-    for i in user_profile.data:
-        # 排除单人局和时间为0的游戏
-        # 茶: 不计算没挖掘的局, 即使apm和lpm也如此
-        if i.num_players == 1 or i.time == 0 or i.dig is None:
-            continue
-        # 加权计算
-        time = i.time / 1000
-        lpm = 24 * (i.pieces / time)
-        apm = (i.attack / time) * 60
-        adpm = ((i.attack + i.dig) / time) * 60
-        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:
-            break
-    if num == 0:
-        return None
-    # TODO)) 如果有效局数小于 {查询数} , 并且没有无dig信息的局, 且 user_profile.data 内有{请求数}个局, 则继续往前获取信息
-    metrics = get_metrics(
-        lpm=weighted_total_lpm / total_time, apm=weighted_total_apm / total_time, adpm=weighted_total_adpm / total_time
-    )
-    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_image(user_info: UserInfoSuccess, game_data: GameData, event_user_info: UserInfo | None) -> bytes:
-    metrics = game_data.metrics
-    sprint_value = (
-        (
-            f'{duration:.3f}s'
-            if (duration := timedelta(milliseconds=float(user_info.data.pb_sprint)).total_seconds()) < 60  # noqa: PLR2004
-            else f'{duration // 60:.0f}m {duration % 60:.3f}s'
-        )
-        if user_info.data.pb_sprint != '2147483647'
-        else 'N/A'
-    )
-    async with HostPage(
-        await render(
-            'v1/tos/info',
-            Info(
-                user=People(
-                    avatar=await get_avatar(event_user_info, 'Data URI', None)
-                    if event_user_info is not None
-                    else get_random_avatar(user_info.data.teaid),
-                    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(f'http://{get_self_netloc()}/host/{page_hash}.html')
-
-
-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':
-        message += '暂无段位统计数据'
-    else:
-        message += f', 段位分 {round(float(user_data.rating_now),2)}±{round(float(user_data.rd_now),2)} ({round(float(user_data.vol_now),2)}) '
-    if game_data is None:
-        message += ', 暂无游戏数据'
-    else:
-        message += f', 最近 {game_data.game_num} 局数据'
-        message += f"\nL'PM: {game_data.metrics.lpm} ( {game_data.metrics.pps} pps )"
-        message += f'\nAPM: {game_data.metrics.apm} ( x{game_data.metrics.apl} )'
-        message += f'\nADPM: {game_data.metrics.adpm} ( x{game_data.metrics.adpl} ) ( {game_data.metrics.vs}vs )'
-    message += f'\n40L: {float(user_data.pb_sprint)/1000:.2f}s' if user_data.pb_sprint != '2147483647' else ''
-    message += f'\nMarathon: {user_data.pb_marathon}' if user_data.pb_marathon != '0' else ''
-    message += f'\nChallenge: {user_data.pb_challenge}' if user_data.pb_challenge != '0' else ''
-    return UniMessage(message)

nonebot_plugin_tetris_stats/utils/browser.py

@@ -1,7 +1,6 @@
 import sys
 from os import environ
 from platform import system
-from re import sub
 
 from nonebot import get_driver
 from nonebot.log import logger
@@ -15,12 +14,12 @@ global_config = driver.config
 
 @driver.on_startup
 async def _():
-    await BrowserManager.init_playwright()
+    await BrowserManager._init_playwright()
 
 
 @driver.on_shutdown
 async def _():
-    await BrowserManager.close_browser()
+    await BrowserManager._close_browser()
 
 
 class BrowserManager:
@@ -29,44 +28,50 @@ class BrowserManager:
     _browser: Browser | None = None
 
     @classmethod
-    async def init_playwright(cls) -> None:
+    async def _init_playwright(cls) -> None:
         if system() == 'Windows' and getattr(global_config, 'fastapi_reload', False):
-            msg = '加载失败, Windows 必须设置 FASTAPI_RELOAD=false 才能正常运行 playwright'
+            raise ImportError('加载失败, Windows 必须设置 FASTAPI_RELOAD=false 才能正常运行 playwright')
-            raise ImportError(msg)
         logger.info('开始 安装/更新 playwright 浏览器')
         environ['PLAYWRIGHT_DOWNLOAD_HOST'] = 'https://npmmirror.com/mirrors/playwright/'
-        if cls._call_playwright(['', 'install', 'firefox']):
+        if cls._handle_error(cls._call_playwright(['', 'install', 'firefox'])):
             logger.success('安装/更新 playwright 浏览器成功')
         else:
             logger.warning('playwright 浏览器 安装/更新 失败, 尝试使用原始仓库下载')
             del environ['PLAYWRIGHT_DOWNLOAD_HOST']
-            if cls._call_playwright(['', 'install', 'firefox']):
+            if cls._handle_error(cls._call_playwright(['', 'install', 'firefox'])):
                 logger.success('安装/更新 playwright 浏览器成功')
             else:
                 logger.error('安装/更新 playwright 浏览器失败')
         try:
             await cls._start_browser()
-        except BaseException as e:  # 不知道会有什么异常, 交给用户解决
+        except BaseException as e:  # noqa: BLE001 不知道会有什么异常, 交给用户解决
-            msg = 'playwright 启动失败, 请尝试在命令行运行 playwright install-deps firefox, 如果仍然启动失败, 请参考上面的报错👆'
+            raise ImportError(
-            raise ImportError(msg) from e
+                'playwright 启动失败, 请尝试在命令行运行 playwright install-deps firefox, 如果仍然启动失败, 请参考上面的报错👆'
+            ) from e
         else:
             logger.success('playwright 启动成功')
 
     @classmethod
-    def _call_playwright(cls, argv: list[str]) -> bool:
+    def _call_playwright(cls, argv: list[str]) -> BaseException:
         """等价于调用 playwright 的命令行程序"""
         argv_backup = sys.argv.copy()
+        from re import sub
+
         sys.argv[0] = sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
         sys.argv = argv
         try:
             main()
-        except SystemExit as e:
+        except BaseException as e:  # noqa: BLE001 不在这里处理 playwright 的异常
-            return e.code == 0
+            return e
-        except BaseException:  # noqa: BLE001
-            return False
         finally:
             sys.argv = argv_backup
-        return True
+        return SystemExit(0)
+
+    @classmethod
+    def _handle_error(cls, error: BaseException) -> bool:
+        if isinstance(error, SystemExit) and error.code == 0:
+            return True
+        return False
 
     @classmethod
     async def _start_browser(cls) -> Browser:
@@ -81,7 +86,7 @@ class BrowserManager:
         return cls._browser or await cls._start_browser()
 
     @classmethod
-    async def close_browser(cls) -> None:
+    async def _close_browser(cls) -> None:
         """关闭浏览器实例"""
         if isinstance(cls._browser, Browser):
             await cls._browser.close()