workout_api - desafio vmj

workout_api/atleta/controller.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 from uuid import uuid4
-from fastapi import APIRouter, Body, HTTPException, status
+from fastapi import APIRouter, Body, HTTPException, status, Query
 from pydantic import UUID4
 
 from workout_api.atleta.schemas import AtletaIn, AtletaOut, AtletaUpdate
@@ -51,6 +51,16 @@ async def post(
 
         atleta_model.categoria_id = categoria.pk_id
         atleta_model.centro_treinamento_id = centro_treinamento.pk_id
+
+    try:
+        db_session.add(atleta_model)
+        await db_session.commit()
+    except IntegrityError:
+        await db_session.rollback()
+        raise HTTPException(
+            status_code=303,
+            detail=f"Já existe um atleta cadastrado com o cpf: {atleta_in.cpf}"
+        )
 
         db_session.add(atleta_model)
         await db_session.commit()

workout_api/main.py

@@ -1,5 +1,8 @@
 from fastapi import FastAPI
+from fastapi_pagination import add_pagination
 from workout_api.routers import api_router
 
 app = FastAPI(title='WorkoutApi')
 app.include_router(api_router)
+
+add_pagination(app)

workout_api_desafiovitor

@@ -0,0 +1 @@
+