Master Real Engineering. One Passing Test at a Time.

class TokenBucket:
    def __init__(self, capacity: int, refill_rate: float):
        self.capacity = capacity
        self.refill_rate = refill_rate
        self.tokens = float(capacity)
        self.last_timestamp = 0.0

    def consume(self, tokens: int, now: float) -> bool:
        # Refill tokens based on elapsed delta
        delta = max(0.0, now - self.last_timestamp)
        self.tokens = min(self.capacity, self.tokens + delta * self.refill_rate)
        self.last_timestamp = now
        
        if self.tokens >= tokens:
            self.tokens -= tokens
            return True
        return False