Files
notifications-api/migrations/versions/0210_remove_monthly_billing.py
Cliff Hill 1157f5639d black, isort, flake8
Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
2023-12-08 21:43:52 -05:00

62 lines
2.0 KiB
Python

"""
Revision ID: 0210_remove_monthly_billing
Revises: 0209_add_cancelled_status
Create Date: 2018-07-31 16:43:00.568972
"""
import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql
revision = "0210_remove_monthly_billing"
down_revision = "0209_add_cancelled_status"
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_index("ix_monthly_billing_service_id", table_name="monthly_billing")
op.drop_table("monthly_billing")
# ### end Alembic commands ###
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.create_table(
"monthly_billing",
sa.Column("id", postgresql.UUID(), autoincrement=False, nullable=False),
sa.Column("service_id", postgresql.UUID(), autoincrement=False, nullable=False),
sa.Column(
"notification_type",
postgresql.ENUM("email", "sms", "letter", name="notification_type"),
autoincrement=False,
nullable=False,
),
sa.Column(
"monthly_totals",
postgresql.JSON(astext_type=sa.Text()),
autoincrement=False,
nullable=False,
),
sa.Column(
"updated_at", postgresql.TIMESTAMP(), autoincrement=False, nullable=False
),
sa.Column(
"start_date", postgresql.TIMESTAMP(), autoincrement=False, nullable=False
),
sa.Column(
"end_date", postgresql.TIMESTAMP(), autoincrement=False, nullable=False
),
sa.ForeignKeyConstraint(
["service_id"], ["services.id"], name="monthly_billing_service_id_fkey"
),
sa.PrimaryKeyConstraint("id", name="monthly_billing_pkey"),
sa.UniqueConstraint(
"service_id", "start_date", "notification_type", name="uix_monthly_billing"
),
)
op.create_index(
"ix_monthly_billing_service_id", "monthly_billing", ["service_id"], unique=False
)
# ### end Alembic commands ###