Merge pull request #2320 from alphagov/letter_branding_data_migration

Letter branding data migration
This commit is contained in:
Leo Hemsted
2019-02-07 14:36:35 +00:00
committed by GitHub

View File

@@ -0,0 +1,31 @@
"""empty message
Revision ID: 0257_letter_branding_migration
Revises: 0256_set_postage_tmplt_hstr
"""
# revision identifiers, used by Alembic.
revision = '0257_letter_branding_migration'
down_revision = '0256_set_postage_tmplt_hstr'
from alembic import op
def upgrade():
op.execute("""INSERT INTO letter_branding (id, name, filename, domain)
SELECT uuid_in(md5(random()::text)::cstring), name, filename, null
from dvla_organisation""")
op.execute("""INSERT INTO service_letter_branding (service_id, letter_branding_id)
SELECT S.id, LB.id
FROM services s
JOIN dvla_organisation d on (s.dvla_organisation_id = d.id)
JOIN letter_branding lb on (lb.filename = d.filename)
WHERE d.id != '001'
""")
def downgrade():
op.execute('delete from service_letter_branding')
op.execute('delete from letter_branding')