50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
"""new request statuses
|
|
|
|
Revision ID: 77b065750596
|
|
Revises: 1f57f784ed5b
|
|
Create Date: 2018-08-07 16:42:11.502361
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.orm.session import sessionmaker
|
|
|
|
from atst.models.request_status_event import RequestStatus
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '77b065750596'
|
|
down_revision = '1f57f784ed5b'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
"""
|
|
Update all existing request statuses so that the state of the
|
|
table reflects the statuses listed in RequestStatus.
|
|
|
|
This involves fixing the casing on existing statuses, and
|
|
deleting statuses that have no match.
|
|
"""
|
|
|
|
db = op.get_bind()
|
|
|
|
status_events = db.execute("SELECT * FROM request_status_events").fetchall()
|
|
for status_event in status_events:
|
|
try:
|
|
status = RequestStatus[status_event["new_status"].upper()]
|
|
query = sa.text("""
|
|
UPDATE request_status_events
|
|
SET new_status = :status
|
|
WHERE id = :id"""
|
|
)
|
|
db.execute(query, id=status_event["id"], status=status.name)
|
|
except KeyError:
|
|
query = sa.text("DELETE FROM request_status_events WHERE id = :id")
|
|
db.execute(query, id=status_event["id"])
|
|
|
|
|
|
def downgrade():
|
|
pass
|