Python中使用wtforms.fieldsFileField()实现文件上传功能

发布时间：2024-01-13 11:10:25

在Python中，使用wtforms.fields.FileField()可以很方便地实现文件上传功能。下面是使用wtforms.fields.FileField()的一个简单例子。

首先，我们需要安装wtforms库。可以使用以下命令进行安装：

pip install wtforms

然后，我们可以创建一个Python文件，命名为file_upload_example.py，并在其中编写以下代码。

from flask import Flask, render_template, request
from wtforms import Form, FileField, validators

app = Flask(__name__)


class FileUploadForm(Form):
    file = FileField('Choose file', [validators.DataRequired()])


@app.route('/upload', methods=['GET', 'POST'])
def upload():
    form = FileUploadForm(request.form)
    if request.method == 'POST' and form.validate():
        file = request.files['file']
        # 以下是文件保存的代码，这里只是简单地保存文件到当前目录下的uploads文件夹中
        file.save(f"uploads/{file.filename}")
        return f"File '{file.filename}' has been uploaded successfully."
    return render_template('upload.html', form=form)


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

上述代码使用Flask框架创建了一个简单的Web应用，并实现了一个文件上传功能。在代码中，我们定义了一个FileUploadForm类，继承自wtforms.Form，并有一个file属性，它是一个FileField对象。在FileField构造函数中，我们通过传入'Choose file'和[validators.DataRequired()]两个参数，设置表单字段的标签文本和验证器。

然后，我们创建了一个路由/upload，该路由既接受GET请求，也接受POST请求。在GET请求的时候，我们创建了一个FileUploadForm对象，并将其作为参数传递给render_template()函数，以便在模板中渲染表单。在POST请求的时候，我们首先通过request.files['file']来获取上传的文件对象，然后可以保存文件到指定的位置。

需要注意的是，上述代码中只是保存了文件到当前目录下的uploads文件夹中，实际情况中可能需要根据具体需求修改保存文件的路径。

最后，我们需要创建一个HTML模板文件，命名为upload.html，并编写以下代码：

<!DOCTYPE html>
<html>
<head>
    <title>File Upload</title>
</head>
<body>
    <h2>File Upload</h2>
    <form method="POST" enctype="multipart/form-data">
        {{ form.file.label }}
        {{ form.file }}
        <p>{{ form.file.errors }}</p>
        <input type="submit" value="Upload">
    </form>
</body>
</html>

在上述HTML模板中，我们使用{{ form.file.label }}和{{ form.file }}来渲染表单字段的标签和输入控件。我们也使用了{{ form.file.errors }}来显示表单字段的错误信息。

完成以上步骤后，我们可以通过运行file_upload_example.py来启动应用，并访问http://localhost:5000/upload来使用文件上传功能。

在这个例子中，我们演示了如何使用wtforms.fields.FileField()实现文件上传功能。你可以根据自己的需求，在保存文件的时候做一些额外的处理，比如校验文件类型、检查文件大小等等。