"""add_prs_codes_to_moment_and_campaign

Revision ID: 22e3392754dc
Revises: 04123319e94d
Create Date: 2021-02-16 12:28:10.834343

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision = '22e3392754dc'
down_revision = '04123319e94d'
branch_labels = None
depends_on = None


def upgrade():
    op.create_unique_constraint('uq_PRSPurchaseOrder_po_number', 'PRSPurchaseOrder', ['po_number'])
    op.add_column('ArtistMoment', sa.Column('po_number', sa.BigInteger(), nullable=True))
    op.add_column('Campaign', sa.Column('po_number', sa.BigInteger(), nullable=True))
    op.create_foreign_key(op.f('fk_ArtistMoment_po_number_PRSPurchaseOrder'), 'ArtistMoment', 'PRSPurchaseOrder',
                          ['po_number'], ['po_number'], ondelete='CASCADE')
    op.create_foreign_key(op.f('fk_Campaign_po_number_PRSPurchaseOrder'), 'Campaign', 'PRSPurchaseOrder',
                          ['po_number'], ['po_number'], ondelete='CASCADE')

def downgrade():
    op.drop_constraint(op.f('uq_PRSPurchaseOrder_po_number'), 'PRSPurchaseOrder', type_='unique')
    op.drop_constraint('fk_ArtistMoment_po_number_PRSPurchaseOrder', 'ArtistMoment', type_='foreignkey')
    op.drop_constraint('fk_Campaign_po_number_PRSPurchaseOrder', 'Campaign', type_='foreignkey')
    op.drop_column('ArtistMoment', 'po_number')
    op.drop_column('Campaign', 'po_number')