notifications-api/migrations/versions/0229_new_letter_rates.py

"""empty message

Revision ID: 0229_new_letter_rates
Revises: 0228_notification_postage

"""

revision = '0229_new_letter_rates'
down_revision = '0228_notification_postage'

import uuid
from datetime import datetime
from alembic import op


start = datetime(2018, 9, 30, 23, 0)

NEW_RATES = [
    (uuid.uuid4(), start, 1, 0.30, False, 'second'),
    (uuid.uuid4(), start, 2, 0.35, True, 'second'),
    (uuid.uuid4(), start, 2, 0.35, False, 'second'),
    (uuid.uuid4(), start, 3, 0.40, True, 'second'),
    (uuid.uuid4(), start, 3, 0.40, False, 'second'),
    (uuid.uuid4(), start, 4, 0.45, True, 'second'),
    (uuid.uuid4(), start, 4, 0.45, False, 'second'),
    (uuid.uuid4(), start, 5, 0.50, True, 'second'),
    (uuid.uuid4(), start, 5, 0.50, False, 'second'),
    (uuid.uuid4(), start, 1, 0.56, True, 'first'),
    (uuid.uuid4(), start, 1, 0.56, False, 'first'),
    (uuid.uuid4(), start, 2, 0.61, True, 'first'),
    (uuid.uuid4(), start, 2, 0.61, False, 'first'),
    (uuid.uuid4(), start, 3, 0.66, True, 'first'),
    (uuid.uuid4(), start, 3, 0.66, False, 'first'),
    (uuid.uuid4(), start, 4, 0.71, True, 'first'),
    (uuid.uuid4(), start, 4, 0.71, False, 'first'),
    (uuid.uuid4(), start, 5, 0.76, True, 'first'),
    (uuid.uuid4(), start, 5, 0.76, False, 'first'),
]


def upgrade():
    conn = op.get_bind()
    for id, start_date, sheet_count, rate, crown, post_class in NEW_RATES:
        conn.execute("""
            INSERT INTO letter_rates (id, start_date, sheet_count, rate, crown, post_class)
                VALUES ('{}', '{}', '{}', '{}', '{}', '{}')
        """.format(id, start_date, sheet_count, rate, crown, post_class))


def downgrade():
    pass