"""

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,
    )