darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-08 04:48:24 -04:00
Code Issues Packages Projects Releases Wiki Activity

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34100 additions and 23589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
migrations/versions/0075_create_rates_table.py
View File

@@ -11,38 +11,50 @@ import uuid
from sqlalchemy import text
revision = '0075_create_rates_table'
down_revision = '0073_add_international_sms_flag'
revision = "0075_create_rates_table"
down_revision = "0073_add_international_sms_flag"
from alembic import op
import sqlalchemy as sa
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')
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)
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)
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')
op.drop_index(op.f("ix_rates_notification_type"), table_name="rates")
op.drop_table("rates")