38 lines
1.6 KiB
Python
38 lines
1.6 KiB
Python
"""Official record import fields and run journal."""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "0002"
|
|
down_revision = "0001"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
status = sa.Enum("running", "success", "partial", "failed", name="importstatus")
|
|
op.add_column("catch_report", sa.Column("source_url", sa.Text()))
|
|
op.add_column("catch_report", sa.Column("source_external_id", sa.String(64)))
|
|
op.add_column("catch_report", sa.Column("raw_payload", sa.JSON()))
|
|
op.create_unique_constraint("uq_catch_report_source_external_id", "catch_report", ["source_external_id"])
|
|
op.create_table(
|
|
"official_record_import",
|
|
sa.Column("id", sa.Uuid(), primary_key=True),
|
|
sa.Column("started_at", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("finished_at", sa.DateTime(timezone=True)),
|
|
sa.Column("status", status, nullable=False),
|
|
sa.Column("source_url", sa.Text(), nullable=False),
|
|
sa.Column("rows_seen", sa.Integer(), nullable=False),
|
|
sa.Column("rows_created", sa.Integer(), nullable=False),
|
|
sa.Column("rows_updated", sa.Integer(), nullable=False),
|
|
sa.Column("error_summary", sa.Text()),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("official_record_import")
|
|
op.drop_constraint("uq_catch_report_source_external_id", "catch_report", type_="unique")
|
|
op.drop_column("catch_report", "raw_payload")
|
|
op.drop_column("catch_report", "source_external_id")
|
|
op.drop_column("catch_report", "source_url")
|
|
sa.Enum(name="importstatus").drop(op.get_bind(), checkfirst=True)
|