FastAPI 使用Python进行文件的发送和接收：FastAPI和requests

在本文中，我们将介绍如何使用FastAPI和requests库在Python中发送和接收文件。FastAPI是一个高性能的Web框架，可以快速构建API。它基于Python 3.7+的类型提示和异步技术，提供了强大的自动化文档和验证功能。requests库是一个常用的HTTP库，用于发送HTTP请求。

阅读更多：FastAPI 教程

发送文件

要发送文件，我们首先需要创建一个POST请求，并在请求中包含文件数据。使用FastAPI可以方便地实现这一功能。首先，我们需要导入必要的库和模块：

from fastapi import FastAPI, File, UploadFile
import requests

接下来，我们创建一个FastAPI应用程序的实例：

app = FastAPI()

然后，我们定义一个/upload的POST路径，用于接收文件并将其发送到指定的URL：

@app.post("/upload")
async def upload_file(file: UploadFile = File(...)):
    contents = await file.read()
    response = requests.post(url, files={"file": (file.filename, contents)})
    return response.text

在这个例子中，我们使用UploadFile和File这两个FastAPI提供的类来处理上传的文件。File类用于定义文件字段，而UploadFile类负责处理上传的文件数据。

在/upload路径的处理函数中，我们首先通过await file.read()读取文件的内容。然后，我们使用requests库的post方法发送包含文件数据的POST请求。文件数据以字典形式传递给files参数，其中文件字段的键为”file”，值为元组，元组的第一个元素为文件名，第二个元素为文件内容。

最后，我们返回请求的响应结果。

接收文件

要接收文件，我们需要定义一个POST请求，用于接收文件数据并进行处理。假设我们有一个用于处理上传文件的接口/receive：

@app.post("/receive")
async def receive_file(file: UploadFile = File(...)):
    contents = await file.read()
    # 对接收到的文件内容进行处理
    ...
    return {"message": "文件已接收"}

在这个例子中，我们仍然使用UploadFile和File这两个FastAPI提供的类来处理上传的文件。

在/receive路径的处理函数中，我们通过await file.read()读取文件的内容，并对接收到的文件内容进行处理。具体的处理逻辑可以根据实际需求进行编写。

最后，我们返回一个包含成功消息的响应。

示例

现在，让我们来看一个完整的示例。假设我们有一个文件上传服务，客户端可以通过FastAPI发送文件，并在服务器端接收并保存文件。

客户端的代码：

import requests

url = "http://example.com/upload"
files = {"file": open("example.txt", "rb")}
response = requests.post(url, files=files)

print(response.text)

服务器端的代码：

from fastapi import FastAPI, File, UploadFile
import requests

app = FastAPI()

@app.post("/upload")
async def upload_file(file: UploadFile = File(...)):
    contents = await file.read()
    response = requests.post(url, files={"file": (file.filename, contents)})
    return response.text

@app.post("/receive")
async def receive_file(file: UploadFile = File(...)):
    contents = await file.read()
    # 对接收到的文件内容进行处理
    ...
    return {"message": "文件已接收"}

通过上述代码，客户端可以将example.txt文件发送到服务器的/upload路径，并返回上传成功的消息。服务器可以通过/receive路径接收文件，并进行处理。

总结

本文介绍了如何使用FastAPI和requests库在Python中发送和接收文件。通过FastAPI，我们可以轻松地创建用于处理文件上传和接收的API路径。借助requests库的post方法，我们可以方便地发送和接收具有文件数据的POST请求。通过这些技术，我们可以构建出强大的文件上传和处理服务。