Merge pull request #3462 from alphagov/nhs-brand-migration

Migration to make NHS email branding ID consistent across environments

migrations/versions/0365_add_nhs_branding.py

@@ -0,0 +1,58 @@
+"""
+
+Revision ID: 0365_add_nhs_branding
+Revises: 0364_drop_old_column
+Create Date: 2022-02-17 16:31:21.415065
+
+"""
+import os
+
+from alembic import op
+
+revision = '0365_add_nhs_branding'
+down_revision = '0364_drop_old_column'
+
+environment = os.environ['NOTIFY_ENVIRONMENT']
+
+
+def upgrade():
+    if environment not in ["live", "production"]:
+        op.execute("""
+            DELETE FROM service_email_branding
+            WHERE email_branding_id in (
+                SELECT id
+                FROM email_branding
+                WHERE name = 'NHS'
+            )
+        """)
+
+        op.execute("""
+            UPDATE organisation SET email_branding_id = null
+            WHERE email_branding_id in(
+                SELECT id
+                FROM email_branding
+                WHERE name = 'NHS'
+            )
+        """)
+
+        op.execute("""
+            DELETE FROM email_branding WHERE name = 'NHS'
+        """)
+
+        op.execute("""
+            INSERT INTO email_branding (
+                id, logo, name, brand_type
+            )
+            VALUES (
+                'a7dc4e56-660b-4db7-8cff-12c37b12b5ea',
+                '1ac6f483-3105-4c9e-9017-dd7fb2752c44-nhs-blue_x2.png',
+                'NHS',
+                'org'
+            )
+        """)
+
+
+def downgrade():
+    """
+    No downgrade step since this is not fully reversible, but won't be run in production.
+    """