"""updated user model to include LD attributes

Revision ID: 899a70e24bf3
Revises: e32509d93b30
Create Date: 2018-08-16 15:46:28.701665

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '899a70e24bf3'
down_revision = 'e32509d93b30'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('user', sa.Column('account_type', sa.String(length=120), nullable=True))
    op.add_column('user', sa.Column('country', sa.String(length=120), nullable=True))
    op.add_column('user', sa.Column('state', sa.String(length=120), nullable=True))
    op.add_column('user', sa.Column('user_type', sa.String(length=120), nullable=True))
    op.drop_column('user', 'beta_tester')
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('user', sa.Column('beta_tester', sa.BOOLEAN(), autoincrement=False, nullable=True))
    op.drop_column('user', 'user_type')
    op.drop_column('user', 'state')
    op.drop_column('user', 'country')
    op.drop_column('user', 'account_type')
    # ### end Alembic commands ###
