Merge pull request #2159 from alphagov/add-dvla-org-filename

Add filename to dvla_organisation table

app/models.py

@@ -247,6 +247,7 @@ class DVLAOrganisation(db.Model):
     __tablename__ = 'dvla_organisation'
     id = db.Column(db.String, primary_key=True)
     name = db.Column(db.String(255), nullable=True)
+    filename = db.Column(db.String(255), nullable=True)
 
 
 INTERNATIONAL_SMS_TYPE = 'international_sms'

migrations/versions/0237_add_filename_to_dvla_org.py

@@ -0,0 +1,56 @@
+"""
+
+Revision ID: 0237_add_filename_to_dvla_org
+Revises: 0236_another_letter_org
+Create Date: 2018-09-28 15:39:21.115358
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.sql import text
+
+
+revision = '0237_add_filename_to_dvla_org'
+down_revision = '0236_another_letter_org'
+
+
+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')