插件窝 干货文章 Python Sanic框架实现文件上传功能

Python Sanic框架实现文件上传功能

文件 上传 class code 49    来源:    2024-10-28

前言

Sanic是一个Python 3.5+的异步Web框架,它的设计理念与Flask相似,但采用了更高效的异步I/O处理。在处理文件上传时,Sanic同样提供了方便、高效的方法。本教程将结合实际案例,详细介绍如何在Sanic框架中实现文件上传的功能。

环境准备

在开始之前,请确保你已经安装了Python 3.5或更高版本,并安装了Sanic框架。如果没有安装Sanic,可以通过pip命令进行安装:

pip install sanic

基本概念

在Sanic中,文件上传通常是通过request.files属性来实现的。当客户端以multipart/form-data的方式发送POST请求时,可以通过request.files.get('filename')来获取指定的文件对象。

文件上传示例

1. 创建Sanic应用

首先,我们需要创建一个Sanic应用实例,并定义一个用于文件上传的路由。
首先创建个文件上传的目录

在这里插入图片描述

代码展示:

from sanic import Sanic
from sanic.response import json

app = Sanic("File-Upload-Example")


@app.route("/upload", methods=["POST"])
async def handle_upload(request):
    # 获取上传的文件,根据请求体中file字段获取上传的文件
    #先打印file对象
    print("files对象",request.files)
    file = request.files.get('file')
    # print("获取的文件对象名",file.name)
    print("查看file有哪些方法",dir(file))

    if file:
        # 对文件进行处理,这里简单地将文件保存到本地

        file_path = './uploads/' + file.name
        with open(file_path, 'wb') as f:
            #上传的文件在file的body中
            f.write(file.body)

        return json({"message": "File uploaded successfully", "filename": file.name})
    else:
        return json({"error": "No file uploaded"}, status=400)


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

2. 客户端上传文件

为了测试文件上传功能,我们需要一个客户端来发送文件。可以使用Postman这样的工具,或者使用Python的requests库来发送请求。

使用Postman

  1. 打开Postman,创建一个新的POST请求。
  2. 在请求URL中输入http://localhost:8000/upload
  3. 设置Headers中的Content-Typemultipart/form-data
  4. 在Body中选择form-data,然后点击Add file选择你要上传的文件。
  5. 然后在Value处点击Search Files选择需要上传的文件。
  6. 发送请求,并查看响应。

在body处选择 form-data类型请求,创建个key为file的键值对。

在这里插入图片描述

点击Search Files选择需要上传的文件

在这里插入图片描述

选择文件

在这里插入图片描述

点击发送请求

在这里插入图片描述

在这里插入图片描述

上传成功

在这里插入图片描述

使用Python requests库

我们现在桌面创建个txt文档

在这里插入图片描述

在这里插入图片描述

import requests

url = 'http://localhost:8000/upload'
files = {'file': open(r'C:\Users\dell\Desktop\text.txt', 'rb')}

response = requests.post(url, files=files)
print(response.text)

在这里插入图片描述

上传成功

在这里插入图片描述

文件上传的进阶用法

3. 验证上传文件的类型

为了防止用户上传不安全的文件类型(如可执行文件、脚本文件等),你需要在服务器端验证上传文件的MIME类型或文件扩展名。

from sanic import HTTPResponse, request, json

@app.route("/upload", methods=["POST"])
async def handle_upload(request):
    file = request.files.get("file")
    
    if not file:
        return json({"error": "没有文件被上传"}, status=400)
    
    # 检查文件类型
    allowed_types = {"image/jpeg", "image/png", "application/pdf"}
    if file.type not in allowed_types:
        return json({"error": "不允许的文件类型"}, status=400)
    
    # 保存文件(这里省略了具体的保存逻辑)
    
    return json({"message": "文件上传成功"})

在这个例子中,我们检查了上传文件的type属性,它通常包含了文件的MIME类型。我们定义了一个allowed_types集合,包含了允许上传的文件类型。如果上传的文件类型不在这个集合中,我们就返回一个错误响应。

4. 安全性考虑

  • 防止目录遍历:确保在保存上传文件时,不要直接使用用户输入的文件名或路径,以防止目录遍历攻击。可以使用哈希函数或UUID来生成唯一的文件名。
  • 限制上传目录的权限:确保上传目录的权限设置得当,以防止未授权访问或执行上传的文件。
  • 使用HTTPS:在生产环境中,始终使用HTTPS来保护用户数据和文件传输的安全性。

5. 进度反馈

对于大文件上传,用户可能会希望看到上传进度。然而,Sanic本身并不直接支持上传进度的跟踪。但是,你可以通过前端JavaScript(如使用XMLHttpRequest或Fetch API的upload事件)来跟踪上传进度,并通过WebSocket、Server-Sent Events (SSE) 或轮询等方式将进度信息发送到服务器,然后服务器可以将这些信息广播给需要它们的客户端。

6. 异步文件处理

如果你需要在文件上传后执行一些耗时的处理(如图像分析、视频转码等),并且这些处理不需要立即返回结果给用户,你可以将这些处理任务放入后台执行。这可以通过将任务提交到消息队列(如RabbitMQ、Kafka)或使用后台工作进程(如Celery)来实现。

7. 监控和日志

对于生产环境中的应用,监控和日志记录是非常重要的。你可以使用Sanic的日志系统来记录关键事件和错误,并使用如Prometheus、Grafana等工具来监控应用的性能。

总结

在使用Sanic进行文件上传时,除了处理文件本身之外,还需要考虑安全性、性能、用户体验等多个方面。通过验证文件类型、限制文件大小、使用异步IO、提供进度反馈以及实施适当的监控和日志记录,你可以构建出既安全又高效的文件上传功能。

以上就是pyhton Sanic框架实现文件上传功能的详细内容,更多关于pyhton Sanic文件上传的资料请关注插件窝其它相关文章!