feat: Add changes to use version 0.100.0 of FastAPI

This versión of FastAPI use the version 2 of Pydantic core rewritten in rust for a mayor efficiency (among other improvements) and implement minor changes for request error handling

Author: GermanMT

app/config.py

@@ -1,5 +1,5 @@
 from functools import lru_cache
-from pydantic import BaseSettings
+from pydantic_settings import BaseSettings
 
 
 class Settings(BaseSettings):

app/controllers/graph_controller.py

@@ -20,11 +20,8 @@ async def get_graph(repository_id: str) -> JSONResponse:
 
     - **repository_id**: the id of a repository
     '''
-    try:
-        graph = await read_graph_by_repository_id(repository_id, '_')
-        return JSONResponse(status_code=status.HTTP_200_OK, content=graph)
-    except HTTPException as error:
-        return JSONResponse(status_code=error.status_code, content=json_encoder({'message': error.detail}))
+    graph = await read_graph_by_repository_id(repository_id, '_')
+    return JSONResponse(status_code=status.HTTP_200_OK, content=graph)
 
 
 @router.post(
@@ -39,13 +36,10 @@ async def init_graph(repository: RepositoryModel) -> JSONResponse:
     - **repository**: a json containing the owner and the name of a repository
     '''
     repository_json = jsonable_encoder(repository)
-    try:
-        repository_id = await read_repository_by_owner_name(repository_json['owner'], repository_json['name'], '_')
-        if not repository_id:
-            await extract_graph(repository_json)
-        else:
-            # TODO: Hacer un método que actualize el grafo existente en la base de datos
-            pass
-        return JSONResponse(status_code=status.HTTP_201_CREATED, content=json_encoder({'message': 'created'}))
-    except HTTPException as error:
-        return JSONResponse(status_code=error.status_code, content=json_encoder({'message': error.detail}))
+    repository_id = await read_repository_by_owner_name(repository_json['owner'], repository_json['name'], '_')
+    if not repository_id:
+        await extract_graph(repository_json)
+    else:
+        # TODO: Hacer un método que actualize el grafo existente en la base de datos
+        pass
+    return JSONResponse(status_code=status.HTTP_201_CREATED, content=json_encoder({'message': 'created'}))

app/main.py

@@ -1,13 +1,16 @@
-from json import loads
 from fastapi import FastAPI, Request
-from fastapi.exceptions import RequestValidationError, ValidationError
+from fastapi.exceptions import RequestValidationError
+from fastapi.exception_handlers import (
+    http_exception_handler,
+    request_validation_exception_handler,
+)
 from starlette.middleware.cors import CORSMiddleware
-from starlette.responses import JSONResponse
+from starlette.exceptions import HTTPException
+from starlette.responses import Response
 from apscheduler.schedulers.background import BackgroundScheduler
 from app.controllers import db_updater
 from app.router import api_router
 from app.services import create_indexes
-from app.utils import json_encoder
 
 DESCRIPTION = '''
 A backend for dependency graph building, atribution of vulnerabilities and reasoning
@@ -39,15 +42,14 @@ async def startup_event() -> None:
     scheduler.start()
 
 
+@app.exception_handler(HTTPException)
+async def custom_http_exception_handler(request: Request, exc: HTTPException) -> Response:
+    return await http_exception_handler(request, exc)
+
+
 @app.exception_handler(RequestValidationError)
-@app.exception_handler(ValidationError)
-async def validation_exception_handler(_: Request, exc: ValidationError | RequestValidationError) -> JSONResponse:
-    exc_json = loads(exc.json())
-    response: dict[str, list[str]] = {'message': []}
-    for error in exc_json:
-        response['message'].append(error['loc'][-1] + f": {error['msg']}")
-
-    return JSONResponse(content=json_encoder(response), status_code=422)
+async def validation_exception_handler(request: Request, exc: RequestValidationError) -> Response:
+    return await request_validation_exception_handler(request, exc)
 
 app.add_middleware(
     CORSMiddleware,

app/models/models.py

@@ -16,9 +16,9 @@ class VersionModel(BaseModel):
     count: int
 
     class Config:
-        allow_population_by_field_name = True
+        populate_by_name = True
         arbitrary_types_allowed = True
-        schema_extra = {
+        json_schema_extra = {
             'example': {
                 'release': '1.26.5',
                 'release_date': datetime.now(),
@@ -34,9 +34,9 @@ class PackageModel(BaseModel):
     versions: list[VersionModel] | None
 
     class Config:
-        allow_population_by_field_name = True
+        populate_by_name = True
         arbitrary_types_allowed = True
-        schema_extra = {
+        json_schema_extra = {
             'example': {
                 'name': 'urllib3',
                 'moment': datetime.now()
@@ -49,9 +49,9 @@ class RequirementFile(BaseModel):
     manager: str
 
     class Config:
-        allow_population_by_field_name = True
+        populate_by_name = True
         arbitrary_types_allowed = True
-        schema_extra = {
+        json_schema_extra = {
             'example': {
                 'name': 'requirements.txt',
                 'manager': 'PIP'
@@ -75,9 +75,9 @@ class RepositoryModel(BaseModel):
     requirement_files: list[RequirementFile] | None
 
     class Config:
-        allow_population_by_field_name = True
+        populate_by_name = True
         arbitrary_types_allowed = True
-        schema_extra = {
+        json_schema_extra = {
             'example': {
                 'owner': 'depexorg',
                 'name': 'pip_test',

requirements.txt

@@ -1,5 +1,6 @@
 wheel==0.40.0
-fastapi[all]==0.98.0
+fastapi[all]==0.100.0
+pydantic-settings==2.0.1
 apscheduler==3.10.1
 motor==3.1.2
 python-dotenv==1.0.0