"""remove duplicate habit log index
Revision ID: 0006
Revises: 0005
"""

import sqlalchemy as sa
from alembic import op

revision = "0006"
down_revision = "0005"
branch_labels = None
depends_on = None


def upgrade() -> None:
    bind = op.get_bind()
    indexes = {index["name"] for index in sa.inspect(bind).get_indexes("habit_logs")}
    if "ix_habit_logs_habit_day" in indexes:
        op.drop_index("ix_habit_logs_habit_day", table_name="habit_logs")


def downgrade() -> None:
    bind = op.get_bind()
    indexes = {index["name"] for index in sa.inspect(bind).get_indexes("habit_logs")}
    if "ix_habit_logs_habit_day" not in indexes:
        op.create_index(
            "ix_habit_logs_habit_day",
            "habit_logs",
            ["habit_id", "day"],
        )