"""Add trim_start and trim_end columns to highlight_candidates.

Revision ID: 021_add_highlight_trim_columns
Revises: 020_add_chapter_status_and_sort_order
"""

from alembic import op
import sqlalchemy as sa


revision = "021_add_highlight_trim_columns"
down_revision = "020_add_chapter_status_and_sort_order"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("highlight_candidates", sa.Column("trim_start", sa.Float(), nullable=True))
    op.add_column("highlight_candidates", sa.Column("trim_end", sa.Float(), nullable=True))


def downgrade() -> None:
    op.drop_column("highlight_candidates", "trim_end")
    op.drop_column("highlight_candidates", "trim_start")