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