如何在FastAPI中實現(xiàn)檔案上傳和處理

FastAPI 是一個現(xiàn)代化的高效能Web框架，簡單易用且功能強大，它提供了原生支援檔案上傳和處理的功能。在本文中，我們將學習如何在FastAPI框架中實作檔案上傳和處理的功能，並提供程式碼範例來說明具體的實作步驟。

首先，我們需要匯入需要的函式庫和模組：

from fastapi import FastAPI, UploadFile, File
from fastapi.responses import JSONResponse
import shutil
import os

接下來，我們需要建立一個FastAPI應用程式實例：

app = FastAPI()

現(xiàn)在，我們可以定義一個路由來接受檔案上傳請求，並將檔案儲存到伺服器上：

@app.post("/upload/")
async def create_upload_file(file: UploadFile = File(...)):
    try:
        # 保存文件到服務器上
        with open(file.filename, "wb") as buffer:
            shutil.copyfileobj(file.file, buffer)
        
        # 返回成功的響應
        return JSONResponse({"message": "File uploaded successfully"})
    except Exception as e:
        # 返回失敗的響應
        return JSONResponse({"error": str(e)}, status_code=500)

在上面的程式碼中，我們定義了一個POST 請求路由/upload/，它接收一個名為file 的參數(shù)，該參數(shù)的類型是UploadFile，我們使用File 函數(shù)將其作為請求體中的檔案進行解析。 File 函數(shù)的第一個參數(shù)是檔案類型的預設值，... 表示必須傳遞該參數(shù)，否則將會傳回錯誤回應。

在處理檔案上傳請求時，我們先使用with open 的方式建立一個檔案的寫入流，然後透過shutil.copyfileobj 函數(shù)將請求體中的文件物件複製到伺服器上。

當檔案上傳成功後，我們傳回一個包含成功訊息的JSON回應；如果檔案上傳過程中出現(xiàn)任何異常，我們將傳回一個包含錯誤訊息的JSON回應並設定回應狀態(tài)碼為500。

在檔案上傳功能已經(jīng)實現(xiàn)後，我們可以繼續(xù)實作檔案處理功能。以下是一個範例路由，用於接受上傳的圖片文件，並將圖片檔案轉(zhuǎn)換為縮圖：

@app.post("/process_image/")
async def process_image(file: UploadFile = File(...)):
    try:
        # 保存文件到服務器上
        with open(file.filename, "wb") as buffer:
            shutil.copyfileobj(file.file, buffer)
        
        # 進行圖片處理，生成縮略圖
        thumbnail_filename = f"thumbnail_{file.filename}"
        # 模擬圖片處理過程
        # 請根據(jù)實際需求進行實現(xiàn)
        # ...
        
        # 返回縮略圖的下載鏈接
        return JSONResponse({"thumbnail_url": f"/download/{thumbnail_filename}"})
    except Exception as e:
        # 返回失敗的響應
        return JSONResponse({"error": str(e)}, status_code=500)

在上面的範例程式碼中，我們使用了相同的文件上傳過程，然後進入圖片處理的邏輯。在這裡，我們使用模擬的方式來處理圖片文件，產(chǎn)生縮圖，並將縮圖的下載連結(jié)傳回給客戶端。

最後，我們也可以定義一個路由來提供下載功能：

@app.get("/download/{filename}")
async def download_file(filename: str):
    try:
        # 返回文件下載鏈接
        return JSONResponse({"download_url": f"/file/{filename}"})
        
    except Exception as e:
        # 返回失敗的響應
        return JSONResponse({"error": str(e)}, status_code=500)

在上面的程式碼中，我們定義了一個GET 請求路由/download /{filename}，該路由接受一個檔案名稱參數(shù)filename，並傳回該檔案的下載連結(jié)。

到此，我們已經(jīng)實現(xiàn)了在FastAPI框架中實現(xiàn)檔案上傳和處理的功能。透過上面的範例程式碼，我們可以了解到在FastAPI中處理檔案上傳和處理的基本流程，以及如何使用FastAPID的API和方法來實現(xiàn)這些功能。當然，具體的文件處理邏輯是可以根據(jù)實際需求進行自訂的。

希望這篇文章對你理解如何在FastAPI中實現(xiàn)檔案上傳和處理功能有所幫助！

以上是如何在FastAPI中實現(xiàn)檔案上傳和處理的詳細內(nèi)容。更多資訊請關(guān)注PHP中文網(wǎng)其他相關(guān)文章！