fastapi 上传文件保存 - Python (1)

📌 相关文章

📜 fastapi 上传文件保存 - Python (1)

📅 最后修改于: 2023-12-03 15:30:44.788000 🧑 作者: Mango

FastAPI 上传文件保存 - Python

在 FastAPI 中，可以方便地使用 UploadFile 类来处理上传文件。同时，也可以使用 asyncio 库中的协程来异步处理上传的文件，并将其保存到磁盘上。

安装 FastAPI

首先，需要安装 FastAPI 和 uvicorn，可以使用 pip 来进行安装：

$ pip install fastapi uvicorn

上传文件处理

为了处理上传的文件，需要在路由函数中添加 File 类型参数，然后使用 save 方法将文件保存到磁盘上。下面是一个上传文件的示例程序：

from fastapi import FastAPI, File, UploadFile

app = FastAPI()

@app.post("/files/")
async def create_file(file: UploadFile = File(...)):
    with open(file.filename, "wb") as buffer:
        buffer.write(await file.read())
    return {"filename": file.filename}

在上面的示例中，create_file 函数接收一个名为 file 的参数，并使用 await file.read() 读取文件内容。然后，将文件内容写入到磁盘上。

异步写入文件

为了提高性能，可以使用 asyncio 库中的协程来异步处理读写文件。下面是一个使用协程异步保存文件的示例程序：

import asyncio
from fastapi import FastAPI, File, UploadFile

app = FastAPI()

async def save_upload_file(file: UploadFile) -> str:
    with open(file.filename, "wb") as buffer:
        while True:
            data = await file.read(1024)
            if not data:
                break
            buffer.write(data)
    return file.filename

@app.post("/files/")
async def create_file(file: UploadFile = File(...)):
    loop = asyncio.get_event_loop()
    return await loop.run_in_executor(None, save_upload_file, file)

在上面的示例中，使用异步函数 save_upload_file 来保存上传的文件。该函数使用协程来异步读写文件，并返回文件名。

然后，从 create_file 函数中使用 run_in_executor 方法来调用异步函数 save_upload_file。该方法将异步函数传递给第一个参数，然后在协程池中运行异步函数并返回结果。

Markdown

本文介绍了如何使用 FastAPI 处理上传文件，并异步保存文件。代码示例已经在本文中给出。

# FastAPI 上传文件保存 - Python

在 FastAPI 中，可以方便地使用 `UploadFile` 类来处理上传文件。同时，也可以使用 `asyncio` 库中的协程来异步处理上传的文件，并将其保存到磁盘上。

## 安装 FastAPI

首先，需要安装 FastAPI 和 uvicorn，可以使用 pip 来进行安装：

```bash
$ pip install fastapi uvicorn