mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-11 07:42:20 -05:00
64 lines
1.3 KiB
Python
64 lines
1.3 KiB
Python
"""
|
|
|
|
Revision ID: 0237_add_filename_to_dvla_org
|
|
Revises: 0235_add_postage_to_pk
|
|
Create Date: 2018-09-28 15:39:21.115358
|
|
|
|
"""
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.sql import text
|
|
|
|
revision = "0237_add_filename_to_dvla_org"
|
|
down_revision = "0235_add_postage_to_pk"
|
|
|
|
|
|
LOGOS = {
|
|
"001": "hm-government",
|
|
"002": "opg",
|
|
"003": "dwp",
|
|
"004": "geo",
|
|
"005": "ch",
|
|
"006": "dwp-welsh",
|
|
"007": "dept-for-communities",
|
|
"008": "mmo",
|
|
"009": "hmpo",
|
|
"500": "hm-land-registry",
|
|
"501": "ea",
|
|
"502": "wra",
|
|
"503": "eryc",
|
|
"504": "rother",
|
|
"505": "cadw",
|
|
"506": "twfrs",
|
|
"507": "thames-valley-police",
|
|
"508": "ofgem",
|
|
"509": "hackney",
|
|
"510": "pension-wise",
|
|
"511": "nhs",
|
|
"512": "vale-of-glamorgan",
|
|
"513": "wdc",
|
|
"514": "brighton-hove",
|
|
}
|
|
|
|
|
|
def upgrade():
|
|
conn = op.get_bind()
|
|
op.add_column(
|
|
"dvla_organisation", sa.Column("filename", sa.String(length=255), nullable=True)
|
|
)
|
|
|
|
for org_id, org_filename in LOGOS.items():
|
|
conn.execute(
|
|
text(
|
|
"""
|
|
UPDATE dvla_organisation SET filename = :filename WHERE id = :id
|
|
"""
|
|
),
|
|
filename=org_filename,
|
|
id=org_id,
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
op.drop_column("dvla_organisation", "filename")
|