fastapi文件上传

最新推荐文章于 2024-05-24 22:13:06 发布

文子阳

最新推荐文章于 2024-05-24 22:13:06 发布

阅读量546

点赞数

分类专栏： python 文章标签： fastapi python 开发语言

本文链接：https://blog.csdn.net/wenxingchen/article/details/128918859

版权

python 专栏收录该内容

127 篇文章 10 订阅

订阅专栏

@app.post("/ai_read_meter")
async def refresh_capacity_energy_db(request: Request, file: bytes = File(...)):
    BASE_DIR = Path(__file__).resolve().parent

    """单文件上传（不能大于5M）"""
    if len(file) > 5 * 1024 * 1024:
        raise HTTPException(status_code=400, detail="每个文件都不能大于5M")
    name = hashlib.md5(file).hexdigest() + ".png"  # 使用md5作为文件名，以免同一个文件多次写入
    subpath = "media/uploads"
    # 如果不存在文件夹，则创建
    if not (folder := BASE_DIR / subpath).exists():
        await AsyncPath(folder).mkdir(parents=True)
    # 如果不存在文件，则写入
    if not (fpath := folder / name).exists():
        await AsyncPath(fpath).write_bytes(file)

    # image = np.asarray(bytearray(file), dtype="uint8")
    # start(i=2, image=cv2.imdecode(image, cv2.IMREAD_COLOR))

    return f"{subpath}/{name}"