From b23851226066d2968a02560f20f56bc07347fdb7 Mon Sep 17 00:00:00 2001 From: Chris Hill-Scott Date: Thu, 17 Jan 2019 17:05:14 +0000 Subject: [PATCH] Add 5 new letter logos --- .../versions/0251_another_letter_org.py | 39 +++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 migrations/versions/0251_another_letter_org.py diff --git a/migrations/versions/0251_another_letter_org.py b/migrations/versions/0251_another_letter_org.py new file mode 100644 index 000000000..2344da9d5 --- /dev/null +++ b/migrations/versions/0251_another_letter_org.py @@ -0,0 +1,39 @@ +"""empty message + +Revision ID: 0251_another_letter_org +Revises: 0250_drop_stats_template_table + +""" + +# revision identifiers, used by Alembic. +revision = '0251_another_letter_org' +down_revision = '0250_drop_stats_template_table' + +from alembic import op + + +NEW_ORGANISATIONS = [ + ('522', 'Anglesey Council', 'anglesey'), + ('523', 'Angus Council', 'angus'), + ('524', 'Cheshire East Council', 'cheshire-east'), + ('525', 'Newham Council', 'newham'), + ('526', 'Warwickshire Council', 'warwickshire'), +] + + +def upgrade(): + for numeric_id, name, filename in NEW_ORGANISATIONS: + op.execute(""" + INSERT + INTO dvla_organisation + VALUES ('{}', '{}', '{}') + """.format(numeric_id, name, filename)) + + +def downgrade(): + for numeric_id, _, _ in NEW_ORGANISATIONS: + op.execute(""" + DELETE + FROM dvla_organisation + WHERE id = '{}' + """.format(numeric_id))