♻️ Refactor (#318)

* 🔧 启用一些 ruff 的新规则

* ➕ 添加开发依赖 nonebot-adapter-qq

* ➕ 添加依赖 nonebot-plugin-session

* ➕ 添加依赖 nonebot-plugin-session-orm

* 🔧 忽略 ruff 规则 ISC001
format 冲突风险

* 🚨 修复 ruff 警报

* ♻️ 重构！

* ♻️ 恢复定时获取 TetraLeague 数据的功能

* ✨ 统一处理需要捕获的错误

* ✨ 记录用户触发的指令

nonebot_plugin_tetris_stats/db/__init__.py

@@ -1,10 +1,23 @@
+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_plugin_orm import AsyncSession
+from nonebot.exception import FinishedException
+from nonebot.log import logger
+from nonebot_plugin_orm import AsyncSession, get_session
 from sqlalchemy import select
 
-from ..utils.typing import GameType
-from .models import Bind
+from ..utils.typing import CommandType, GameType
+from .models import Bind, TriggerHistoricalData
+
+UTC = timezone.utc
+
+if TYPE_CHECKING:
+    from ..game_data_processor.io_data_processor.api.models import TETRIOHistoricalData
+    from ..game_data_processor.top_data_processor.api.models import TOPHistoricalData
+    from ..game_data_processor.tos_data_processor.api.models import TOSHistoricalData
 
 
 class BindStatus(Enum):
@@ -55,3 +68,75 @@ async def create_or_update_bind(
         message = BindStatus.UPDATE
     await session.commit()
     return message
+
+
+T = TypeVar('T', 'TETRIOHistoricalData', 'TOPHistoricalData', 'TOSHistoricalData')
+
+
+async def anti_duplicate_add(cls: type[T], model: T) -> None:
+    async with get_session() as session:
+        result = (
+            await session.scalars(
+                select(cls)
+                .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: CommandType | Literal['rank'],
+    command_args: list[str],
+) -> AsyncGenerator:
+    yield
+
+
+@asynccontextmanager
+@overload
+async def trigger(
+    session_persist_id: int,
+    game_platform: GameType,
+    command_type: CommandType,
+    command_args: list[str],
+) -> AsyncGenerator:
+    yield
+
+
+@asynccontextmanager
+async def trigger(
+    session_persist_id: int,
+    game_platform: GameType,
+    command_type: CommandType | Literal['rank'],
+    command_args: list[str],
+) -> AsyncGenerator:
+    logger.debug('running')
+    trigger_time = datetime.now(UTC)
+    try:
+        yield
+    except FinishedException:
+        logger.debug('yield')
+        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