基于FastAPI的文件上传和下载

基于FastAPI的文件上传和下载

一、前言

为了实现ASR的可视化界面,在各个博客中寻觅了一波找找文件上传和下载的例子,没有找到能完整实现这个功能的,有也只是有一部分(菜菜求捞捞),看了甚是烦恼,后来找gpt大爷来回交互了几下,把功能实现了。记录一下过程。

二、实现步骤

对了,能来查这个问题的兄弟,默认你对fastapi有了解了hh,就不多介绍了,单刀直入主题,怎么实现功能。

1、环境配置

这个功能实现涉及到三个库: fastapi、uvicorn、 requests。

pip install fastapi uvicorn requests -i

https://pypi.tuna.tsinghua.edu.cn/simple

2、服务端代码

文件名:upload.py

python 复制代码
import os.path

import uvicorn

from fastapi import FastAPI, File, UploadFile
from fastapi.responses import FileResponse

app = FastAPI()

@app.get("/")
async def hello():

    return {"ret": 'hello'}

@app.post("/uploadfile/")
async def create_upload_file(file: UploadFile = File(...)):
    print(file)
    if not os.path.exists('uploaded_files'):
        os.mkdir('uploaded_files')
    with open(f"uploaded_files/{file.filename}", "wb") as f:
        f.write(await file.read())
    return {"filename": file.filename}

@app.get("/downloadfile/{filename}")
async def download_file(filename: str):
    file_path = f"uploaded_files/{filename}"
    return FileResponse(file_path, media_type="application/octet-stream", filename=filename)


if __name__ == '__main__':
    uvicorn.run('upload:app', host='127.0.0.1', port=18005, reload=True)

运行结果参考:

3、客户端代码

python 复制代码
import os

import requests
def upfile(file_path):
    url = "http://127.0.0.1:18005/uploadfile/"

    try:
        with open(file_path, "rb") as f:
            files = {"file": (file_path, f)}
            response = requests.post(url, files=files)

        if response.status_code == 200:
            print(response.json())
            return True
        else:
            print(f"Request failed with status code {response.status_code}")
            return False
    except Exception as e:
        print(e)
        return False
def download(fileName):
    url = "http://127.0.0.1:18005/downloadfile/{}".format(fileName)
    try:
        res=requests.get(url)
        if res.status_code==200:
            with open(fileName,'wb') as w:
                w.write(res.content)

            return True
        else:

            return False
    except Exception as e:
        print(e)
        return False
def main():
    file_path = r"C:\Users\Admin\Pictures\LesNo2Text.png"

    upfile(file_path)
    fileName=os.path.basename(file_path)
    print('待下载的文件名字:'.format(fileName))
    ret=download(fileName)
    if ret:
        print('下载成功:' + fileName)
    else:
        print('下载失败:'+fileName)
if __name__=="__main__":
    main()

运行结果参考:
到这里就结束了,有遇到什么bug欢迎在下面评论区提,相对其他博主我还是学生相对活跃一些hh。

相关推荐
夏末蝉未鸣012 分钟前
python transformers笔记(TrainingArguments类)
python·自然语言处理·transformer
德育处主任Pro7 分钟前
「py数据分析」04如何将 Python 爬取的数据保存为 CSV 文件
数据库·python·数据分析
咸鱼鲸17 分钟前
【PyTorch】PyTorch中数据准备工作(AI生成)
人工智能·pytorch·python
遇见你很高兴37 分钟前
Pycharm中体验通义灵码来AI辅助编程
python
大虫小呓42 分钟前
50个Python处理Excel示例代码,覆盖95%日常使用场景-全网最全
python·excel
大模型真好玩1 小时前
做题王者,实战拉跨!是时候给马斯克的Grok4泼盆冷水了!(Grok 4模型详细测评报告)
人工智能·python·mcp
羊八井1 小时前
使用 Earth2Studio 和 AI 模型进行全球天气预测:太阳辐照
pytorch·python·nvidia
向左转, 向右走ˉ1 小时前
PyTorch随机擦除:提升模型抗遮挡能力
人工智能·pytorch·python·深度学习
马特说1 小时前
金融时间序列机器学习训练前的数据格式验证系统设计与实现
python·机器学习·金融
Blue桃之夭夭1 小时前
基于OpenCV的实时人脸检测系统实现指南 ——Python+Haar级联分类器从环境搭建到完整部署
人工智能·python·opencv