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

首先,您需要安装 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/')  
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 应用程序。

你可能感兴趣的:(python,flask,python,后端)