PythonFlask-WTF文件上传后的后续处理方法

发布时间：2023-12-11 06:22:37

在Python Flask中，可以使用Flask-WTF扩展来处理文件上传。Flask-WTF是一个用于集成WTForms表单功能的Flask插件，它提供了一个方便的方法来处理文件上传。

以下是使用Flask-WTF处理文件上传的步骤：

1. 安装Flask-WTF库。可以使用pip命令进行安装：

pip install Flask-WTF

2. 创建一个Flask应用程序，并导入必要的模块：

from flask import Flask, render_template, request
from flask_wtf import FlaskForm
from wtforms import FileField, SubmitField

3. 创建一个表单类，继承自FlaskForm，并定义一个FileField字段用于接收文件上传：

class MyForm(FlaskForm):
    file = FileField('Upload File')
    submit = SubmitField('Submit')

4. 创建一个视图函数来处理表单的GET和POST请求，并在模板中渲染表单：

@app.route('/', methods=['GET', 'POST'])
def upload_file():
    form = MyForm()
    
    if form.validate_on_submit():
        file = form.file.data
        filename = file.filename
        file.save(filename)
        return 'File uploaded successfully'
    
    return render_template('upload.html', form=form)

5. 创建一个模板文件upload.html来渲染表单：

<!DOCTYPE html>
<html>
<head>
    <title>File Upload</title>
</head>
<body>
    <h1>File Upload</h1>
    <form method="post" enctype="multipart/form-data">
        {{ form.csrf_token }}
        {{ form.file.label }}
        {{ form.file }}
        {{ form.submit }}
    </form>
</body>
</html>

在上述代码中，首先创建了一个表单类MyForm，它包含一个FileField字段和一个SubmitField字段。然后，创建了一个视图函数upload_file来处理表单的GET和POST请求。在GET请求中，将表单对象传递给模板文件进行渲染；在POST请求中，如果表单验证成功，则获取上传的文件对象并保存到本地。

可以通过form.validate_on_submit()方法来判断表单是否通过验证。如果验证成功，则可以通过form.file.data来获取文件对象，并使用file.save()方法将文件保存到指定路径。最后，返回上传成功的提示信息。

启动Flask应用程序后，访问http://localhost:5000即可看到文件上传表单。选择一个文件并点击提交按钮，文件将被上传到服务器，并显示上传成功的提示信息。

总结：

Python Flask-WTF扩展提供了一个简单而强大的方法来处理文件上传。通过定义一个包含FileField字段的表单类，并在视图函数中处理文件保存等后续操作，可以实现文件上传功能。