4.8 KiB
Clase personalizada de Request y APIRoute
En algunos casos, puede que quieras sobrescribir la lógica utilizada por las clases Request y APIRoute.
En particular, esta puede ser una buena alternativa a la lógica en un middleware.
Por ejemplo, si quieres leer o manipular el request body antes de que sea procesado por tu aplicación.
/// danger | Advertencia
Esta es una funcionalidad "avanzada".
Si apenas estás comenzando con FastAPI, quizás quieras saltar esta sección.
///
Casos de uso
Algunos casos de uso incluyen:
- Convertir request bodies no-JSON a JSON (por ejemplo,
msgpack). - Descomprimir request bodies comprimidos con gzip.
- Registrar automáticamente todos los request bodies.
Manejo de codificaciones personalizadas de request body
Veamos cómo hacer uso de una subclase personalizada de Request para descomprimir requests gzip.
Y una subclase de APIRoute para usar esa clase de request personalizada.
Crear una clase personalizada GzipRequest
/// tip | Consejo
Este es un ejemplo de juguete para demostrar cómo funciona, si necesitas soporte para Gzip, puedes usar el GzipMiddleware proporcionado.
///
Primero, creamos una clase GzipRequest, que sobrescribirá el método Request.body() para descomprimir el request body si hay un header apropiado.
Si no hay gzip en el header, no intentará descomprimir el request body.
De esa manera, la misma clase de ruta puede manejar requests comprimidos con gzip o no comprimidos.
{* ../../docs_src/custom_request_and_route/tutorial001_an_py310.py hl[9:16] *}
Crear una clase personalizada GzipRoute
A continuación, creamos una subclase personalizada de fastapi.routing.APIRoute que hará uso de GzipRequest.
Esta vez, sobrescribirá el método APIRoute.get_route_handler().
Este método devuelve una función. Y esa función es la que recibirá un request y devolverá un response.
Aquí lo usamos para crear un GzipRequest a partir del request original.
{* ../../docs_src/custom_request_and_route/tutorial001_an_py310.py hl[19:27] *}
/// note | Detalles técnicos
Un Request tiene un atributo request.scope, que es simplemente un dict de Python que contiene los metadatos relacionados con el request.
Un Request también tiene un request.receive, que es una función para "recibir" el body del request.
El dict scope y la función receive son ambos parte de la especificación ASGI.
Y esas dos cosas, scope y receive, son lo que se necesita para crear una nueva instance de Request.
Para aprender más sobre el Request, revisa la documentación de Starlette sobre Requests.
///
La única cosa que la función devuelta por GzipRequest.get_route_handler hace diferente es convertir el Request en un GzipRequest.
Haciendo esto, nuestro GzipRequest se encargará de descomprimir los datos (si es necesario) antes de pasarlos a nuestras path operations.
Después de eso, toda la lógica de procesamiento es la misma.
Pero debido a nuestros cambios en GzipRequest.body, el request body se descomprimirá automáticamente cuando sea cargado por FastAPI si es necesario.
Accediendo al request body en un manejador de excepciones
/// tip | Consejo
Para resolver este mismo problema, probablemente sea mucho más fácil usar el body en un manejador personalizado para RequestValidationError (Manejo de Errores).
Pero este ejemplo sigue siendo válido y muestra cómo interactuar con los componentes internos.
///
También podemos usar este mismo enfoque para acceder al request body en un manejador de excepciones.
Todo lo que necesitamos hacer es manejar el request dentro de un bloque try/except:
{* ../../docs_src/custom_request_and_route/tutorial002_an_py310.py hl[14,16] *}
Si ocurre una excepción, el instance de Request todavía estará en el alcance, así que podemos leer y hacer uso del request body cuando manejamos el error:
{* ../../docs_src/custom_request_and_route/tutorial002_an_py310.py hl[17:19] *}
Clase personalizada APIRoute en un router
También puedes establecer el parámetro route_class de un APIRouter:
{* ../../docs_src/custom_request_and_route/tutorial003_py310.py hl[26] *}
En este ejemplo, las path operations bajo el router usarán la clase personalizada TimedRoute, y tendrán un header X-Response-Time extra en el response con el tiempo que tomó generar el response:
{* ../../docs_src/custom_request_and_route/tutorial003_py310.py hl[13:20] *}