Files
notifications-api/migrations/versions/0257_letter_branding_migration.py
Rebecca Law 429997e778 Data migration to populate letter_branding and service_letter_branding.
Only service_letter_branding is added where the service.dvla_org_id is not 001 (hm-government)
2019-02-07 14:28:09 +00:00

32 lines
959 B
Python

"""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')