notifications-api/migrations/versions/0075_create_rates_table.py

"""empty message

Revision ID: 0075_create_rates_table
Revises: 0073_add_international_sms_flag
Create Date: 2017-04-24 15:12:18.907629

"""

# revision identifiers, used by Alembic.
import uuid

from sqlalchemy import text

revision = "0075_create_rates_table"
down_revision = "0073_add_international_sms_flag"

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql


def upgrade():
    notification_types = postgresql.ENUM(
        "email", "sms", "letter", name="notification_type", create_type=False
    )
    op.create_table(
        "rates",
        sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column("valid_from", sa.DateTime(), nullable=False),
        sa.Column("rate", sa.Numeric(), nullable=False),
        sa.Column("notification_type", notification_types, nullable=False),
        sa.PrimaryKeyConstraint("id"),
    )

    op.create_index(
        op.f("ix_rates_notification_type"), "rates", ["notification_type"], unique=False
    )

    conn = op.get_bind()
    input_params = {"id": uuid.uuid4()}
    conn.execute(
        text(
            "INSERT INTO rates(id, valid_from, rate, notification_type) "
            "VALUES(:id, '2016-05-18 00:00:00', 1.65, 'sms')"
        ),
        input_params,
    )
    input_params = {"id": uuid.uuid4()}
    conn.execute(
        text(
            "INSERT INTO rates(id, valid_from, rate, notification_type) "
            "VALUES(:id, '2017-04-01 00:00:00', 1.58, 'sms')"
        ),
        input_params,
    )


def downgrade():
    op.drop_index(op.f("ix_rates_notification_type"), table_name="rates")
    op.drop_table("rates")