ขั้นตอนการเขียนโค้ด · การรับด้วยถังโทเคน

from math import isfinite
def admit(tokens, last, now, capacity, rate, cost):
    values = (tokens, last, now, capacity, rate, cost)
    if not all(type(x) in (int, float) and isfinite(x) for x in values):
        raise ValueError("finite numbers required")
    if not (capacity > 0 and rate > 0 and 0 < cost <= capacity
            and 0 <= tokens <= capacity and now >= last):
        raise ValueError("invalid bucket state")
    refill = min(capacity, tokens + (now - last) * rate)
    allowed = refill >= cost
    return (refill - cost if allowed else refill, now), allowed