baozaotumao2025
28 lines
827 B
Python
28 lines
827 B
Python
"""persist idempotency keys for run creation
|
|
|
|
Revision ID: 0002
|
|
Revises: 0001
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision = "0002"
|
|
down_revision = "0001"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
with op.batch_alter_table("test_runs") as batch:
|
|
batch.add_column(sa.Column("idempotency_key", sa.String(length=128), nullable=True))
|
|
batch.add_column(sa.Column("request_fingerprint", sa.String(length=128), nullable=False, server_default=""))
|
|
batch.create_unique_constraint("uq_test_runs_idempotency_key", ["idempotency_key"])
|
|
|
|
|
|
def downgrade():
|
|
with op.batch_alter_table("test_runs") as batch:
|
|
batch.drop_constraint("uq_test_runs_idempotency_key", type_="unique")
|
|
batch.drop_column("request_fingerprint")
|
|
batch.drop_column("idempotency_key")
|