在 FastAPI 中上传多部分形式的图像作为数据

Question

我正在尝试为图像上传编写端点，它将只接受 base64 格式的图像。 我已经阅读了如何从 FastAPI 中的请求中获取文件并以这种方式实现：

服务器端

@router.post("/v1/installation/{some_identifier}/image")
@db.create_connection
async def upload_installation_image(some_identifier: int, data: UploadFile = File(...)):
    ...
    content = await data.read()
    image_uuid = await save_image(installation.uuid, content, data.content_type)
    
    return {"image_uuid": image_uuid}

它工作正常，但在这种情况下，我需要以这种方式发送数据：

客户端

def test_upload_image(...):
    ...
    response = session.post(
            f"/v1/installation/{relink_serial}/image", files={"file": ("test_file_name.png", data, "image/png")}
        )
    ...

但我希望能够像这样上传图像：

客户端

def test_upload_image(...):
    ...
    response = session.post(
        f"/v1/installation/{installation_uuid}/image", data=data, headers={"content-type": "image/png"}
    )

我已经阅读了很多文章和其他问题，但他们都建议使用 UploadFile 并将数据作为 json 与file或body参数发送。 这是否可能，像我展示的第二个测试一样只加载 1 个文件？

Answer 1

我找到了解决方案。 我可以直接从请求 object 中获取所有数据，而不是使用预定义的 FastAPI 类型，所以我的代码如下所示：

@router.post("/v1/installation/{some_identifier}/image")
@db.create_connection
async def upload_installation_image(some_identifier: int, request: Request):
    ...
    content = await request.body()
    image_uuid = await save_image(installation.uuid, content, content_type)
    
    return {"image_uuid": image_uuid}

这正是我正在寻找的，因为在我的案例中，客户端应用程序已将整个文件发送给我。