"""Add started_at and completed_at to notes."""

from alembic import op
import sqlalchemy as sa

revision = "0032"
down_revision = "0031"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column("notes", sa.Column("started_at", sa.DateTime(timezone=True), nullable=True))
    op.add_column("notes", sa.Column("completed_at", sa.DateTime(timezone=True), nullable=True))


def downgrade() -> None:
    op.drop_column("notes", "completed_at")
    op.drop_column("notes", "started_at")
