style: 统一 Ruff 代码格式
This commit is contained in:
@@ -38,7 +38,9 @@ def _add_column_if_missing(bind, table_name: str, column: sa.Column) -> None:
|
||||
op.add_column(table_name, column)
|
||||
|
||||
|
||||
def _create_index_if_missing(bind, name: str, table_name: str, columns: list[str], unique: bool = False) -> None:
|
||||
def _create_index_if_missing(
|
||||
bind, name: str, table_name: str, columns: list[str], unique: bool = False
|
||||
) -> None:
|
||||
if name not in _indexes(bind, table_name):
|
||||
op.create_index(name, table_name, columns, unique=unique)
|
||||
|
||||
@@ -46,16 +48,40 @@ def _create_index_if_missing(bind, name: str, table_name: str, columns: list[str
|
||||
def upgrade() -> None:
|
||||
bind = op.get_bind()
|
||||
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("uid", sa.String(length=32), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("nickname", sa.String(length=128), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("points", sa.Integer(), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("game_name", sa.String(length=128), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("game_channel", sa.String(length=128), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("gold_balance", sa.Integer(), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("exchange_balance", sa.Integer(), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("bind_status", sa.String(length=32), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("change_role_wait_time", sa.Integer(), nullable=True))
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("updated_at", sa.DateTime(), nullable=True))
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("uid", sa.String(length=32), nullable=True)
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("nickname", sa.String(length=128), nullable=True)
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("points", sa.Integer(), nullable=True)
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("game_name", sa.String(length=128), nullable=True)
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind,
|
||||
"accounts",
|
||||
sa.Column("game_channel", sa.String(length=128), nullable=True),
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("gold_balance", sa.Integer(), nullable=True)
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("exchange_balance", sa.Integer(), nullable=True)
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("bind_status", sa.String(length=32), nullable=True)
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind,
|
||||
"accounts",
|
||||
sa.Column("change_role_wait_time", sa.Integer(), nullable=True),
|
||||
)
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("updated_at", sa.DateTime(), nullable=True)
|
||||
)
|
||||
_create_index_if_missing(bind, "ix_accounts_uid", "accounts", ["uid"])
|
||||
|
||||
if not _has_table(bind, "douyu_tasks"):
|
||||
@@ -75,8 +101,12 @@ def upgrade() -> None:
|
||||
sa.ForeignKeyConstraint(["created_by"], ["users.id"]),
|
||||
sa.PrimaryKeyConstraint("id"),
|
||||
)
|
||||
_create_index_if_missing(bind, "ix_douyu_tasks_batch_id", "douyu_tasks", ["batch_id"])
|
||||
_create_index_if_missing(bind, "ix_douyu_tasks_task_type", "douyu_tasks", ["task_type"])
|
||||
_create_index_if_missing(
|
||||
bind, "ix_douyu_tasks_batch_id", "douyu_tasks", ["batch_id"]
|
||||
)
|
||||
_create_index_if_missing(
|
||||
bind, "ix_douyu_tasks_task_type", "douyu_tasks", ["task_type"]
|
||||
)
|
||||
|
||||
if not _has_table(bind, "douyu_config"):
|
||||
op.create_table(
|
||||
@@ -121,7 +151,10 @@ def downgrade() -> None:
|
||||
if _has_table(bind, "douyu_goods_snapshot"):
|
||||
indexes = _indexes(bind, "douyu_goods_snapshot")
|
||||
if "ix_douyu_goods_snapshot_commodity_id" in indexes:
|
||||
op.drop_index("ix_douyu_goods_snapshot_commodity_id", table_name="douyu_goods_snapshot")
|
||||
op.drop_index(
|
||||
"ix_douyu_goods_snapshot_commodity_id",
|
||||
table_name="douyu_goods_snapshot",
|
||||
)
|
||||
op.drop_table("douyu_goods_snapshot")
|
||||
if _has_table(bind, "douyu_config"):
|
||||
op.drop_table("douyu_config")
|
||||
|
||||
Reference in New Issue
Block a user