"""Add weather and track_condition to races.

Revision ID: 004_race_conditions
Revises: 003_tab_api
Create Date: 2026-03-01 00:00:00.000000
"""

from collections.abc import Sequence

import sqlalchemy as sa
from alembic import op

revision: str = "004_race_conditions"
down_revision: str | None = "003_tab_api"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    op.add_column(
        "races",
        sa.Column(
            "weather",
            sa.String(length=64),
            nullable=True,
            comment="Weather at race time",
        ),
    )
    op.add_column(
        "races",
        sa.Column(
            "track_condition",
            sa.String(length=64),
            nullable=True,
            comment="Track condition",
        ),
    )


def downgrade() -> None:
    op.drop_column("races", "track_condition")
    op.drop_column("races", "weather")