如何使用Flask-Uploads来处理文件上传和下载

2301_78316786

于 2023-08-02 23:06:33 发布

阅读量674

点赞数

分类专栏： python 文章标签： flask python 后端

本文链接：https://blog.csdn.net/2301_78316786/article/details/132052795

版权

python 专栏收录该内容

99 篇文章 1 订阅

订阅专栏

首先，您需要安装 Flask 和 Flask-Uploads。使用 pip 非常容易，只需运行以下命令：

pip install Flask Flask-Uploads

然后，导入必要的库并设置应用程序：

from flask import Flask, render_template, request, redirect, url_for  
from flask_uploads import UploadSet, configure_uploads, patch_request_class  
  
app = Flask(__name__)  
  
# 配置文件上传  
app.config['UPLOADS_DEFAULT_DEST'] = 'static/uploads'  
app.config['UPLOADS_DEFAULT_URL'] = '/static/uploads'  
app.config['UPLOADSet'] = ('my_uploads', 'uploads', 'all')  
  
# 初始化上传套件  
my_uploads = UploadSet(app)  
configure_uploads(app, my_uploads)  
patch_request_class(app)

现在，您需要定义一个简单的路由，以便用户可以上传文件：

@app.route('/upload', methods=['GET', 'POST'])  
def upload():  
    if request.method == 'POST':  
        # 从表单中获取文件  
        file = request.files['file']  
        # 将文件保存到上传目录中  
        file.save(my_uploads.join('uploads', file.filename))  
        # 重定向到成功页面  
        return redirect(url_for('uploaded'))  
    # 如果是 GET 请求，则显示上传表单  
    return render_template('upload.html')

但是，在此之前，您需要创建另一个路由来显示已上传的文件：

@app.route('/uploads/<filename>')  
def uploaded(filename):  
    return render_template('uploaded.html', filename=filename)

接下来，您需要创建上传表单的 HTML 模板。在 Flask 中，通常将模板保存在 templates 目录中。在这个例子中，我们将文件上传表单模板保存在 upload.html 中，并将已上传文件的模板保存在 uploaded.html 中。

这是 upload.html 的内容：

<!DOCTYPE html>  
<html>  
  <head>  
    <title>文件上传</title>  
  </head>  
  <body>  
    <h1>上传文件</h1>  
    <form method="POST" enctype="multipart/form-data">  
      <input type="file" name="file">  
      <input type="submit" value="上传">  
    </form>  
  </body>  
</html>

这是 uploaded.html 的内容：

<!DOCTYPE html>  
<html>  
  <head>  
    <title>已上传文件</title>  
  </head>  
  <body>  
    <h1>已上传文件</h1>  
    <p>您已上传的文件是：{{ filename }}</p>  
  </body>  
</html>

现在，您的 Flask-Uploads 应用程序已经准备就绪，可以开始上传和下载文件了！

以上是使用 Flask-Uploads 处理文件上传和下载的基本步骤和代码示例。下面我们再深入一步，介绍一些扩展功能和用法。

自定义上传目录
默认情况下，上传的文件会被保存在应用程序的静态目录中。如果您需要将文件保存在其他目录中，可以设置 UploadSet 的自定义目录：

my_uploads = UploadSet(app, 'custom_dir')

这将上传文件保存到名为 “custom_dir” 的目录中。

自定义文件名
默认情况下，上传的文件会保留原始文件名。如果您需要为文件指定一个新的文件名，可以使用 Flask-Uploads 的 save() 方法，并传递一个新的文件名作为参数：

file.save('custom_filename.txt')

这将上传文件保存到上传目录中，并使用 “custom_filename.txt” 作为文件名。

限制文件大小
如果您需要限制上传文件的大小，可以在表单中添加一个 input 元素，并设置 name 属性为 MAX_FILE_SIZE，例如：

<input type="hidden" name="MAX_FILE_SIZE" value="10485760">

这将限制用户上传的文件大小不能超过 10MB。

上传多个文件
如果您需要允许用户上传多个文件，可以在表单中添加多个 input 元素，并设置 name 属性为 file，例如：

<input type="file" name="file">  
<input type="file" name="file">  
<input type="file" name="file">

然后，在 Flask 路由中，您可以使用 request.files 获取所有上传的文件，并使用循环遍历它们：

for file in request.files.getlist('file'):  
    # 处理每个文件

使用 Flask-Uploads 蓝图
如果您使用 Flask-Uploads 的蓝图，可以将上传文件保存到不同的目录中，以便更好地组织应用程序的文件。例如，您可以在蓝图目录中创建一个名为 “uploads” 的目录，然后在应用程序配置中设置 UPLOADSet 为 (‘my_uploads’, ‘uploads’, ‘all’)。这将将上传文件保存到蓝图目录中的 “uploads” 目录中。

自定义 URL
如果您需要为上传的文件生成自定义的 URL，可以在应用程序配置中设置 UPLOADSet 的 url 参数：

app.config['UPLOADS_DEFAULT_URL'] = 'http://example.com/uploads/%(path)s'

这将为上传的文件生成自定义的 URL，例如 http://example.com/uploads/myfile.txt。

自定义文件名转换器
如果您需要自定义文件名转换器，可以继承 FileConverter 类，并实现自己的转换器方法。例如，以下代码将文件名中的空格替换为下划线：

from flask_uploads import FileConverter  
  
class MyFileConverter(FileConverter):  
    @staticmethod  
    def rename(filename):  
        return filename.replace(' ', '_')

然后，在应用程序配置中注册自定义转换器：

app.config['UPLOADS_DEFAULT_FILENAME_Converter'] = MyFileConverter

这将使用自定义的转换器重命名上传的文件。

以上是使用 Flask-Uploads 处理文件上传和下载的一些扩展功能和用法。根据您的具体需求，您可以根据自己的情况进一步定制和扩展 Flask-Uploads 应用程序。

2301_78316786

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
如何使用Flask-Uploads来处理文件上传和下载

如果您使用 Flask-Uploads 的蓝图，可以将上传文件保存到不同的目录中，以便更好地组织应用程序的文件。例如，您可以在蓝图目录中创建一个名为 “uploads” 的目录，然后在应用程序配置中设置 UPLOADSet 为 (‘my_uploads’, ‘uploads’, ‘all’)。在这个例子中，我们将文件上传表单模板保存在 upload.html 中，并将已上传文件的模板保存在 uploaded.html 中。默认情况下，上传的文件会被保存在应用程序的静态目录中。
复制链接

扫一扫