FastAPI 如何在FastAPI端点中返回一个字典和一张图片

在本文中，我们将介绍如何在FastAPI端点中返回一个字典和一张图片。FastAPI是一个基于Python的现代化Web框架，具有快速、易用和强大的特性。它被广泛用于构建高性能的API，并提供了丰富的功能和扩展性。

阅读更多：FastAPI 教程

返回一个字典

要在FastAPI端点中返回一个字典，我们可以使用Python的字典数据结构，并将其作为响应的主体返回。以下是一个示例代码：

from fastapi import FastAPI

app = FastAPI()

@app.get("/data")
async def get_data():
    data = {"name": "John", "age": 30}
    return data

上面的代码创建了一个FastAPI应用程序，并定义了一个GET端点/data，该端点返回一个字典。在这个示例中，字典中包含了名为”name”和”age”的键值对。当我们访问/data端点时，将返回这个字典作为响应的主体。

返回一张图片

要在FastAPI端点中返回一张图片，我们可以使用FileResponse类，并将图片的二进制数据作为响应的主体返回。以下是一个示例代码：

from fastapi import FastAPI
from fastapi.responses import FileResponse

app = FastAPI()

@app.get("/image")
async def get_image():
    image_path = "path_to_image.jpg"
    return FileResponse(image_path, media_type="image/jpeg")

在上面的示例中，我们使用FileResponse类来返回一张图片。我们需要指定图片的路径image_path，并通过media_type参数设置响应的媒体类型为”image/jpeg”。在访问/image端点时，将返回指定路径的图片作为响应的主体。

返回一个字典和一张图片

要在FastAPI端点中同时返回一个字典和一张图片，我们可以将字典和图片的响应包装在一个Response对象中。以下是一个示例代码：

from fastapi import FastAPI
from fastapi.responses import FileResponse
from fastapi import Response

app = FastAPI()

@app.get("/data-and-image")
async def get_data_and_image():
    data = {"name": "John", "age": 30}
    image_path = "path_to_image.jpg"
    response = Response(content=data, media_type="application/json")
    response.headers["Content-Disposition"] = 'attachment; filename="data.json"'
    response.body = open(image_path, "rb").read()
    response.media_type = "image/jpeg"
    return response

在上面的示例中，我们首先创建了一个data字典和一个image_path变量来代表图片的路径。然后，我们创建了一个Response对象，并将字典和图片的响应内容分别设置为其content和body属性。我们还通过media_type参数将content的媒体类型设置为”application/json”，将body的媒体类型设置为”image/jpeg”。最后，我们将response对象作为响应的主体返回。

总结

在本文中，我们介绍了在FastAPI端点中返回一个字典和一张图片的方法。我们首先使用字典的数据结构来返回一个字典，然后使用FileResponse类来返回一张图片。最后，我们演示了如何同时返回一个字典和一张图片，通过将它们的响应包装在一个Response对象中。通过这些方法，我们可以轻松地在FastAPI应用程序中返回丰富和多样的响应。