"""change_unique_constraint_for_prs_po

Revision ID: 8d48d33caa58
Revises: 9f25f16ba5a5
Create Date: 2021-03-10 20:28:27.843746

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '8d48d33caa58'
down_revision = '9f25f16ba5a5'
branch_labels = None
depends_on = None


def upgrade():
    op.drop_constraint('uq_PRSPurchaseOrder_po_number_gras_project_id', 'PRSPurchaseOrder', type_='unique')
    op.create_unique_constraint('uq_PRSPurchaseOrder_po_number_gras_project_id',
                                'PRSPurchaseOrder',
                                ['po_number', 'gras_project_id', 'group_id', 'category_id'])


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_constraint('uq_PRSPurchaseOrder_po_number_gras_project_id', 'PRSPurchaseOrder', type_='unique')
    op.create_unique_constraint('uq_PRSPurchaseOrder_po_number_gras_project_id',
                                'PRSPurchaseOrder',
                                ['po_number', 'gras_project_id'])
    # ### end Alembic commands ###