PyrateLimiter example with Redis

pyrate_limiter.py

from pyrate_limiter import RedisBucket, Limiter, Duration, Rate, BucketFullException
from redis import Redis

max_requests = 5
time_period = Duration.SECOND * 2

rate = Rate(max_requests, time_period)

redis_connection = Redis(host="localhost")
redis_bucket = RedisBucket.init([rate], redis_connection, "my-bucket-name")

limiter = Limiter(redis_bucket)

for request in range(max_requests + 1):
    try:
        limiter.try_acquire(request)
    except BucketFullException as err:
        print(err)