token_bucket_rate_limiter_quickstart_with_memory_storage.py

1from token_bucket import Limiter, MemoryStorage
2
3# Rate limit: 10 requests per second
4# Initial burst: 1 token
5limiter = Limiter(rate=10, capacity=1, storage=MemoryStorage())
6
7def process_request(request_id):
8    if limiter.consume('default_key'):
9        print(f"Request {request_id}: Allowed")
10    else:
11        print(f"Request {request_id}: Rate limited")
12
13# Example usage
14for i in range(5):
15    process_request(i)