From 558edff5972eaf062c2b1aca2a75010cbdfd18a0 Mon Sep 17 00:00:00 2001
From: Pea Tyczynska <pea.tyczynska@digital.cabinet-office.gov.uk>
Date: Fri, 31 Jan 2020 10:57:56 +0000
Subject: [PATCH] Populate email_access_validated_at_column and make it
 non-nullable

---
 .../versions/0314_populate_email_access.py    | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 migrations/versions/0314_populate_email_access.py

diff --git a/migrations/versions/0314_populate_email_access.py b/migrations/versions/0314_populate_email_access.py
new file mode 100644
index 000000000..6f1f17ad7
--- /dev/null
+++ b/migrations/versions/0314_populate_email_access.py
@@ -0,0 +1,43 @@
+"""
+
+Revision ID: 0314_populate_email_access
+Revises: 0313_email_access_validated_at
+Create Date: 2020-01-31 10:35:44.524606
+
+"""
+from alembic import op
+
+
+revision = '0314_populate_email_access'
+down_revision = '0313_email_access_validated_at'
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    # if user has email_auth, set email_access_validated_at on last login, else set it at user created_at date.
+    op.execute("""
+        UPDATE
+            users
+        SET
+            email_access_validated_at = created_at
+        WHERE
+            email_access_validated_at IS NULL
+    """)
+    op.execute("""
+        UPDATE
+            users
+        SET
+            email_access_validated_at = logged_in_at
+        WHERE
+            auth_type = 'email_auth'
+        AND
+            logged_in_at IS NOT NULL
+    """)
+    op.alter_column('users', 'email_access_validated_at', nullable=False)
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('users', 'email_access_validated_at', nullable=True)
+    # ### end Alembic commands ###