"""extend labels

Revision ID: 57d7f103dc0f
Revises: dfba6001b36c
Create Date: 2019-09-02 15:42:26.770304

"""
import sqlalchemy as sa
from alembic import op
from models.labels import Label

# revision identifiers, used by Alembic.
revision = '57d7f103dc0f'
down_revision = 'dfba6001b36c'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###

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

    labels = (
        {'id': 6, 'name': 'US Latin'},
        {'id': 7, 'name': 'Arista'},
        {'id': 8, 'name': 'Latin Regional'},
    )

    labels = [Label(id=label['id'], name=label['name']) for label in labels]
    session.bulk_save_objects(labels)

    session.commit()
    session.close()
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    bind = op.get_bind()
    session = sa.orm.Session(bind=bind)

    session.query(Label).filter(Label.id > 5).delete()

    session.commit()
    session.close()
    # ### end Alembic commands ###