from fastapi import FastAPI,File, UploadFile
import shutil
from pathlib import Path
from tempfile import NamedTemporaryFile
app = FastAPI()
@app.post("/file_upload")
async def file_upload(file: UploadFile = File(...)):
# 图片文件夹
# 本地:这里是磁盘根目录开始
p = Path('/')
# 是否存在文件夹,不存在就创建一个
save_dir = p/"assets"
if not save_dir.exists():
save_dir.mkdir()
# 多级目录创建:parents=True创建父级目录,exist_ok=True父级目录存在时不抛出异常
# save_dir = p/"assets/image"
# if not save_dir.exists():
# save_dir.mkdir(parents=True,exist_ok=True)
try:
suffix = Path(file.filename).suffix
with NamedTemporaryFile(delete=False, suffix=suffix, dir=save_dir) as tmp:
shutil.copyfileobj(file.file, tmp)
tmp_file_name = Path(tmp.name).name
return {"code":0,"msg": "success","data":tmp_file_name}
except Exception as e:
return {"code":1,"msg": str(e)}
【python】图片上传至指定文件夹
于 2022-11-26 10:50:39 首次发布