FastAPI搭建文件上传服务器

怼烎@ 2023-02-20 12:28 383阅读 0赞

本文共435个字，预计阅读时间需要2分钟。  
**FastAPI** 是一个高性能 Web 框架，也是一个Python包，用于构建 API，适合利用极少的代码搭建服务器后端，实现前后端分离。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h5aXN2_size_16_color_FFFFFF_t_70_pic_center]

下面给出了一个任务：利用FastAPI搭建文件上传服务器，给出上传接口，并保存到服务器指定位置。

**需要使用的Python包：fastapi和uvicorn。**

## 服务器代码 ##

其中with open(file.filename, “wb”)是将客户上传的文件保存起来，上传的url地址为host:port/file\_upload，可以自定义端口和host。

注意如果在前端配置的时候出现跨域问题需要添加FastAPI跨域规则。

import time
    import uvicorn
    from fastapi import FastAPI, File, UploadFile
    
    app = FastAPI()
    
    
    @app.post("/file_upload")
    async def file_upload(file: UploadFile = File(...)):
        start = time.time()
        try:
            res = await file.read()
            with open(file.filename, "wb") as f:
                f.write(res)
            return {
        "message": "success", 'time': time.time() - start, 'filename': file.filename}
        except Exception as e:
            return {
        "message": str(e), 'time': time.time() - start, 'filename': file.filename}
    
    
    if __name__ == '__main__':
        uvicorn.run(app=app, host="127.0.0.1", port=8000, workers=1)

## 客户端代码示例 ##

此客户端的需要上传的文件在path中，url即为文件上传的API。

import requests
    url = "http://127.0.0.1:8000/file_upload"
    path = "C:\\Users\\omegaxyz\\Desktop\\pic.jpeg"
    files = {
        'file': open(path, 'rb')}
    r = requests.post(url, files=files)
    print(r.url)
    print(r.text)

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h5aXN2_size_16_color_FFFFFF_t_70_pic_center]: https://img-blog.csdnimg.cn/20200627101328816.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3h5aXN2,size_16,color_FFFFFF,t_70#pic_center
[omegaxyz.com]: https://www.omegaxyz.com/