mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-09 06:32:11 -05:00
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
"""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")
|