"""

Revision ID: 0351_unique_key_annual_billing
Revises: 0349_add_ft_processing_time
Create Date: 2021-04-12 09:02:45.098875

"""

import os

from alembic import op

revision = "0351_unique_key_annual_billing"
down_revision = "0349_add_ft_processing_time"

environment = os.environ["NOTIFY_ENVIRONMENT"]


def upgrade():
    index = """
        CREATE UNIQUE INDEX CONCURRENTLY uix_service_id_financial_year_start
        ON annual_billing (service_id, financial_year_start)
    """
    constraint = """
        ALTER TABLE annual_BILLING add constraint uix_service_id_financial_year_start
         UNIQUE USING INDEX uix_service_id_financial_year_start
    """
    op.execute("COMMIT")
    op.execute(index)
    op.execute(constraint)


def downgrade():
    op.drop_constraint(
        "uix_service_id_financial_year_start", "annual_billing", type_="unique"
    )