alembic migration fix
This commit is contained in:
+9
-5
@@ -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",
|
||||||
Reference in New Issue
Block a user