zhifu-platform/migrations/versions/0012_tenant_reservations.py

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

revision = "0012_tenant_reservations"
down_revision = "0011_tenant_quotas"
branch_labels = None
depends_on = None


def upgrade():
    op.add_column(
        "tenant_quotas",
        sa.Column(
            "interactive_reserved", mysql.BIGINT(unsigned=True), nullable=False, server_default="0"
        ),
    )
    op.create_check_constraint(
        "reserved_limit", "tenant_quotas", "interactive_reserved <= interactive_limit"
    )
    policy = op.create_table(
        "capacity_policy", sa.Column("id", mysql.BIGINT(unsigned=True), primary_key=True)
    )
    op.bulk_insert(policy, [{"id": 1}])


def downgrade():
    op.drop_table("capacity_policy")
    op.drop_constraint("ck_tenant_quotas_reserved_limit", "tenant_quotas", type_="check")
    op.drop_column("tenant_quotas", "interactive_reserved")