"""update_ccp_tables

Revision ID: dbcf68de163d
Revises: 55911bcdbc01
Create Date: 2021-11-23 13:56:07.800072

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'dbcf68de163d'
down_revision = '55911bcdbc01'
branch_labels = None
depends_on = None


def upgrade():
    op.execute('''
        update "PRSImportInfo"
        set imported_ccp_date = '2021-11-22'
        where id = 1
    ''')
    op.drop_table("RawCCPBudgets")
    op.drop_table("RawCCPPurchaseOrders")
    op.create_table(
        'RawCCPBudgets',
        sa.Column('project_budget_code', sa.String(), nullable=False),
        sa.Column('description', sa.String(), nullable=False),
        sa.Column('commodity_lvl_2', sa.String(), nullable=True),
        sa.Column('commodity_lvl_2_description', sa.String(), nullable=True),
        sa.Column('commodity_lvl_3', sa.String(), nullable=True),
        sa.Column('commodity_lvl_3_description', sa.String(), nullable=True),
        sa.Column('expense_type', sa.String(), nullable=True),
        sa.Column('expense_type_description', sa.String(), nullable=True),
        sa.Column('expense_group', sa.String(), nullable=True),
        sa.Column('expense_group_description', sa.String(), nullable=True),
        sa.Column('total_project_amount', sa.String(), nullable=False),
        sa.Column('currency', sa.String(), nullable=False),
        sa.Column('creation_date', sa.Date(), nullable=False),
        sa.Column('last_modified_date', sa.Date(), nullable=True),
        sa.Column('region', sa.String(), nullable=True),
    )
    op.create_table(
        'RawCCPPurchaseOrders',
        sa.Column('po_id', sa.String(), nullable=False),
        sa.Column('po_line', sa.String(), nullable=False),
        sa.Column('commodity_lvl_2', sa.String(), nullable=True),
        sa.Column('expense_type', sa.String(), nullable=True),
        sa.Column('creation_date', sa.Date(), nullable=False),
        sa.Column('last_modified_date', sa.Date(), nullable=True),
        sa.Column('status', sa.String(), nullable=False),
        sa.Column('line_item_total_amount', sa.DECIMAL(), nullable=False),
        sa.Column('line_item_total_amount_currency', sa.String(), nullable=False),
        sa.Column('paid_line_item_amount', sa.DECIMAL(), nullable=True),
        sa.Column('paid_line_item_amount_currency', sa.String(), nullable=True),
        sa.Column('project_budget_code', sa.String(), nullable=True),
        sa.Column('supplier', sa.String(), nullable=True),
        sa.Column('short_description', sa.String(), nullable=False),
        sa.Column('region', sa.String(), nullable=False),
    )


def downgrade():
    op.drop_table("RawCCPBudgets")
    op.drop_table("RawCCPPurchaseOrders")

    op.create_table(
        'RawCCPBudgets',
        sa.Column('project_budget_code', sa.String(), nullable=False),
        sa.Column('description', sa.String(), nullable=False),
        sa.Column('commodity_lvl_2', sa.String(), nullable=True),
        sa.Column('commodity_lvl_3', sa.String(), nullable=True),
        sa.Column('total_project_amount', sa.String(), nullable=False),
        sa.Column('currency', sa.String(), nullable=False),
        sa.Column('creation_date', sa.Date(), nullable=False),
        sa.Column('last_modified_date', sa.Date(), nullable=True),
        sa.Column('region', sa.String(), nullable=True),
    )

    op.create_table(
        'RawCCPPurchaseOrders',
        sa.Column('po_id', sa.String(), nullable=False),
        sa.Column('po_line', sa.String(), nullable=False),
        sa.Column('commodity_lvl_2', sa.String(), nullable=True),
        sa.Column('expense_type', sa.String(), nullable=True),
        sa.Column('creation_date', sa.Date(), nullable=False),
        sa.Column('last_modified_date', sa.Date(), nullable=True),
        sa.Column('status', sa.String(), nullable=False),
        sa.Column('line_item_total_amount', sa.DECIMAL(), nullable=False),
        sa.Column('line_item_total_amount_currency', sa.String(), nullable=False),
        sa.Column('paid_line_item_amount', sa.DECIMAL(), nullable=True),
        sa.Column('paid_line_item_amount_currency', sa.String(), nullable=True),
        sa.Column('project_budget_code', sa.String(), nullable=True),
        sa.Column('supplier', sa.String(), nullable=True),
        sa.Column('short_description', sa.String(), nullable=False),
        sa.Column('notes', sa.String(), nullable=True),
        sa.Column('region', sa.String(), nullable=False),
    )