notifications-api/migrations/versions/0191_ft_billing_pkey.py

"""

Revision ID: 0191_ft_billing_pkey
Revises: 0189_ft_billing_data
Create Date: 2018-05-21 14:24:27.229511

"""
from alembic import op

revision = '0191_ft_billing_pkey'
down_revision = '0189_ft_billing_data_type'


def upgrade():
    op.get_bind()
    op.execute("ALTER TABLE ft_billing DROP CONSTRAINT ft_billing_pkey")
    sql = """ALTER TABLE ft_billing ADD CONSTRAINT 
    ft_billing_pkey PRIMARY KEY 
    (bst_date, template_id, service_id, rate_multiplier, provider, notification_type, international, rate)"""
    op.execute(sql)


def downgrade():
    op.get_bind()
    op.execute("ALTER TABLE ft_billing DROP CONSTRAINT ft_billing_pkey")
    sql = """ALTER TABLE ft_billing ADD CONSTRAINT 
    ft_billing_pkey PRIMARY KEY 
    (bst_date, template_id, service_id, rate_multiplier, provider, notification_type, international)"""
    op.execute(sql)