Files
notifications-api/migrations/versions/0156_set_temp_letter_contact.py

39 lines
1.0 KiB
Python

"""
Revision ID: 0156_set_temp_letter_contact
Revises: 0155_revert_0153
Create Date: 2018-01-05 17:04:20.596271
"""
from alembic import op
revision = '0156_set_temp_letter_contact'
down_revision = '0155_revert_0153'
def upgrade():
query = """
UPDATE templates t SET service_letter_contact_id = (
SELECT id from service_letter_contacts
WHERE service_id = t.service_id
and is_default = true
)
WHERE template_type = 'letter'
"""
op.execute(query)
history_query = """
UPDATE templates_history t_history set service_letter_contact_id = (
SELECT service_letter_contact_id from templates
where service_letter_contact_id is not null
and id = t_history.id
and version = t_history.version
)
"""
op.execute(history_query)
def downgrade():
op.execute("UPDATE templates set service_letter_contact_id = null")
op.execute("UPDATE templates_history set service_letter_contact_id = null")