Flask 文件上传限制

在本文中，我们将介绍如何在 Flask 中设置文件上传的限制。

阅读更多：Flask 教程

什么是文件上传限制？

文件上传限制是指在使用网站或应用程序时，设置了上传文件的大小、类型或数量的限制。这是为了确保服务器的安全性和性能，并避免恶意用户滥用系统资源。

Flask 文件上传限制的默认设置

在 Flask 中，默认情况下，文件上传的限制是非常宽松的。它允许上传最大为16MB的任意类型文件。此外，Flask 还提供了方便的方法来访问上传的文件。

下面是一个简单的示例，演示了如何在 Flask 中接收文件上传：

from flask import Flask, request

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_file():
    file = request.files['file']
    file.save('uploads/' + file.filename)
    return '文件上传成功！'

if __name__ == '__main__':
    app.run()

在上面的示例中，我们创建了一个 /upload 的路由，使用 POST 方法接收文件上传。通过 request.files 我们可以访问上传的文件并保存到服务器的指定目录中。

然而，这种默认的设置并不适合所有的应用场景。有时候我们希望限制上传文件的大小、类型或数量，以避免服务器的滥用。

限制文件上传的大小

为了限制上传文件的大小，可以使用 Flask-WTF 扩展包提供的 FileAllowed 和 FileSize 验证。首先，我们需要安装 Flask-WTF：

pip install flask-wtf

然后，我们可以在表单中使用这些验证规则：

from flask import Flask, render_template
from flask_wtf import FlaskForm
from wtforms import FileField
from wtforms.validators import FileAllowed, FileSize

app = Flask(__name__)
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 设置上传文件的最大限制为16MB

class UploadForm(FlaskForm):
    file = FileField('上传文件', validators=[FileAllowed(['txt', 'pdf']), FileSize(5 * 1024 * 1024)])

@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
    form = UploadForm()
    if form.validate_on_submit():
        file = form.file.data
        file.save('uploads/' + file.filename)
        return '文件上传成功！'
    return render_template('upload.html', form=form)

if __name__ == '__main__':
    app.run()

在上面的代码中，我们通过设置 app.config['MAX_CONTENT_LENGTH'] 来限制上传文件的大小，这里将最大限制设置为16MB。同时，我们在 UploadForm 表单类中使用了 FileAllowed 和 FileSize 验证器来限制上传的文件类型和大小。

在HTML模板中，我们可以使用 jinja2 模板语言来显示表单并处理验证错误。

限制文件上传的类型

为了限制上传文件的类型，我们可以使用 Flask-WTF 提供的 FileAllowed 验证器。它允许我们指定允许上传的文件类型列表。

下面是一个示例代码，演示了如何限制上传文件的类型为图像文件：

from flask import Flask, render_template
from flask_wtf import FlaskForm
from wtforms import FileField
from wtforms.validators import FileAllowed

app = Flask(__name__)

class UploadForm(FlaskForm):
    file = FileField('上传文件', validators=[FileAllowed(['jpg', 'jpeg', 'png'])])

@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
    form = UploadForm()
    if form.validate_on_submit():
        file = form.file.data
        file.save('uploads/' + file.filename)
        return '文件上传成功！'
    return render_template('upload.html', form=form)

if __name__ == '__main__':
    app.run()

在上面的代码中，我们使用了 FileAllowed 验证器，并指定了允许上传的文件类型为 jpg、jpeg 和 png。

限制文件上传的数量

为了限制上传文件的数量，我们可以在保存文件之前，检查 request.files 的长度。

下面是一个示例代码，演示了如何限制一次只能上传一个文件：

from flask import Flask, request

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_file():
    files = request.files.getlist('file')
    if len(files) != 1:
        return '一次只能上传一个文件！'
    file = files[0]
    file.save('uploads/' + file.filename)
    return '文件上传成功！'

if __name__ == '__main__':
    app.run()

在上面的示例中，我们通过 request.files.getlist() 方法获取上传的文件列表，并检查列表的长度是否为1，如果不是则返回错误信息。