Files
notifications-api/migrations/versions/0226_new_letter_rates.py
Pea Tyczynska b3191dbc2a Filter letter rates by post_class in get_rate
Also adjust existing tests.
2018-09-25 15:10:15 +01:00

55 lines
1.7 KiB
Python

"""empty message
Revision ID: 0226_new_letter_rates
Revises: 0225_another_letter_org
"""
revision = '0226_new_letter_rates'
down_revision = '0225_another_letter_org'
import uuid
import pytz
from datetime import datetime
from alembic import op
start = datetime(2018, 10, 1, 0, 0, tzinfo=pytz.utc)
NEW_RATES = [
(uuid.uuid4(), start, 1, 0.30, True, 'second'),
(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