zhifu-platform/migrations/versions/0025_business_wait.py

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import mysql

revision = "0025_business_wait"
down_revision = "0024_channel_inbox"
branch_labels = None
depends_on = None


def upgrade():
    op.add_column("agent_runs", sa.Column("business_ticket_id", mysql.BINARY(16), nullable=True))
    op.add_column("agent_runs", sa.Column(
        "business_cursor", mysql.BIGINT(unsigned=True), nullable=False, server_default="0"))
    op.alter_column("agent_runs", "business_cursor", server_default=None,
                    existing_type=mysql.BIGINT(unsigned=True))
    op.drop_constraint(op.f("ck_agent_runs_status"), "agent_runs", type_="check")
    op.create_check_constraint("status", "agent_runs",
        "status IN ('QUEUED','RUNNING','WAITING_USER','WAITING_APPROVAL',"
        "'COMPLETED','FAILED','CANCELLED')")


def downgrade():
    op.drop_constraint(op.f("ck_agent_runs_status"), "agent_runs", type_="check")
    op.create_check_constraint("status", "agent_runs",
        "status IN ('QUEUED','RUNNING','WAITING_USER','COMPLETED','FAILED','CANCELLED')")
    op.drop_column("agent_runs", "business_cursor")
    op.drop_column("agent_runs", "business_ticket_id")