86 lines
2.1 KiB
Python
86 lines
2.1 KiB
Python
|
from json import JSONDecodeError
|
||
|
|
from fastapi import Request, status
|
||
|
|
from fastapi.responses import JSONResponse
|
||
|
|
from fastapi.encoders import jsonable_encoder
|
||
|
|
from fastapi.exceptions import RequestValidationError
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
try:
|
||
|
|
from Context.common import uvicorn_log
|
||
|
|
except Exception as e:
|
||
|
|
pass
|
||
|
|
uvicorn_log = None
|
||
|
|
|
||
|
|
|
||
|
|
class ReqData(BaseModel):
|
||
|
|
url: str = ""
|
||
|
|
method: str = ""
|
||
|
|
client_ip: str = ""
|
||
|
|
body: bytes = b''
|
||
|
|
data: dict = {}
|
||
|
|
|
||
|
|
|
||
|
|
async def set_body(request: Request):
|
||
|
|
receive_ = await request._receive()
|
||
|
|
|
||
|
|
async def receive():
|
||
|
|
return receive_
|
||
|
|
|
||
|
|
request._receive = receive
|
||
|
|
|
||
|
|
|
||
|
|
async def format_request(req: Request) -> ReqData:
|
||
|
|
url = req.url
|
||
|
|
method = req.method
|
||
|
|
client_ip = f"{req.client[0]}:{req.client[1]}"
|
||
|
|
|
||
|
|
req_data = ReqData()
|
||
|
|
req_data.url = str(url)
|
||
|
|
req_data.method = method
|
||
|
|
req_data.client_ip = client_ip
|
||
|
|
body=b''
|
||
|
|
data = {}
|
||
|
|
try:
|
||
|
|
await set_body(req)
|
||
|
|
body = await req.body()
|
||
|
|
try:
|
||
|
|
data = await req.json()
|
||
|
|
except JSONDecodeError:
|
||
|
|
data = {}
|
||
|
|
except RuntimeError:
|
||
|
|
pass
|
||
|
|
req_data.body = body
|
||
|
|
req_data.data = data
|
||
|
|
return req_data
|
||
|
|
|
||
|
|
|
||
|
|
async def exception_handler(req: Request, exc: Exception):
|
||
|
|
req_data = await format_request(req)
|
||
|
|
if uvicorn_log:
|
||
|
|
uvicorn_log.info("请求错误")
|
||
|
|
uvicorn_log.info(req_data)
|
||
|
|
else:
|
||
|
|
print("请求错误")
|
||
|
|
print(req_data)
|
||
|
|
raise exc
|
||
|
|
|
||
|
|
|
||
|
|
async def validation_exception_handler(req: Request, exc: RequestValidationError):
|
||
|
|
req_data = await format_request(req)
|
||
|
|
if uvicorn_log:
|
||
|
|
uvicorn_log.info("请求错误")
|
||
|
|
uvicorn_log.info(req_data)
|
||
|
|
else:
|
||
|
|
print("请求错误")
|
||
|
|
print(req_data)
|
||
|
|
return JSONResponse(
|
||
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
||
|
|
content=jsonable_encoder({"detail": exc.errors(),
|
||
|
|
"body": exc.body}))
|
||
|
|
|
||
|
|
|
||
|
|
async def logger_request(req: Request, call_next):
|
||
|
|
req_data = await format_request(req)
|
||
|
|
response = await call_next(req)
|
||
|
|
return response
|