mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
"""
|
|
|
|
Revision ID: 0014_drop_monthly_billing_cols
|
|
Revises: 0113_job_created_by_nullable
|
|
Create Date: 2017-07-27 13:36:37.304344
|
|
|
|
"""
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision = "0014_drop_monthly_billing_cols"
|
|
down_revision = "0113_job_created_by_nullable"
|
|
|
|
|
|
def upgrade():
|
|
op.drop_index("uix_monthly_billing", table_name="monthly_billing")
|
|
op.create_unique_constraint(
|
|
"uix_monthly_billing",
|
|
"monthly_billing",
|
|
["service_id", "start_date", "notification_type"],
|
|
)
|
|
op.drop_column("monthly_billing", "year")
|
|
op.drop_column("monthly_billing", "month")
|
|
|
|
|
|
def downgrade():
|
|
op.add_column(
|
|
"monthly_billing",
|
|
sa.Column("month", sa.VARCHAR(), autoincrement=False, nullable=True),
|
|
)
|
|
op.add_column(
|
|
"monthly_billing",
|
|
sa.Column(
|
|
"year",
|
|
postgresql.DOUBLE_PRECISION(precision=53),
|
|
autoincrement=False,
|
|
nullable=True,
|
|
),
|
|
)
|
|
op.drop_constraint("uix_monthly_billing", "monthly_billing", type_="unique")
|
|
op.create_index(
|
|
"uix_monthly_billing",
|
|
"monthly_billing",
|
|
["service_id", "start_date", "notification_type"],
|
|
unique=True,
|
|
)
|