Files
notifications-api/migrations/versions/0048_job_scheduled_time.py
Cliff Hill 1157f5639d black, isort, flake8
Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
2023-12-08 21:43:52 -05:00

45 lines
1.6 KiB
Python

"""empty message
Revision ID: 0048_job_scheduled_time
Revises: 0046_organisations_and_branding
Create Date: 2016-08-24 13:21:51.744526
"""
# revision identifiers, used by Alembic.
revision = "0048_job_scheduled_time"
down_revision = "0046_organisations_and_branding"
import sqlalchemy as sa
from alembic import op
def upgrade():
op.create_table(
"job_status",
sa.Column("name", sa.String(length=255), nullable=False),
sa.PrimaryKeyConstraint("name"),
)
op.add_column("jobs", sa.Column("job_status", sa.String(length=255), nullable=True))
op.add_column("jobs", sa.Column("scheduled_for", sa.DateTime(), nullable=True))
op.create_index(op.f("ix_jobs_job_status"), "jobs", ["job_status"], unique=False)
op.create_index(
op.f("ix_jobs_scheduled_for"), "jobs", ["scheduled_for"], unique=False
)
op.create_foreign_key(None, "jobs", "job_status", ["job_status"], ["name"])
op.execute("insert into job_status values ('pending')")
op.execute("insert into job_status values ('in progress')")
op.execute("insert into job_status values ('finished')")
op.execute("insert into job_status values ('sending limits exceeded')")
op.execute("insert into job_status values ('scheduled')")
def downgrade():
op.drop_constraint("jobs_job_status_fkey", "jobs", type_="foreignkey")
op.drop_index(op.f("ix_jobs_scheduled_for"), table_name="jobs")
op.drop_index(op.f("ix_jobs_job_status"), table_name="jobs")
op.drop_column("jobs", "scheduled_for")
op.drop_column("jobs", "job_status")
op.drop_table("job_status")