laantungir/routstr-core
1
0
Fork 0
mirror of https://github.com/Routstr/routstr-core.git synced 2026-08-09 02:54:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix migration

Browse Source
This commit is contained in:
Shroominic
2025-11-28 13:45:32 +09:00
parent 9be8aba417
commit bd3aacd1f5
1 changed files with 33 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
migrations/versions/5f6ac1e4fa9f_the_refactor.py
+33 -30
View File
@@ -19,39 +19,42 @@ def upgrade() -> None:
# Rename the table
op.rename_table("api_keys", "temporary_credit")
# Perform column modifications in batch mode for SQLite support
with op.batch_alter_table("temporary_credit", schema=None) as batch_op:
batch_op.add_column(sa.Column("created", sa.DateTime(), nullable=True))
batch_op.add_column(
sa.Column("refund_expiration_time", sa.Integer(), nullable=True)
)
batch_op.drop_column("key_expiry_time")
batch_op.drop_column("total_spent")
batch_op.drop_column("total_requests")
op.add_column(
"temporary_credit", sa.Column("created", sa.DateTime(), nullable=True)
)
op.add_column(
"temporary_credit",
sa.Column("refund_expiration_time", sa.Integer(), nullable=True),
)
op.drop_column("temporary_credit", "key_expiry_time")
op.drop_column("temporary_credit", "total_spent")
op.drop_column("temporary_credit", "total_requests")
def downgrade() -> None:
# Revert column modifications
with op.batch_alter_table("temporary_credit", schema=None) as batch_op:
batch_op.add_column(
sa.Column(
"total_requests",
sa.Integer(),
nullable=False,
server_default=sa.text("0"),
)
)
batch_op.add_column(
sa.Column(
"total_spent",
sa.Integer(),
nullable=False,
server_default=sa.text("0"),
)
)
batch_op.add_column(sa.Column("key_expiry_time", sa.Integer(), nullable=True))
batch_op.drop_column("refund_expiration_time")
batch_op.drop_column("created")
op.add_column(
"temporary_credit",
sa.Column(
"total_requests",
sa.Integer(),
nullable=False,
server_default=sa.text("0"),
),
)
op.add_column(
"temporary_credit",
sa.Column(
"total_spent",
sa.Integer(),
nullable=False,
server_default=sa.text("0"),
),
)
op.add_column(
"temporary_credit", sa.Column("key_expiry_time", sa.Integer(), nullable=True)
)
op.drop_column("temporary_credit", "refund_expiration_time")
op.drop_column("temporary_credit", "created")
# Revert table rename
op.rename_table("temporary_credit", "api_keys")