"""remove_unique_po_number_constraint

Revision ID: 5508ac229aca
Revises: 11468bd7913d
Create Date: 2021-03-03 14:05:27.033444

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '5508ac229aca'
down_revision = '11468bd7913d'
branch_labels = None
depends_on = None


def upgrade():
    op.drop_constraint(op.f('fk_Campaign_po_number_PRSPurchaseOrder'), 'Campaign', type_='foreignkey')
    op.drop_constraint(op.f('fk_ArtistMoment_po_number_PRSPurchaseOrder'), 'ArtistMoment', type_='foreignkey')

    op.execute('''ALTER TABLE "PRSPurchaseOrder" DROP CONSTRAINT "uq_PRSPurchaseOrder_po_number" CASCADE''')
    op.create_unique_constraint('uq_PRSPurchaseOrder_po_number_gras_project_id', 'PRSPurchaseOrder', ['po_number', 'gras_project_id'])


def downgrade():
    op.drop_index('uq_PRSPurchaseOrder_po_number_gras_project_id', table_name='PRSPurchaseOrder')

    op.create_unique_constraint('uq_PRSPurchaseOrder_po_number', 'PRSPurchaseOrder', ['po_number'])

    op.create_foreign_key(op.f('fk_Campaign_po_number_PRSPurchaseOrder'), 'Campaign', 'PRSPurchaseOrder', ['po_number'],
                          ['po_number'])
    op.create_foreign_key(op.f('fk_ArtistMoment_po_number_PRSPurchaseOrder'), 'ArtistMoment', 'PRSPurchaseOrder',
                          ['po_number'], ['po_number'])