"""Artist level

Revision ID: c85297ab2124
Revises: 67e604d86a08
Create Date: 2019-05-28 18:41:21.880449

"""
from alembic import op
import sqlalchemy as sa

from migrations import utils
from models.artist_levels import ArtistLevel


# revision identifiers, used by Alembic.
revision = 'c85297ab2124'
down_revision = '67e604d86a08'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table(
        'ArtistLevel',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('name', sa.String(), nullable=False),
        sa.PrimaryKeyConstraint('id', name=op.f('pk_ArtistLevel'))
    )

    bind = op.get_bind()
    session = sa.orm.Session(bind=bind)

    # fill artist levels
    artist_levels = (
        {'id': 1, 'name': 'New'},
        {'id': 2, 'name': 'Developing'},
        {'id': 3, 'name': 'Established'},
    )
    utils.add_to_session(session, ArtistLevel, artist_levels)

    session.commit()
    session.close()

    op.add_column(
        'Campaign',
        sa.Column(
            'artist_level_id',
            sa.Integer(),
            nullable=False,
            server_default='1'
        )
    )
    op.create_foreign_key(
        op.f('fk_Campaign_artist_level_id_ArtistLevel'),
        'Campaign',
        'ArtistLevel',
        ['artist_level_id'],
        ['id']
    )
    op.alter_column('Campaign', 'artist_level_id', server_default=None)
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_constraint(
        op.f('fk_Campaign_artist_level_id_ArtistLevel'),
        'Campaign',
        type_='foreignkey'
    )
    op.drop_column('Campaign', 'artist_level_id')
    op.drop_table('ArtistLevel')
    # ### end Alembic commands ###