使用Flask-WTF的FileField()实现文件上传和保存

发布时间：2024-01-17 12:00:48

使用Flask-WTF的FileField()可以很方便地实现文件上传和保存的功能。FileField()是Flask-WTF中的一个表单字段类，用于处理文件上传。

首先，需要安装Flask-WTF库。可以使用以下命令进行安装：

pip install Flask-WTF

接下来，在Flask应用中导入必要的模块和类：

from flask import Flask, render_template
from flask_wtf import FlaskForm
from wtforms import FileField, SubmitField
from werkzeug.utils import secure_filename

然后，创建一个表单类，继承自FlaskForm，并定义一个FileField()字段和一个SubmitField()字段：

class UploadForm(FlaskForm):
    upload = FileField('Upload File')
    submit = SubmitField('Submit')

接下来，需要配置Flask应用的相关设置。需要设置一个秘钥，并指定文件上传的目录。可以使用以下代码进行配置：

app = Flask(__name__)
app.config['SECRET_KEY'] = 'your_secret_key'
app.config['UPLOAD_FOLDER'] = '/path/to/upload/folder'

然后，创建一个视图函数来处理文件上传和保存的逻辑：

@app.route('/', methods=['GET', 'POST'])
def upload_file():
    form = UploadForm()
    
    if form.validate_on_submit():
        file = form.upload.data
        filename = secure_filename(file.filename)
        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
        
        return 'File uploaded successfully!'
    
    return render_template('upload.html', form=form)

在视图函数中，首先实例化表单对象，并在验证通过时获取上传的文件数据。然后，使用secure_filename()函数来确保文件名的安全，并保存文件到指定目录中。

最后，创建一个HTML模板来渲染表单，并将其与视图函数的逻辑进行绑定。可以创建一个名为upload.html的模板文件：

<!doctype html>
<html>
<head>
    <title>File Upload</title>
</head>
<body>
    <h1>File Upload</h1>
    
    <form method="POST" enctype="multipart/form-data" action="/">
        {{ form.csrf_token }}
        {{ form.upload.label }} {{ form.upload }}
        {{ form.submit }}
    </form>
</body>
</html>

在模板中，使用form.csrf_token来添加一个CSRF令牌，确保表单的安全性。然后，将表单字段渲染为HTML输入元素，并添加一个submit按钮。

运行Flask应用，并访问根URL，即可看到一个文件上传表单。选择一个文件并点击提交按钮，文件将被上传到指定目录中，并显示上传成功的消息。

以上就是使用Flask-WTF的FileField()实现文件上传和保存的示例。使用Flask-WTF的FileField()可以快速方便地实现文件上传功能，并且提供了一些验证选项来确保文件的有效性和安全性。可以根据具体的需求进行修改和扩展。