From 7792c18efb46cfa03a11ee13eff9a7b2ac1bbcb6 Mon Sep 17 00:00:00 2001 From: Rebecca Law Date: Fri, 5 Jan 2018 17:34:51 +0000 Subject: [PATCH] Migration script to update the templates and templates_history (latest version only) with the service default letter contact block. --- .../versions/0156_set_temp_letter_contact.py | 38 +++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 migrations/versions/0156_set_temp_letter_contact.py diff --git a/migrations/versions/0156_set_temp_letter_contact.py b/migrations/versions/0156_set_temp_letter_contact.py new file mode 100644 index 000000000..92e848201 --- /dev/null +++ b/migrations/versions/0156_set_temp_letter_contact.py @@ -0,0 +1,38 @@ +""" + +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")