Merge pull request #2977 from alphagov/add-broadcast-org

add broadcast org

migrations/versions/0331_add_broadcast_org.py

@@ -0,0 +1,81 @@
+"""
+
+Revision ID: 0331_add_broadcast_org
+Revises: 0330_broadcast_invite_email
+Create Date: 2020-09-23 10:11:01.094412
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import os
+
+revision = '0331_add_broadcast_org'
+down_revision = '0330_broadcast_invite_email'
+
+environment = os.environ['NOTIFY_ENVIRONMENT']
+
+organisation_id = '38e4bf69-93b0-445d-acee-53ea53fe02df'
+
+
+def upgrade():
+    # we've already done this manually on production
+    if environment != "production":
+        insert_sql = """
+            INSERT INTO organisation
+            (
+                id,
+                name,
+                active,
+                created_at,
+                agreement_signed,
+                crown,
+                organisation_type
+            )
+            VALUES (
+                :id,
+                :name,
+                :active,
+                current_timestamp,
+                :agreement_signed,
+                :crown,
+                :organisation_type
+            )
+        """
+        update_service_set_broadcast_org_sql = """
+        UPDATE services
+        SET organisation_id = :organisation_id
+        WHERE id in (
+            SELECT service_id
+            FROM service_permissions
+            WHERE permission = 'broadcast'
+        )
+        """
+        conn = op.get_bind()
+        conn.execute(
+            sa.text(insert_sql),
+            id=organisation_id,
+            name=f'Broadcast Services ({environment})',
+            active=True,
+            agreement_signed=None,
+            crown=None,
+            organisation_type='central',
+        )
+        conn.execute(
+            sa.text(update_service_set_broadcast_org_sql),
+            organisation_id=organisation_id
+        )
+
+
+def downgrade():
+    update_service_remove_org_sql = """
+        UPDATE services
+        SET organisation_id = NULL, updated_at = current_timestamp
+        WHERE organisation_id = :organisation_id
+    """
+    delete_sql = """
+        DELETE FROM organisation
+        WHERE id = :organisation_id
+    """
+    conn = op.get_bind()
+    conn.execute(sa.text(update_service_remove_org_sql), organisation_id=organisation_id)
+    conn.execute(sa.text(delete_sql), organisation_id=organisation_id)