mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-09 23:02:13 -05:00
85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
"""
|
|
|
|
Revision ID: 0331_add_broadcast_org
|
|
Revises: 0330_broadcast_invite_email
|
|
Create Date: 2020-09-23 10:11:01.094412
|
|
|
|
"""
|
|
import os
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
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)
|