Files
dodo/migrations/versions/0006_remove_duplicate_habit_index.py
T

31 lines
776 B
Python

"""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"],
)