From 9991b40871ccd70ee4359b8e44e9a3a5aa15fd8a Mon Sep 17 00:00:00 2001
From: Chris Hill-Scott <me@quis.cc>
Date: Mon, 5 Nov 2018 10:03:47 +0000
Subject: [PATCH] Add Buckinghamshire and Worcestershire letter logos

---
 .../versions/0243_another_letter_org.py       | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 migrations/versions/0243_another_letter_org.py

diff --git a/migrations/versions/0243_another_letter_org.py b/migrations/versions/0243_another_letter_org.py
new file mode 100644
index 000000000..853de5e76
--- /dev/null
+++ b/migrations/versions/0243_another_letter_org.py
@@ -0,0 +1,36 @@
+"""empty message
+
+Revision ID: 0243_another_letter_org
+Revises: 0242_template_folders
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '0243_another_letter_org'
+down_revision = '0242_template_folders'
+
+from alembic import op
+
+
+NEW_ORGANISATIONS = [
+    ('516', 'Worcestershire County Council', 'worcestershire'),
+    ('517', 'Buckinghamshire County Council', 'buckinghamshire'),
+]
+
+
+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))