43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
"""Add upload task metadata."""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision: str = "20260802_0002"
|
|
down_revision: str | Sequence[str] | None = "20260802_0001"
|
|
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("project_id", sa.String(36), nullable=True))
|
|
batch.add_column(sa.Column("description", sa.Text(), nullable=True))
|
|
batch.add_column(sa.Column("version", sa.String(100), nullable=True))
|
|
batch.add_column(sa.Column("purpose", sa.String(200), nullable=True))
|
|
batch.add_column(sa.Column("tags_json", sa.Text(), nullable=False, server_default="[]"))
|
|
batch.add_column(
|
|
sa.Column("overwrite_allowed", sa.Boolean(), nullable=False, server_default=sa.false())
|
|
)
|
|
batch.create_foreign_key(
|
|
"fk_upload_tasks_project_id", "projects", ["project_id"], ["id"], ondelete="SET NULL"
|
|
)
|
|
batch.create_index("ix_upload_tasks_project_id", ["project_id"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
with op.batch_alter_table("upload_tasks") as batch:
|
|
batch.drop_index("ix_upload_tasks_project_id")
|
|
batch.drop_constraint("fk_upload_tasks_project_id", type_="foreignkey")
|
|
for column in (
|
|
"overwrite_allowed",
|
|
"tags_json",
|
|
"purpose",
|
|
"version",
|
|
"description",
|
|
"project_id",
|
|
):
|
|
batch.drop_column(column)
|