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

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

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

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

接下來(lái)，我們需要建立一個(gè)FastAPI應(yīng)用程式實(shí)例：

app = FastAPI()

現(xiàn)在，我們可以定義一個(gè)路由來(lái)接受檔案上傳請(qǐng)求，並將檔案儲(chǔ)存到伺服器上：

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

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

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

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

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

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

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

最後，我們也可以定義一個(gè)路由來(lái)提供下載功能：

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

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

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

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

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