FastAPI, 파일 업로드와 다운로드 API 예제

FastAPI, 파일 업로드와 다운로드 API 예제 

 

https://blog.jcharistech.com/2022/12/11/create-an-upload-and-download-files-api-endpoint-with-fastapi/

Create an Upload and Download Files API Endpoint with FastAPI

 

https://www.youtube.com/watch?v=m6Ma6B6VlFs 

 

 

import time
from fastapi import FastAPI, UploadFile
from fastapi.exceptions import HTTPException
from fastapi.responses import FileResponse
import uvicorn
import json
import os
import yaml


app = FastAPI()

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
timestr = time.strftime("%Y%m%d-%H%M%S")

@app.get("/")
def read_root():
    return {"Hello": "FastAPI"}


@app.post("/file/upload")
def upload_file(file: UploadFile):
    if file.content_type != "application/json":
        raise HTTPException(400, detail="Invalid document type")
    else:
        data = json.loads(file.file.read())
    return {"content": data, "filename": file.filename}


@app.post("/file/uploadndownload")
def upload_n_downloadfile(file: UploadFile):
    """Return a YAML FIle for the uploaded JSON File"""
    if file.content_type != "application/json":
        raise HTTPException(400, detail="Invalid document type")
    else:
        json_data = json.loads(file.file.read())
        new_filename = "{}_{}.yaml".format(os.path.splitext(file.filename)[0], timestr)
        # Write the data to a file
        # Store the saved file
        SAVE_FILE_PATH = os.path.join(UPLOAD_DIR, new_filename)
        with open(SAVE_FILE_PATH, "w") as f:
            yaml.dump(json_data, f)

        # Return as a download
        return FileResponse(
            path=SAVE_FILE_PATH,
            media_type="application/octet-stream",
            filename=new_filename,
        )


if __name__ == "__main__":
    uvicorn.run(app, host="127.0.0.1", port=8000)