""" Adding display_tas to staging tables and SF133 table

Revision ID: de5e3fa1d2d2
Revises: d753553fa79b
Create Date: 2019-10-29 17:43:50.519330

"""

# revision identifiers, used by Alembic.
revision = 'de5e3fa1d2d2'
down_revision = 'd753553fa79b'
branch_labels = None
depends_on = None

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

def upgrade(engine_name):
    globals()["upgrade_%s" % engine_name]()


def downgrade(engine_name):
    globals()["downgrade_%s" % engine_name]()





def upgrade_data_broker():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('appropriation', sa.Column('display_tas', sa.Text()))
    op.add_column('award_financial', sa.Column('display_tas', sa.Text()))
    op.add_column('object_class_program_activity', sa.Column('display_tas', sa.Text()))
    op.add_column('sf_133', sa.Column('display_tas', sa.Text(), nullable=True))
    # ### end Alembic commands ###


def downgrade_data_broker():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('object_class_program_activity', 'display_tas')
    op.drop_column('award_financial', 'display_tas')
    op.drop_column('appropriation', 'display_tas')
    op.drop_column('sf_133', 'display_tas')
    # ### end Alembic commands ###

