"""Add execution_mode to workflow_runs.

Revision ID: 064
Revises: 063
"""
from alembic import op
import sqlalchemy as sa

revision = "064"
down_revision = "063"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column(
        "workflow_runs",
        sa.Column(
            "execution_mode",
            sa.String(length=20),
            nullable=False,
            server_default="legacy",
        ),
    )
    op.execute(
        """
        UPDATE workflow_runs
        SET execution_mode = 'legacy'
        WHERE execution_mode IS NULL OR execution_mode = ''
        """
    )
    op.alter_column("workflow_runs", "execution_mode", server_default=None)


def downgrade() -> None:
    op.drop_column("workflow_runs", "execution_mode")