36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
"""Persist upload activity provenance."""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision: str = "20260806_0013"
|
|
down_revision: str | Sequence[str] | None = "20260806_0012"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
with op.batch_alter_table("upload_tasks") as batch:
|
|
batch.add_column(sa.Column("usage_profile_id", sa.String(length=36), nullable=True))
|
|
batch.add_column(
|
|
sa.Column("origin", sa.String(length=20), nullable=False, server_default="user")
|
|
)
|
|
batch.create_index("ix_upload_tasks_usage_profile_id", ["usage_profile_id"])
|
|
batch.create_foreign_key(
|
|
"fk_upload_tasks_usage_profile_id",
|
|
"usage_profiles",
|
|
["usage_profile_id"],
|
|
["id"],
|
|
ondelete="SET NULL",
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
with op.batch_alter_table("upload_tasks") as batch:
|
|
batch.drop_constraint("fk_upload_tasks_usage_profile_id", type_="foreignkey")
|
|
batch.drop_index("ix_upload_tasks_usage_profile_id")
|
|
batch.drop_column("origin")
|
|
batch.drop_column("usage_profile_id")
|