mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-07-05 03:07:49 +02:00
Alexander Myasoedov
18 lines
570 B
Python
18 lines
570 B
Python
from fastapi import Request
|
|
from loguru import logger
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
|
|
|
|
class LogNon200ResponsesMiddleware(BaseHTTPMiddleware):
|
|
async def dispatch(self, request: Request, call_next):
|
|
try:
|
|
response = await call_next(request)
|
|
except Exception as e:
|
|
logger.exception("Yikes")
|
|
raise e
|
|
if response.status_code != 200:
|
|
logger.error(
|
|
f"{request.method} {request.url} - Status code: {response.status_code}"
|
|
)
|
|
return response
|