从WTForms字段获取上传的文件

时间:2015-05-16 18:47:03

标签: python flask wtforms flask-wtforms

在Flask文档中,文件上传示例使用<input type="file" name="file">然后使用request.files['file']来获取文件。我正在使用WTForms FileField。如何在使用WTForms时获取上传的文件,而不是自己编写输入html?

1 个答案:

答案 0 :(得分:19)

request.files是一个字典,其中键是文件字段的名称。您可以使用my_form.my_field.name获取WTForms字段的名称。因此,您可以使用request.files[my_form.my_field.name]访问从该字段上传的数据。

您可以使用Flask-WTF FileField,而不是使用WTForms FileField。它提供了data属性,可以为您获取文件数据。 This is described in the documentation.

from flask import url_for, redirect, render_template
from flask_wtf import Form
from flask_wtf.file import FileField
from werkzeug import secure_filename

class UploadForm(Form):
    file = FileField()

@app.route('/upload', methods=['GET', 'POST'])
def upload():
    form = UploadForm()

    if form.validate_on_submit():
        filename = secure_filename(form.file.data.filename)
        form.file.data.save('uploads/' + filename)
        return redirect(url_for('upload'))

    return render_template('upload.html', form=form)
<html>
<head>
<title>Upload</title>
</head>
<body>
<form method="post" enctype="multipart/form-data">
    {{ form.hidden_tag() }}
    {{ form.file }}
    <input type="submit">
</form>
</body>
</html>