diff --git a/nonebot_plugin_tetris_stats/db/__init__.py b/nonebot_plugin_tetris_stats/db/__init__.py index 5d208ae..76908c7 100644 --- a/nonebot_plugin_tetris_stats/db/__init__.py +++ b/nonebot_plugin_tetris_stats/db/__init__.py @@ -92,7 +92,7 @@ async def anti_duplicate_add(cls: type[T], model: T) -> None: async def trigger( session_persist_id: int, game_platform: Literal['IO'], - command_type: CommandType | Literal['rank'], + command_type: CommandType | Literal['rank', 'config', 'record'], command_args: list[str], ) -> AsyncGenerator: yield @@ -113,7 +113,7 @@ async def trigger( async def trigger( session_persist_id: int, game_platform: GameType, - command_type: CommandType | Literal['rank'], + command_type: CommandType | Literal['rank', 'config', 'record'], command_args: list[str], ) -> AsyncGenerator: trigger_time = datetime.now(UTC) diff --git a/nonebot_plugin_tetris_stats/db/models.py b/nonebot_plugin_tetris_stats/db/models.py index 3f63add..dd66195 100644 --- a/nonebot_plugin_tetris_stats/db/models.py +++ b/nonebot_plugin_tetris_stats/db/models.py @@ -76,6 +76,6 @@ class TriggerHistoricalData(MappedAsDataclass, Model): trigger_time: Mapped[datetime] = mapped_column(DateTime) session_persist_id: Mapped[int] game_platform: Mapped[GameType] = mapped_column(String(32), index=True) - command_type: Mapped[CommandType | Literal['rank']] = mapped_column(String(16), index=True) + command_type: Mapped[CommandType | Literal['rank', 'config', 'record']] = mapped_column(String(16), index=True) command_args: Mapped[list[str]] = mapped_column(JSON) finish_time: Mapped[datetime] = mapped_column(DateTime)