소개

API 서버를 Flack에서 FastAPI로 마이그레이션할 기회가 있었으므로 FastAPI에서 파일 업로드 기능을 구현하는 방법에 대해 메모했습니다.

또한 FastAPI 업로드된 파일을 일단 저장하지 않고 그대로 S3에 업로드하여 업로드한 객체의 경로를 얻는 방법에 대해서도 소개하고 있습니다.

준비

FastAPI에서 파일 업로드 기능을 사용하려면 python-multipart를 설치해야 합니다.

sudo pip install python-multipart

코드

main.py

from fastapi import FastAPI, File, UploadFile

app = FastAPI()

@app.post("/file_upload")
async def file_upload(file: UploadFile = File(None)):
    s3_bucket = "バケット名"
    s3_dir = "ディレクト"
    region_name = "ap-northeast-1"

    s3 = boto3.client("s3", region_name = region_name)

    response = s3.put_object(
        Body = file.file,
        Bucket = s3_bucket,
        Key = f"{s3_dir}/{file.filename}"
    )

    # S3にアップロードしたオブジェクトのパス
    file_url = "https://%s.s3-%s.amazonaws.com/%s" % (s3_bucket, region_name, s3_dir+"/"+file.filename)

    response = {
            "file_url": file_url
        }

    return response

확인

/docs에 액세스ファイルを選択 다음 Execute

업로드에 성공하면 S3의 오브젝트 URL이 취득됩니다.


이상, FastAPIでファイルアップロードしてそのままS3にアップロードする 의 메모였습니다.

앞으로도 이러한 쁘띠 스킬도 기사로 해 나가고 싶습니다.
끝까지 봐 주셔서 감사합니다.