flask利用flask-wtf验证上传的文件的方法

介绍

在使用 Flask 框架进行 web 开发时，文件上传是一个常见的需求，而且对上传文件进行有效的验证更是必不可少的。Flask-WTF 是一个方便且强大的库，可以轻松实现文件上传的验证。本文将介绍如何使用 Flask 和 Flask-WTF 进行文件上传验证。

安装

首先，确保已经安装了 Flask 和 Flask-WTF 这两个库。可以使用 pip 命令进行安装：

pip install flask
pip install flask-wtf

创建 Flask 应用

首先，我们需要创建一个 Flask 应用。可以在一个项目目录下创建一个名为 app.py 的文件，并写入以下内容：

from flask import Flask, render_template, request
from flask_wtf import FlaskForm
from wtforms import FileField, SubmitField
from wtforms.validators import DataRequired
app = Flask(__name__)
app.config['SECRET_KEY'] = 'your-secret-key'
class UploadForm(FlaskForm):
    file = FileField('Upload File', validators=[DataRequired()])
    submit = SubmitField('Submit')
@app.route('/', methods=['GET', 'POST'])
def upload_file():
    form = UploadForm()
    if form.validate_on_submit():
        file = form.file.data
        # 处理上传的文件
        return 'File uploaded successfully!'
    return render_template('upload.html', form=form)
if __name__ == '__main__':
    app.run(debug=True)

上述代码中，我们创建了一个名为 UploadForm 的表单类，其中包含了一个文件字段和一个提交按钮。在上传文件时，我们会通过 FlaskForm 类的 validate_on_submit 方法进行验证，并获取上传的文件对象。在实际项目中，可以根据需要对上传的文件进行进一步的处理。

创建 HTML 模板

接下来，我们需要创建一个 HTML 模板用于渲染表单。在项目目录下创建一个名为 templates 的文件夹，并在文件夹中创建一个名为 upload.html 的文件，并写入以下内容：

<form method="POST" enctype="multipart/form-data">
    {{ form.csrf_token }}
    {{ form.file.label }} {{ form.file }}
    {{ form.submit }}
</form>

在上述代码中，我们使用了 Flask-WTF 提供的表单模板渲染功能，通过 `{{ form.csrf_token }}` 渲染 CSRF 令牌，`{{ form.file.label }}` 渲染文件字段的标签，`{{ form.file }}` 渲染文件字段本身，`{{ form.submit }}` 渲染提交按钮。

运行应用

现在，我们可以运行我们的 Flask 应用，并访问 http://localhost:5000 来查看上传表单。上传一个文件并点击提交按钮，如果文件上传成功，会显示 "File uploaded successfully!"。

总结

在本文中，我们介绍了如何使用 Flask 和 Flask-WTF 进行文件上传验证。通过使用 Flask-WTF 提供的表单验证功能，我们可以轻松地对上传的文件进行验证，并在需要时进行进一步的处理。这是一个简单但实用的技巧，可以帮助我们更好地处理文件上传相关的需求。