Files
notifications-api/migrations/versions/0351_unique_key_annual_billing.py

37 lines
866 B
Python
Raw Normal View History

"""
Revision ID: 0351_unique_key_annual_billing
2023-07-19 09:19:40 -07:00
Revises: 0349_add_ft_processing_time
Create Date: 2021-04-12 09:02:45.098875
"""
2024-04-01 15:12:33 -07:00
import os
from alembic import op
2023-08-29 14:54:30 -07:00
revision = "0351_unique_key_annual_billing"
down_revision = "0349_add_ft_processing_time"
2023-08-29 14:54:30 -07:00
environment = os.environ["NOTIFY_ENVIRONMENT"]
def upgrade():
index = """
2023-11-17 09:47:32 -05:00
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
"""
2023-08-29 14:54:30 -07:00
op.execute("COMMIT")
op.execute(index)
op.execute(constraint)
def downgrade():
2023-08-29 14:54:30 -07:00
op.drop_constraint(
"uix_service_id_financial_year_start", "annual_billing", type_="unique"
)