notifications-api/migrations/versions/0134_add_email_2fa_template_.py

"""

Revision ID: 0134_add_email_2fa_template
Revises: 0133_set_services_sms_prefix
Create Date: 2017-11-03 13:52:59.715203

"""
from datetime import datetime

from alembic import op
from flask import current_app


revision = '0134_add_email_2fa_template'
down_revision = '0133_set_services_sms_prefix'

template_id = '299726d2-dba6-42b8-8209-30e1d66ea164'


def upgrade():
    template_insert = """
        INSERT INTO templates (id, name, template_type, created_at, content, archived, service_id, subject, created_by_id, version, process_type)
        VALUES ('{}', '{}', '{}', '{}', '{}', False, '{}', '{}', '{}', 1, '{}')
    """
    template_history_insert = """
        INSERT INTO templates_history (id, name, template_type, created_at, content, archived, service_id, subject, created_by_id, version, process_type)
        VALUES ('{}', '{}', '{}', '{}', '{}', False, '{}', '{}', '{}', 1, '{}')
    """

    template_content = '\n'.join([
        'Hi ((name)),',
        '',
        'To sign in to GOV.​UK Notify please open this link:',
        '((url))',
    ])

    template_name = "Notify email verify code"
    template_subject = 'Sign in to GOV.UK Notify'

    op.execute(
        template_history_insert.format(
            template_id,
            template_name,
            'email',
            datetime.utcnow(),
            template_content,
            current_app.config['NOTIFY_SERVICE_ID'],
            template_subject,
            current_app.config['NOTIFY_USER_ID'],
            'normal'
        )
    )

    op.execute(
        template_insert.format(
            template_id,
            template_name,
            'email',
            datetime.utcnow(),
            template_content,
            current_app.config['NOTIFY_SERVICE_ID'],
            template_subject,
            current_app.config['NOTIFY_USER_ID'],
            'normal'
        )
    )

# If you are copying this migration, please remember about an insert to TemplateRedacted,
# which was not originally included here either by mistake or because it was before TemplateRedacted existed
    # op.execute(
    #     """
    #         INSERT INTO template_redacted (template_id, redact_personalisation, updated_at, updated_by_id)
    #         VALUES ('{}', '{}', '{}', '{}')
    #         ;
    #     """.format(template_id, False, datetime.utcnow(), current_app.config['NOTIFY_USER_ID'])
    # )


def downgrade():
    op.execute("DELETE FROM notifications WHERE template_id = '{}'".format(template_id))
    op.execute("DELETE FROM notification_history WHERE template_id = '{}'".format(template_id))
    op.execute("DELETE FROM template_redacted WHERE template_id = '{}'".format(template_id))
    op.execute("DELETE FROM templates_history WHERE id = '{}'".format(template_id))
    op.execute("DELETE FROM templates WHERE id = '{}'".format(template_id))