darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-09 14:42:24 -05:00
Code Issues Packages Projects Releases Wiki Activity

Simpify org types migration script and introduce foreign key

Browse Source 
Don't transform org types on prod
This commit is contained in:
Pea Tyczynska
2019-07-24 16:37:23 +01:00
parent 5b256fa64e
commit 38bb2c1cf6
5 changed files with 82 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

145
migrations/versions/0300_migrate_org_types.py
View File

@@ -1,116 +1,75 @@
import os
"""
Revision ID: 0300_migrate_org_types
Revises: 0299_org_types_table
Create Date: 2019-07-19 11:13:41.286472
Create Date: 2019-07-24 16:18:27.467361
"""
from alembic import op
import sqlalchemy as sa
revision = '0300_migrate_org_types'
down_revision = '0299_org_types_table'
environment = os.environ['NOTIFY_ENVIRONMENT']
def upgrade():
op.execute("""
UPDATE
organisation
SET
organisation_type = 'nhs_local'
FROM
organisation_to_service, annual_billing
WHERE
organisation.organisation_type = 'nhs'
AND
annual_billing.service_id = organisation_to_service.service_id
AND
organisation_to_service.organisation_id = organisation.id
AND
annual_billing.free_sms_fragment_limit = 25000
""")
if environment != "production":
op.execute("""
UPDATE
organisation
SET
organisation_type = 'nhs_local'
WHERE
organisation.organisation_type = 'nhs'
""")
op.execute("""
UPDATE
services
SET
organisation_type = 'nhs_local'
FROM
annual_billing
WHERE
services.organisation_type = 'nhs'
AND
annual_billing.service_id = services.id
AND
annual_billing.free_sms_fragment_limit = 25000
""")
op.execute("""
UPDATE
services
SET
organisation_type = 'nhs_local'
WHERE
services.organisation_type = 'nhs'
""")
op.execute("""
UPDATE
organisation
SET
organisation_type = 'nhs_central'
FROM
organisation_to_service, annual_billing
WHERE
organisation.organisation_type = 'nhs'
AND
annual_billing.service_id = organisation_to_service.service_id
AND
organisation_to_service.organisation_id = organisation.id
AND
annual_billing.free_sms_fragment_limit = 250000
""")
op.alter_column('organisation_types', 'name', existing_type=sa.VARCHAR(), type_=sa.String(length=255))
op.execute("""
UPDATE
services
SET
organisation_type = 'nhs_central'
FROM
annual_billing
WHERE
services.organisation_type = 'nhs'
AND
annual_billing.service_id = services.id
AND
annual_billing.free_sms_fragment_limit = 250000
""")
op.create_foreign_key(
'organisation_organisation_type_fkey', 'organisation', 'organisation_types', ['organisation_type'], ['name']
)
op.create_foreign_key(
'services_organisation_type_fkey', 'services', 'organisation_types', ['organisation_type'], ['name']
)
def downgrade():
op.execute("""
UPDATE
organisation
SET
organisation_type = 'nhs'
WHERE
organisation_type = 'nhs_central'
""")
op.drop_constraint('services_organisation_type_fkey', 'services', type_='foreignkey')
op.execute("""
UPDATE
services
SET
organisation_type = 'nhs'
WHERE
organisation_type = 'nhs_central'
""")
op.drop_constraint('organisation_organisation_type_fkey', 'organisation', type_='foreignkey')
op.execute("""
UPDATE
organisation
SET
organisation_type = 'nhs'
WHERE
organisation_type = 'nhs_local'
""")
op.alter_column('organisation_types', 'name', existing_type=sa.String(length=255), type_=sa.VARCHAR())
op.execute("""
UPDATE
services
SET
organisation_type = 'nhs'
WHERE
organisation_type = 'nhs_local'
""")
if environment != "production":
op.execute("""
UPDATE
organisation
SET
organisation_type = 'nhs'
WHERE
organisation_type = 'nhs_local'
""")
op.execute("""
UPDATE
services
SET
organisation_type = 'nhs'
WHERE
organisation_type = 'nhs_local'
""")