"""sanitize nullable and non-nullable fields

Revision ID: 15a14d428bff
Revises: 05f3ae5db7c7
Create Date: 2020-09-10 12:27:31.889962

"""
from alembic import op


# revision identifiers, used by Alembic.
revision = "15a14d428bff"
down_revision = "05f3ae5db7c7"
branch_labels = None
depends_on = None


def upgrade():
    nullable = {
        "full_name": "users",
        "email": "users",
        "mobile": "users",
        "name": "cities",
        "name": "games",
        "price": "games",
        "name": "web_bookings",
        "email": "web_bookings",
        "contact": "web_bookings",
        "message": "web_bookings",
        "game": "web_bookings",
        "city": "web_bookings",
        "address": "venues",
        "name": "venues",
        "spanish_name": "sports",
        "rating": "user_ratings",
        "name": "sports",
    }
    non_nullable = {
        "social_id": "users",
        "type": "users",
    }
    for field, table in nullable.items():
        query = "UPDATE {0} SET {1} = '' WHERE {1} IS NULL".format(table, field)
        op.execute(query)
        with op.batch_alter_table(table) as batch_op:
            batch_op.alter_column(
                column_name=field, nullable=False, server_default=None
            )
    for field, table in non_nullable.items():
        with op.batch_alter_table(table) as batch_op:
            batch_op.alter_column(column_name=field, nullable=True, server_default=None)
        query = "UPDATE {0} SET {1} = NULL WHERE {1} = '' OR {1} = '0'".format(
            table, field
        )
        op.execute(query)


def downgrade():
    pass