fastapi_slowapi_rate_limiting_by_ip_address.py

1from fastapi import FastAPI, Request
2from slowapi import Limiter, _rate_limit_exceeded_handler
3from slowapi.util import get_remote_address
4from slowapi.errors import RateLimitExceeded
5
6# Initialize the limiter using the client's IP address
7limiter = Limiter(key_func=get_remote_address)
8app = FastAPI()
9
10# Add the limiter to the application state and handle exceptions
11app.state.limiter = limiter
12app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
13
14@app.get("/home")
15@limiter.limit("5/minute")
16async def homepage(request: Request):
17    return {"message": "Welcome to the homepage!"}
18
19@app.get("/mars")
20@limiter.limit("1/minute")
21async def mars(request: Request):
22    return {"message": "Welcome to Mars!"}
23
24if __name__ == "__main__":
25    import uvicorn
26    uvicorn.run(app, host="0.0.0.0", port=8000)