Implement rate limiting (#54)

app/api/endpoints/players.py

@@ -47,7 +47,6 @@ def get_player_stats(player_id: str):
     return player_stats
 
 
-# Define the endpoint for injuries
 @router.get("/{player_id}/injuries")
 def get_player_injuries(player_id: str, page_number: Optional[int] = 1):
     tfmkt = TransfermarktPlayerInjuries(player_id=player_id, page_number=page_number)

app/main.py

@@ -1,10 +1,23 @@
 import uvicorn
 from fastapi import FastAPI
+from slowapi import Limiter, _rate_limit_exceeded_handler
+from slowapi.errors import RateLimitExceeded
+from slowapi.middleware import SlowAPIMiddleware
+from slowapi.util import get_remote_address
 from starlette.responses import RedirectResponse
 
 from app.api.api import api_router
+from app.settings import settings
 
+limiter = Limiter(
+    key_func=get_remote_address,
+    default_limits=[settings.RATE_LIMITING_FREQUENCY],
+    enabled=settings.RATE_LIMITING_ENABLE,
+)
 app = FastAPI(title="Transfermarkt API")
+app.state.limiter = limiter
+app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
+app.add_middleware(SlowAPIMiddleware)
 app.include_router(api_router)
 
 

app/settings.py

@@ -0,0 +1,10 @@
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class Settings(BaseSettings):
+    model_config = SettingsConfigDict(env_file=".env")
+    RATE_LIMITING_ENABLE: bool = False
+    RATE_LIMITING_FREQUENCY: str = "2/3seconds"
+
+
+settings = Settings()