Implement Rate Limiter

app/routers/frontend.py

@@ -1,8 +1,8 @@
 # -*- coding: utf-8 -*-
 from typing import Annotated, List
-from fastapi import APIRouter, Depends, HTTPException
+from fastapi import APIRouter, Depends, HTTPException, Request
 from tortoise.exceptions import ValidationError
-
+from fastapi_simple_rate_limiter import rate_limiter
 from app.dependencies import (
     get_current_frontend_user
 )
@@ -45,9 +45,11 @@ async def get_user_info(
 
 
 @router.get("/patient/header/{cpf}")
+@rate_limiter(limit=5, seconds=60)
 async def get_patient_header(
     _: Annotated[User, Depends(get_current_frontend_user)],
     cpf: str,
+    request: Request,
 ) -> PatientHeader:
     validator = CPFValidator()
     try:
@@ -79,9 +81,11 @@ async def get_patient_header(
 
 
 @router.get("/patient/summary/{cpf}")
+@rate_limiter(limit=5, seconds=60)
 async def get_patient_summary(
     _: Annotated[User, Depends(get_current_frontend_user)],
     cpf: str,
+    request: Request,
 ) -> PatientSummary:
 
     results = await read_bq(
@@ -114,9 +118,11 @@ async def get_filter_tags(
 
 
 @router.get("/patient/encounters/{cpf}")
+@rate_limiter(limit=5, seconds=60)
 async def get_patient_encounters(
     _: Annotated[User, Depends(get_current_frontend_user)],
     cpf: str,
+    request: Request,
 ) -> List[Encounter]:
 
     results = await read_bq(

app/types/frontend.py

@@ -10,10 +10,10 @@ def __init__(
         self,
         username: str,
         password: str,
-        totp: str,
+        totp_code: str,
     ):
         super().__init__(username=username, password=password)
-        self.totp = totp
+        self.totp_code = totp_code
 
 
 # Clinic Family model

pyproject.toml

@@ -33,6 +33,7 @@ nltk = "^3.9.1"
 asyncer = "^0.0.8"
 qrcode = "^7.4.2"
 pyotp = "^2.9.0"
+fastapi-simple-rate-limiter = "^0.0.4"
 
 
 [tool.poetry.group.dev.dependencies]