"""create_prs_cache_tables

Revision ID: eeb45cac0d8b
Revises: qfep3td233b2
Create Date: 2021-02-05 12:21:38.386420

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

# revision identifiers, used by Alembic.
revision = 'eeb45cac0d8b'
down_revision = '322986bdfefe'
branch_labels = None
depends_on = None


def upgrade():
    op.create_table(
        'RawPRSProjects',
        sa.Column('sub_project_cd', sa.String(), nullable=False, primary_key=True),
        sa.Column('sub_project_description', sa.String(), nullable=False),
        sa.Column('recording_project_no', sa.Integer(), nullable=False),
        sa.Column('recording_project_artist_tx', sa.String(), nullable=True),
        sa.Column('recording_project_title_tx', sa.String(), nullable=True),
        sa.Column('rep_owner_key', sa.String(), nullable=True),
        sa.Column('recording_project_artist_num', sa.Integer(), nullable=True),
        sa.Column('admin_group_cd', sa.String(), nullable=True),
        sa.Column('admin_group_name_tx', sa.String(), nullable=True),
        sa.PrimaryKeyConstraint('sub_project_cd', name='pk_RawPRSProjects'),
        sa.UniqueConstraint('sub_project_cd', name=op.f('uq_RawPRSProjects_sub_project_cd')),
    )
    op.create_table(
        'RawPRSBudgets',
        sa.Column('sub_project_cd', sa.String(), nullable=False, primary_key=True),
        sa.Column('form_id', sa.Integer(), nullable=False, primary_key=True),
        sa.Column('phase_id', sa.Integer(), nullable=False, primary_key=True),
        sa.Column('phase_short_desc_tx', sa.String(), nullable=True),
        sa.Column('inactive_in', sa.String(), nullable=True),
        sa.Column('group_id', sa.Integer(), nullable=False, primary_key=True),
        sa.Column('category_id', sa.Integer(), nullable=False, primary_key=True),
        sa.Column('proposed_am', sa.DECIMAL(), nullable=True),
        sa.Column('created_user', sa.String(), nullable=True),
        sa.Column('created_at', sa.String(), nullable=True),
        sa.Column('updated_user', sa.String(), nullable=True),
        sa.Column('updated_at', sa.String(), nullable=True),
        sa.PrimaryKeyConstraint('sub_project_cd', 'form_id', 'phase_id', 'group_id', 'category_id', name='pk_RawPRSBudgets'),
        sa.ForeignKeyConstraint(
            ['sub_project_cd'],
            ['RawPRSProjects.sub_project_cd'],
            name=op.f('fk_RawPRSBudgets_sub_project_cd_RawPRSProjects'),
            ondelete='CASCADE'
        ),
    )
    op.create_table(
        'RawPRSPurchaseOrders',
        sa.Column('po_num', sa.Integer(), nullable=True, primary_key=True),
        sa.Column('item_no', sa.Integer(), nullable=True, primary_key=True),
        sa.Column('mps_form_id', sa.Integer(), nullable=True),
        sa.Column('mps_budget_phase_id', sa.Integer(), nullable=True),
        sa.Column('budget_group', sa.Integer(), nullable=True),
        sa.Column('budget_category', sa.Integer(), nullable=True),
        sa.Column('po_created_on', sa.Date(), nullable=True),
        sa.Column('po_modified_on', sa.Date(), nullable=True),
        sa.Column('po_status', sa.String(), nullable=True),
        sa.Column('po_total', sa.DECIMAL(), nullable=True),
        sa.Column('po_total_paid', sa.DECIMAL(), nullable=True),
        sa.Column('sub_project_cd', sa.String(), nullable=False),
        sa.Column('vendor_name', sa.String(), nullable=True),
        sa.Column('is_req_under_blanket', sa.String(), nullable=True),
        sa.Column('blanket', sa.String(), nullable=True),
        sa.Column('blanket_name', sa.String(), nullable=True),
        sa.Column('blanket_po_id', sa.Integer(), nullable=True),
        sa.Column('blanket_item_no', sa.Integer(), nullable=True),
        sa.Column('total_blanket_remain', sa.String(), nullable=True),
        sa.Column('item_desc', sa.String(), nullable=True),
        sa.Column('item_full_desc', sa.String(), nullable=True),
        sa.Column('is_unknown_vendor', sa.String(), nullable=True),
        sa.PrimaryKeyConstraint('po_num', 'item_no', name='pk_RawPRSPurchaseOrders'),
        sa.ForeignKeyConstraint(
            ['sub_project_cd'],
            ['RawPRSProjects.sub_project_cd'],
            name=op.f('fk_RawPRSPurchaseOrders_sub_project_code_RawPRSProjects'),
            ondelete='CASCADE'
        ),
    )

    op.create_table(
        'RawPRSPurchaseOrdersMap',
        sa.Column('label_id', sa.Integer(), nullable=False),
        sa.Column('budget_group_id', sa.Integer(), nullable=False),
        sa.Column('budget_category_id', sa.Integer(), nullable=False),
        sa.Column('mapping_entity_type', sa.Integer(), nullable=False),
        sa.Column('campaign_taxonomy', sa.JSON(), nullable=True),
        sa.Column('moment_group_id', sa.Integer(), nullable=True),
        sa.Column('moment_type_id', sa.Integer(), nullable=True),

        sa.ForeignKeyConstraint(
            ['label_id'],
            ['Label.id'],
            name=op.f('fk_PRSPurchaseOrdersMap_label_id_Label'),
            ondelete='CASCADE'
        ),
        sa.ForeignKeyConstraint(
            ['budget_group_id'],
            ['PRSBudgetGroup.id'],
            name=op.f('fk_PRSPurchaseOrdersMap_budget_group_id_PRSBudgetGroup'),
            ondelete='CASCADE'
        ),
        sa.ForeignKeyConstraint(
            ['budget_category_id'],
            ['PRSBudgetCategory.id'],
            name=op.f('fk_PRSPurchaseOrdersMap_budget_category_id_PRSBudgetCategory'),
            ondelete='CASCADE'
        ),
        sa.ForeignKeyConstraint(
            ['moment_group_id'],
            ['ArtistMomentTypeGroup.id'],
            name=op.f('fk_PRSPurchaseOrdersMap_moment_group_id_ArtistMomentTypeGroup'),
            ondelete='CASCADE'
        ),
        sa.ForeignKeyConstraint(
            ['moment_type_id'],
            ['ArtistMomentType.id'],
            name=op.f('fk_PRSPurchaseOrdersMap_moment_type_id_ArtistMomentType'),
            ondelete='CASCADE'
        ),
    )


def downgrade():
    op.drop_table('RawPRSPurchaseOrdersMap')
    op.drop_table('RawPRSBudgets')
    op.drop_table('RawPRSPurchaseOrders')
    op.drop_table('RawPRSProjects')