From 1c960272a62cbdcb8a5d6a7ce7440d43416bef63 Mon Sep 17 00:00:00 2001 From: Chris Hill-Scott Date: Fri, 29 Jun 2018 10:09:18 +0100 Subject: [PATCH] Add Ofgem letter logo --- .../versions/0200_another_letter_org.py | 36 +++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 migrations/versions/0200_another_letter_org.py diff --git a/migrations/versions/0200_another_letter_org.py b/migrations/versions/0200_another_letter_org.py new file mode 100644 index 000000000..ed5ce48f0 --- /dev/null +++ b/migrations/versions/0200_another_letter_org.py @@ -0,0 +1,36 @@ +"""empty message + +Revision ID: 0200_another_letter_org +Revises: 0199_another_letter_org +Create Date: 2017-06-29 12:44:16.815039 + +""" + +# revision identifiers, used by Alembic. +revision = '0200_another_letter_org' +down_revision = '0199_another_letter_org' + +from alembic import op + + +NEW_ORGANISATIONS = [ + ('508', 'Ofgem'), +] + + +def upgrade(): + for numeric_id, name in NEW_ORGANISATIONS: + op.execute(""" + INSERT + INTO dvla_organisation + VALUES ('{}', '{}') + """.format(numeric_id, name)) + + +def downgrade(): + for numeric_id, _ in NEW_ORGANISATIONS: + op.execute(""" + DELETE + FROM dvla_organisation + WHERE id = '{}' + """.format(numeric_id))