mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-10 23:32:27 -05:00
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
"""
|
|
|
|
Revision ID: 0156_set_temp_letter_contact
|
|
Revises: 0152_kill_service_free_fragments
|
|
Create Date: 2018-01-05 17:04:20.596271
|
|
|
|
"""
|
|
from alembic import op
|
|
|
|
revision = "0156_set_temp_letter_contact"
|
|
down_revision = "0152_kill_service_free_fragments"
|
|
|
|
|
|
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")
|