token-bucket

from token_bucket import TokenBucket import time # Create a token bucket with a capacity of 10 tokens, # refilling at a rate of 2 tokens per second. # This allows for a burst of 10 requests, then 2 requests/second. bucket = TokenBucket(capacity=10, rate=2.0) print(f"Initial tokens: {bucket.tokens}") # Try to consume 5 tokens if bucket.consume(5): print("Consumed 5 tokens successfully.") else: print("Failed to consume 5 tokens.") print(f"Tokens after first consumption: {bucket.tokens}") # Wait a bit for tokens to refill time.sleep(2.0) # Try to consume 8 tokens (after 2 seconds, 4 tokens should have refilled) if bucket.consume(8): print("Consumed 8 tokens successfully after refill.") else: print("Failed to consume 8 tokens after refill.") print(f"Tokens after second consumption: {bucket.tokens}")