"""add_is_unknown_field_to_artist

Revision ID: 4a52a3db8fc7
Revises: f6735ec22d02
Create Date: 2021-07-29 12:05:10.880857

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy import or_

from migrations.utils import get_table


# revision identifiers, used by Alembic.
revision = '4a52a3db8fc7'
down_revision = 'f6735ec22d02'
branch_labels = None
depends_on = None


def upgrade():
    # SELECT * FROM "Artist" where name = 'Unknown' or name ilike '%Various%';
    op.add_column('Artist', sa.Column('is_unknown', sa.Boolean(), server_default='FALSE', nullable=False))

    bind = op.get_bind()
    artist_table = get_table(bind, "Artist")

    bind.execute(
        artist_table.update().values(is_unknown=True).where(
            or_(artist_table.c.name == 'Unknown', artist_table.c.name.ilike('%Various%'))
        )
    )


def downgrade():
    op.drop_column('Artist', 'is_unknown')