ADD: search races in season endpoint

Author: Maxbleu

app/api/season.py

@@ -1,9 +1,12 @@
+import requests, os
 from app.db import get_db
+from sqlalchemy import func
 from app.utils import serialize
 from sqlalchemy.orm import Session
-from fastapi.responses import JSONResponse
 from fastapi import Depends, APIRouter
-from app.models import SeasonEntrantConstructor, SeasonEntrantDriver, SeasonEntrantChassis, SeasonEntrant, SeasonConstructor, SeasonDriver, SeasonTyreManufacturer, SeasonEngineManufacturer, SeasonConstructorStanding, SeasonDriverStanding, Race
+from fastapi.responses import JSONResponse
+from app.schema.season_grand_prix import SeasonGrandPrix
+from app.models import SeasonEntrantConstructor, SeasonEntrantDriver, SeasonEntrantChassis, SeasonEntrant, SeasonConstructor, SeasonDriver, SeasonTyreManufacturer, SeasonEngineManufacturer, SeasonConstructorStanding, SeasonDriverStanding, Race, GrandPrix
 
 router = APIRouter()
 
@@ -109,4 +112,25 @@ async def get_season_constructor_standing_by_year(year: int, db: Session = Depen
             SeasonConstructorStanding.year == year
         ).all()
     constructor_standing_json = serialize(constructor_standing)
-    return JSONResponse(content=constructor_standing_json)
+    return JSONResponse(content=constructor_standing_json)
+
+@router.post("/races/search")
+async def post_season_grand_prix(request: SeasonGrandPrix, db: Session = Depends(get_db)) -> JSONResponse:
+    """Obtiene un grand prix de una temporada"""
+
+    grand_prix = db.query(GrandPrix)\
+        .order_by(
+            func.similarity(GrandPrix.short_name, request.grand_prix).desc()
+        )\
+        .filter(
+            func.similarity(GrandPrix.short_name, request.grand_prix) > 0.3
+        )\
+        .first()
+
+    grand_prix = db.query(Race)\
+        .filter(
+            Race.year == request.year,
+            Race.grand_prix_id == grand_prix.id
+        ).first()
+    grand_prix_json = serialize(grand_prix)
+    return JSONResponse(content=grand_prix_json)

app/schema/season_grand_prix.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+class SeasonGrandPrix(BaseModel):
+    year: int
+    grand_prix: str