notifications-api/migrations/versions/0298_add_mou_signed_receipt.py

"""

Revision ID: 0298_add_mou_signed_receipt
Revises: 0297_template_redacted_fix
Create Date: 2019-05-22 16:58:52.929661

"""
from alembic import op
from flask import current_app
from sqlalchemy import text

revision = '0298_add_mou_signed_receipt'
down_revision = '0297_template_redacted_fix'


templates = [
    {
        'id': '4fd2e43c-309b-4e50-8fb8-1955852d9d71',
        'name': 'MOU Signed By Receipt',
        'type': 'email',
        'subject': 'You’ve accepted the GOV.​UK Notify data sharing and financial agreement',
        'content_lines': [
            'Hi ((signed_by_name)),',
            '',
            '((org_name)) has accepted the GOV.​UK Notify data sharing and financial agreement. ',
            '',
            'If you need another copy of the agreement you can download it here: ((mou_link))',
            '',
            'If you need to add Cabinet Office as a supplier, here are the details you need:',
            '',
            'TO BE ADDED MANUALLY',
            '',
            'Thanks,',
            'GOV.​UK Notify team',
            '',
            'https://www.gov.uk/notify',
        ],
    },

    {
        'id': 'c20206d5-bf03-4002-9a90-37d5032d9e84',
        'name': 'MOU Signed On Behalf Of Receipt - Signed by',
        'type': 'email',
        'subject': 'You’ve accepted the GOV.​UK Notify data sharing and financial agreement',
        'content_lines': [
            'Hi ((signed_by_name)),',
            '',
            '((org_name)) has accepted the GOV.​UK Notify data sharing and financial agreement. We’ve emailed ((on_behalf_of_name)) to let them know too.',
            '',
            'If you need another copy of the agreement you can download it here: ((mou_link))',
            '',
            'If you need to add Cabinet Office as a supplier, here are the details you need:',
            '',
            'TO BE ADDED MANUALLY',
            '',
            'Thanks,',
            'GOV.​UK Notify team',
            '',
            'https://www.gov.uk/notify',
        ],
    },

    {
        'id': '522b6657-5ca5-4368-a294-6b527703bd0b',
        'name': 'MOU Signed On Behalf Of Receipt - On Behalf Of',
        'type': 'email',
        'subject': '((org_name)) has accepted the GOV.​UK Notify data sharing and financial agreement',
        'content_lines': [
            'Hi ((on_behalf_of_name)),',
            '',
            '((signed_by_name)) has accepted the GOV.​UK Notify data sharing and financial agreement on your behalf, for ((org_name)).',
            '',
            'GOV.​UK Notify lets teams in the public sector send emails, text messages and letters. It’s built and run by a team in the Government Digital Service (part of Cabinet Office).',
            '',
            'If you need another copy of the agreement you can download it here: ((mou_link))',
            '',
            'If you need to add Cabinet Office as a supplier, here are the details you need.',
            '',
            'TO BE ADDED MANUALLY',
            '',
            'Thanks,',
            'GOV.​UK Notify team',
            '',
            'https://www.gov.uk/notify',
        ],
    },

    {
        'id': 'd0e66c4c-0c50-43f0-94f5-f85b613202d4',
        'name': 'MOU Signed Notify Team Alert',
        'type': 'email',
        'subject': 'Someone signed an MOU for an org on Notify',
        'content_lines': [
            'What’s up Notifiers,',
            '',
            '((signed_by_name)) just accepted the data sharing and financial agreement for ((org_name)).',
            '',
            'See how ((org_name)) is using Notify here: ((org_dashboard_link))',
        ],
    },
]


def upgrade():
    insert_t = """
        INSERT INTO templates (id, name, template_type, created_at, content, archived, service_id, subject,
        created_by_id, version, process_type, hidden)
        VALUES (:template_id, :template_name, :template_type, current_timestamp, 
        :content, False, :notify_service_id, :subject, :user_id, 1, :process_type, false)
    """
    insert_th = """
            INSERT INTO templates_history (id, name, template_type, created_at, content, archived, service_id, subject,
            created_by_id, version, process_type, hidden)
        VALUES (:template_id, :template_name, :template_type, current_timestamp, 
        :content, False, :notify_service_id, :subject, :user_id, 1, :process_type, false)
        
        """

    for template in templates:
        input_params = {
            "template_id": template['id'],
            "template_name": template['name'],
            "template_type": template['type'],
            "content": '\n'.join(template['content_lines']),
            "notify_service_id": current_app.config['NOTIFY_SERVICE_ID'],
            "subject": template.get('subject'),
            "user_id": current_app.config['NOTIFY_USER_ID'],
            "process_type": 'normal'
        }
        conn = op.get_bind()

        conn.execute(
            text(insert_t), input_params
        )

        conn.execute(
            text(insert_th), input_params
        )

        input_params = {
            "template_id": template['id'],
            "user_id": current_app.config['NOTIFY_USER_ID']
        }
        conn.execute(
            text("""
            INSERT INTO template_redacted
            (
                template_id,
                redact_personalisation,
                updated_at,
                updated_by_id
            ) VALUES ( :template_id, false, current_timestamp, :user_id )
            """), input_params
        )


def downgrade():
    conn = op.get_bind()
    for template in templates:
        input_params = {
            "template_id": template['id']
        }
        conn.execute(text("DELETE FROM notifications WHERE template_id = :template_id"), input_params)
        conn.execute(text("DELETE FROM notification_history WHERE template_id = :template_id"), input_params)
        conn.execute(text("DELETE FROM template_redacted WHERE template_id = :template_id"), input_params)
        conn.execute(text("DELETE FROM templates WHERE id = :template_id"), input_params)
        conn.execute(text("DELETE FROM templates_history WHERE id = :template_id"), input_params)