add email 2fa template

2026-02-01 07:35:34 -05:00 · 2017-11-03 14:25:58 +00:00
parent b2756ac99d
commit 57b7a392b2
1 changed files with 71 additions and 0 deletions
--- a/migrations/versions/0134_add_email_2fa_template_.py
+++ b/migrations/versions/0134_add_email_2fa_template_.py
@@ -0,0 +1,71 @@
+"""
+
+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'
+        )
+    )
+
+
+def downgrade():
+   op.execute("DELETE FROM templates_history WHERE id = '{}'".format(template_id))
+   op.execute("DELETE FROM templates WHERE id = '{}'".format(template_id))