PythonFlask-WTF文件上传前的预处理方法

发布时间：2023-12-11 06:20:58

在Python Flask框架中使用WTForms扩展进行文件上传时，可以在文件上传前对上传的文件进行一些预处理操作。预处理方法用于验证文件的合法性、对文件进行处理或转换等操作。

下面是一个使用例子，展示了如何在文件上传前进行预处理：

from flask import Flask, render_template, request
from flask_wtf import FlaskForm
from wtforms import FileField
from wtforms.validators import FileRequired

app = Flask(__name__)
app.config['SECRET_KEY'] = 'secret_key'


class FileForm(FlaskForm):
    file = FileField('File', validators=[FileRequired()])


def preprocess_file(file):
    # 验证文件类型是否为图片
    allowed_extensions = {'jpg', 'png', 'jpeg', 'gif'}
    if file.filename.split('.')[-1].lower() not in allowed_extensions:
        raise ValueError('Only image files are allowed.')

    # 对图片进行处理或转换操作
    # 这里可以根据需求对文件进行任何操作，比如压缩、切割、转换格式等

    # 返回处理后的文件对象
    return file


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

    if request.method == 'POST' and form.validate_on_submit():
        file = form.file.data
        try:
            processed_file = preprocess_file(file)
        except ValueError as e:
            return str(e)

        # 在这里可以保存处理后的文件，或进行其他操作
        # processed_file.save('processed_image.jpg')

        return 'File uploaded and processed successfully.'

    return render_template('index.html', form=form)


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

在上面的例子中，我们首先定义了一个FileForm类，继承自FlaskForm，其中有一个FileField字段用于接收文件上传。我们为该字段添加了一个验证器FileRequired()，用于确保上传的文件不为空。

然后，我们定义了一个preprocess_file方法，用于预处理上传的文件。在该方法中，我们首先验证文件类型是否为图片，只允许上传图片文件，如果文件类型不符合要求，则抛出ValueError异常。之后，可以根据需求对文件进行任何处理或转换操作，例如，压缩、切割、转换格式等。最后，返回处理后的文件对象。

在路由函数upload_file中，我们创建了一个FileForm实例，并在模板中渲染该表单。当请求方法为POST且表单通过验证时，我们从表单中获取文件对象，并调用preprocess_file方法对文件进行预处理。如果在预处理中抛出了异常，则返回异常信息。否则，我们可以在该方法中保存处理后的文件，或进行其他操作。

需要注意的是，实际使用中，我们要根据具体需求对文件进行预处理和后续操作。上述代码中只是一个简单的示例，具体操作需要根据实际需求进行调整。

同时，还需要在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.file.errors }}
        <input type="submit" value="Upload">
    </form>
</body>
</html>

在上述示例中，我们使用了Flask-WTF扩展和WTForms库，通过定义表单类和预处理方法，可以对文件上传进行预处理，提高代码的可维护性和灵活性。