FastAPI 在 FastAPI 中无法读取上传的 CSV 文件

在本文中，我们将介绍在 FastAPI 中无法读取上传的 CSV 文件的问题，并提供解决方案和示例代码。

阅读更多：FastAPI 教程

问题描述

在使用 FastAPI 开发 Web 应用程序时，有时我们需要上传文件并对其进行处理。然而，在处理上传的 CSV 文件时，可能会遇到无法读取文件内容的问题。

解决方案

要解决在 FastAPI 中无法读取上传的 CSV 文件的问题，我们可以使用 csv 模块来处理文件。下面是一个示例代码，演示了如何读取上传的 CSV 文件内容：

from fastapi import FastAPI, UploadFile, File
import csv

app = FastAPI()

@app.post("/upload")
async def upload_file(file: UploadFile = File(...)):
    if file.content_type == "text/csv":
        content = await file.read()
        decoded_content = content.decode("utf-8")

        reader = csv.reader(decoded_content.splitlines(), delimiter=',')
        rows = list(reader)

        # 对文件内容进行处理，这里只是打印每一行的内容
        for row in rows:
            print(row)

        return {"message": "File uploaded successfully."}
    else:
        return {"message": "Invalid file format. Only CSV files are allowed."}

在上面的代码中，我们定义了一个 /upload 路由，用于接收上传的文件。在该路由中，我们首先检查文件的内容类型是否为 CSV，如果是则读取文件内容，并将其解码为 UTF-8 编码。然后，我们使用 csv.reader() 函数将文件内容转换为行列表。最后，我们对每一行的内容进行处理，这里只是简单地打印每一行的内容。

需要注意的是，这里我们假设所有上传的文件都是 CSV 格式的，并且使用了默认的 CSV 分隔符（逗号）。如果需要处理其他格式的文件或使用不同的分隔符，可以相应地修改代码。

示例

假设我们有一个名为 data.csv 的 CSV 文件，内容如下：

Name,Age,Gender
John,25,Male
Alice,32,Female

我们可以使用如下的 cURL 命令将该文件上传到 FastAPI 服务器：

curl -X POST -F "file=@data.csv" localhost:8000/upload

在上传成功后，我们可以在 FastAPI 服务器的终端输出中看到每一行的内容：

['Name', 'Age', 'Gender']
['John', '25', 'Male']
['Alice', '32', 'Female']

这表明我们成功读取了上传的 CSV 文件，并可以对其内容进行进一步处理。

总结

本文介绍了在 FastAPI 中无法读取上传的 CSV 文件的问题，并给出了解决方案和示例代码。我们可以使用 csv 模块来处理上传的 CSV 文件，并对文件内容进行进一步操作。通过本文的指导，您将能够在 FastAPI 中正确读取和处理上传的 CSV 文件。