diff --git a/nonebot_plugin_tetris_stats/config/migrations/2ff388a8c486_add_verify_field.py b/nonebot_plugin_tetris_stats/config/migrations/2ff388a8c486_add_verify_field.py new file mode 100644 index 0000000..3bd50bc --- /dev/null +++ b/nonebot_plugin_tetris_stats/config/migrations/2ff388a8c486_add_verify_field.py @@ -0,0 +1,42 @@ +"""add verify field + +迁移 ID: 2ff388a8c486 +父迁移: 3588702dd3a4 +创建时间: 2025-07-22 18:09:09.734164 + +""" + +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 = '2ff388a8c486' +down_revision: str | Sequence[str] | None = '3588702dd3a4' +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('nb_t_bind', schema=None) as batch_op: + batch_op.add_column(sa.Column('verify', sa.Boolean(), nullable=False, server_default='false')) + + # ### end Alembic commands ### + + +def downgrade(name: str = '') -> None: + if name: + return + # ### commands auto generated by Alembic - please adjust! ### + with op.batch_alter_table('nb_t_bind', schema=None) as batch_op: + batch_op.drop_column('verify') + + # ### end Alembic commands ### diff --git a/nonebot_plugin_tetris_stats/db/__init__.py b/nonebot_plugin_tetris_stats/db/__init__.py index 2f56be9..d739b36 100644 --- a/nonebot_plugin_tetris_stats/db/__init__.py +++ b/nonebot_plugin_tetris_stats/db/__init__.py @@ -42,6 +42,8 @@ async def create_or_update_bind( user: User, game_platform: GameType, game_account: str, + *, + verify: bool = False, ) -> BindStatus: bind = await query_bind_info( session=session, @@ -53,6 +55,7 @@ async def create_or_update_bind( user_id=user.id, game_platform=game_platform, game_account=game_account, + verify=verify, ) session.add(bind) status = BindStatus.SUCCESS diff --git a/nonebot_plugin_tetris_stats/db/models.py b/nonebot_plugin_tetris_stats/db/models.py index 6e43b4c..40ff98e 100644 --- a/nonebot_plugin_tetris_stats/db/models.py +++ b/nonebot_plugin_tetris_stats/db/models.py @@ -71,6 +71,7 @@ class Bind(MappedAsDataclass, Model): user_id: Mapped[int] = mapped_column(index=True) game_platform: Mapped[GameType] = mapped_column(String(32)) game_account: Mapped[str] + verify: Mapped[bool] class TriggerHistoricalDataV2(MappedAsDataclass, Model):