在 router.post fastapi 中捕获唯一 ID

Question

我正在尝试使用以下方法在 python 中使用 fastapi 捕获 api 所花费的时间：

import string, time, random, logging
from starlette.requests import Request

@app.middleware("http")
async def log_requests(request: Request, call_next):
    idem = ''.join(random.choices(string.ascii_uppercase + string.digits, k=6))
    logger.info(f"rid={idem} start request path={request.url.path}")
    start_time = time.time()
    response = await call_next(request)
    process_time = (time.time() - start_time) * 1000
    formatted_process_time = '{0:.2f}'.format(process_time)
    logger.info(f"rid={idem} completed_in={formatted_process_time}ms status_code={response.status_code}")
    return response

我想在@router.post部分中捕获唯一的id idem ，如下所示，但它给出了错误：

import logging as log

@router.post("/infer/", response_description="AD Data Infer")
async def ad_infer_algorithm(ad_infer_config: ADInferConfigSchema = Body(...)):
  log.info("Executing AD Train Algorithm with config"+idem)
  return "2"

请帮助

Answer 1

FastAPI 在request object 上有一个state属性。 这允许您将 state 附加到来自中间件的请求，然后在 controller（或其他中间件）中访问该 state。

由于您的中间件中已经有request object 可用，您只需将idem附加到请求 state 那里：

async def log_requests(request: Request, call_next):
    idem = ''.join(random.choices(string.ascii_uppercase + string.digits, k=6))
    request.state.idem = idem

（旁注：您可能可以将random.choices调用替换为secrets.token_urlsafe以获得随机生成的标识符（除非您需要控制大写和小写标识符 - 因为您没有包含 ascii_lowercase）。

然后，您可以在视图 controller 中访问该 state：

@router.post("/infer/", response_description="AD Data Infer")
async def ad_infer_algorithm(request: Request, ad_infer_config: ADInferConfigSchema = Body(...)):
  log.info("Executing AD Train Algorithm with config" + request.state.idem)
  return "2"

（另一个注意事项：日志记录命令通常应使用占位符（ %s ），然后将变量用作 arguments： log.info("Executing AD Train Algorithm with config %s", request.state.idem)