alembic migration fix

This commit is contained in:
Snowy-Fluffy
2026-02-12 10:17:17 +03:00
parent 799bd01a79
commit bce3d1a046
@@ -13,8 +13,9 @@ import sqlalchemy as sa
# revision identifiers, used by Alembic. # revision identifiers, used by Alembic.
revision: str = "0003_promo_current_activations_not_null" revision: str = "0003_promo_curr_act_not_null"
down_revision: Union[str, Sequence[str], None] = "0002_active_discount_expires_at" down_revision: Union[str, Sequence[str],
None] = "0002_active_discount_expires_at"
branch_labels: Union[str, Sequence[str], None] = None branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None depends_on: Union[str, Sequence[str], None] = None
@@ -41,11 +42,13 @@ def upgrade() -> None:
if not inspector.has_table("promo_codes"): if not inspector.has_table("promo_codes"):
return return
promo_columns = {column["name"] for column in inspector.get_columns("promo_codes")} promo_columns = {column["name"]
for column in inspector.get_columns("promo_codes")}
if "current_activations" not in promo_columns: if "current_activations" not in promo_columns:
op.add_column( op.add_column(
"promo_codes", "promo_codes",
sa.Column("current_activations", sa.Integer(), nullable=False, server_default=sa.text("0")), sa.Column("current_activations", sa.Integer(),
nullable=False, server_default=sa.text("0")),
) )
return return
@@ -80,7 +83,8 @@ def downgrade() -> None:
if not inspector.has_table("promo_codes"): if not inspector.has_table("promo_codes"):
return return
promo_columns = {column["name"] for column in inspector.get_columns("promo_codes")} promo_columns = {column["name"]
for column in inspector.get_columns("promo_codes")}
if "current_activations" in promo_columns: if "current_activations" in promo_columns:
op.alter_column( op.alter_column(
"promo_codes", "promo_codes",