darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-25 08:43:55 -04:00
Code Issues Packages Projects Releases Wiki Activity

swap out org value on existing data before truncating

Browse Source
This commit is contained in:
stvnrlly
2022-10-11 20:59:37 +00:00
parent 0383a4963e
commit 47904bf0bd
1 changed files with 26 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
migrations/versions/0378_add_org_names.py
View File

@@ -17,14 +17,37 @@ def upgrade():
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
op.get_bind() op.get_bind()
op.execute("TRUNCATE TABLE organisation_types CASCADE;") # bluntly swap out data
op.execute("INSERT INTO organisation_types VALUES ('state','f','250000'),('federal','f','250000'),('other','f','250000');") op.execute("INSERT INTO organisation_types VALUES ('state','f','250000'),('federal','f','250000');")
op.execute("UPDATE services SET organisation_type = 'federal';")
op.execute("UPDATE organisation SET organisation_type = 'federal';")
op.execute("UPDATE services_history SET organisation_type = 'federal';")
# remove uk values
service_delete = """DELETE FROM organisation_types WHERE name IN
('central','local','nhs','nhs_central','nhs_local','emergency_service','school_or_college','nhs_gp')
"""
op.execute(service_delete)
# ### end Alembic commands ### # ### end Alembic commands ###
def downgrade(): def downgrade():
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
pass service_insert = """INSERT INTO organisation_types VALUES
('central','','250000')
('local','f','25000')
('nhs','','25000')
('nhs_central','t','250000')
('nhs_local','f','25000')
('emergency_service','f','25000')
('school_or_college','f','25000')
('nhs_gp','f','25000')
"""
op.execute(service_insert)
op.execute("UPDATE services SET organisation_type = 'central';")
op.execute("UPDATE organisation SET organisation_type = 'central';")
op.execute("UPDATE services_history SET organisation_type = 'central';")
op.execute("DELETE FROM organisation_types WHERE name IN ('federal','state')")
# ### end Alembic commands ### # ### end Alembic commands ###