From beb7fda32b216aeaf553b3630ba85eeafeec23f6 Mon Sep 17 00:00:00 2001 From: Rebecca Law Date: Thu, 2 Mar 2017 11:42:37 +0000 Subject: [PATCH] Added the provider details for DVLA. The rates are a bogus number. --- migrations/versions/0066_add_dvla_provider.py | 31 +++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 migrations/versions/0066_add_dvla_provider.py diff --git a/migrations/versions/0066_add_dvla_provider.py b/migrations/versions/0066_add_dvla_provider.py new file mode 100644 index 000000000..5f8a0c6ee --- /dev/null +++ b/migrations/versions/0066_add_dvla_provider.py @@ -0,0 +1,31 @@ +"""empty message + +Revision ID: 0066_add_dvla_provider +Revises: 0065_users_current_session_id +Create Date: 2017-03-02 10:32:28.984947 + +""" +import uuid +from datetime import datetime + +revision = '0066_add_dvla_provider' +down_revision = '0065_users_current_session_id' + +from alembic import op + + +def upgrade(): + provider_id = str(uuid.uuid4()) + op.execute( + "INSERT INTO provider_details (id, display_name, identifier, priority, notification_type, active, version) values ('{}', 'DVLA', 'dvla', 10, 'letter', true, 1)".format(provider_id) + ) + op.execute( + "INSERT INTO provider_details_history (id, display_name, identifier, priority, notification_type, active, version) values ('{}', 'DVLA', 'mmg', 10, 'sms', true, 1)".format(provider_id) + ) + op.execute("INSERT INTO provider_rates (id, valid_from, rate, provider_id) VALUES ('{}', '{}', 1.0, '{}')".format(uuid.uuid4(), datetime.utcnow(), provider_id)) + + +def downgrade(): + op.execute("DELETE FROM provider_rates where provider_id = (SELECT id from provider_details where display_name='DVLA')") + op.execute("DELETE FROM provider_details_history where display_name = 'DVLA'") + op.execute("DELETE FROM provider_details where display_name = 'DVLA'")