mirror of
https://github.com/GSA/notifications-api.git
synced 2026-07-05 08:57:22 -04:00
57 lines
1.1 KiB
Python
57 lines
1.1 KiB
Python
"""
|
|
|
|
Revision ID: 0140_sms_prefix_non_nullable
|
|
Revises: 0139_migrate_sms_allowance_data
|
|
Create Date: 2017-11-07 13:04:04.077142
|
|
|
|
"""
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from flask import current_app
|
|
from sqlalchemy import text
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision = "0140_sms_prefix_non_nullable"
|
|
down_revision = "0139_migrate_sms_allowance_data"
|
|
|
|
|
|
def upgrade():
|
|
conn = op.get_bind()
|
|
input_params = {"id": current_app.config["NOTIFY_SERVICE_ID"]}
|
|
conn.execute(
|
|
text("""
|
|
update services
|
|
set prefix_sms = false
|
|
where id = :id
|
|
"""),
|
|
input_params,
|
|
)
|
|
|
|
op.alter_column(
|
|
"services",
|
|
"prefix_sms",
|
|
existing_type=sa.BOOLEAN(),
|
|
nullable=False,
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
op.alter_column(
|
|
"services",
|
|
"prefix_sms",
|
|
existing_type=sa.BOOLEAN(),
|
|
nullable=True,
|
|
)
|
|
|
|
conn = op.get_bind()
|
|
input_params = {"id": current_app.config["NOTIFY_SERVICE_ID"]}
|
|
conn.execute(
|
|
text("""
|
|
update services
|
|
set prefix_sms = null
|
|
where id = :id
|
|
"""),
|
|
input_params,
|
|
)
|