0011_tenant_quotas.py 939 B

1234567891011121314151617181920212223242526
  1. import sqlalchemy as sa
  2. from alembic import op
  3. from sqlalchemy.dialects import mysql
  4. revision = "0011_tenant_quotas"
  5. down_revision = "0010_tenant_dispatch"
  6. branch_labels = None
  7. depends_on = None
  8. def upgrade():
  9. op.create_table(
  10. "tenant_quotas",
  11. sa.Column("tenant_id", mysql.BINARY(16), nullable=False),
  12. sa.Column("interactive_limit", mysql.BIGINT(unsigned=True), nullable=False),
  13. sa.Column("background_limit", mysql.BIGINT(unsigned=True), nullable=False),
  14. sa.Column("version", mysql.BIGINT(unsigned=True), nullable=False),
  15. sa.ForeignKeyConstraint(["tenant_id"], ["tenants.id"], ondelete="CASCADE"),
  16. sa.PrimaryKeyConstraint("tenant_id"),
  17. sa.CheckConstraint("interactive_limit BETWEEN 1 AND 8", name="interactive_limit"),
  18. sa.CheckConstraint("background_limit BETWEEN 0 AND 8", name="background_limit"),
  19. )
  20. def downgrade():
  21. op.drop_table("tenant_quotas")