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

revision = "0011_tenant_quotas"
down_revision = "0010_tenant_dispatch"
branch_labels = None
depends_on = None


def upgrade():
    op.create_table(
        "tenant_quotas",
        sa.Column("tenant_id", mysql.BINARY(16), nullable=False),
        sa.Column("interactive_limit", mysql.BIGINT(unsigned=True), nullable=False),
        sa.Column("background_limit", mysql.BIGINT(unsigned=True), nullable=False),
        sa.Column("version", mysql.BIGINT(unsigned=True), nullable=False),
        sa.ForeignKeyConstraint(["tenant_id"], ["tenants.id"], ondelete="CASCADE"),
        sa.PrimaryKeyConstraint("tenant_id"),
        sa.CheckConstraint("interactive_limit BETWEEN 1 AND 8", name="interactive_limit"),
        sa.CheckConstraint("background_limit BETWEEN 0 AND 8", name="background_limit"),
    )


def downgrade():
    op.drop_table("tenant_quotas")