关于Flask高级_flask_wtf验证上传的文件的方法

NewBurlaSKing

于 2022-08-31 17:16:12 发布

阅读量215

点赞数 1

分类专栏： Flask 文章标签： flask python 后端

本文链接：https://blog.csdn.net/qq_55961861/article/details/126628061

版权

Flask 专栏收录该内容

47 篇文章 1 订阅

订阅专栏

Flask高级_flask_wtf验证上传的文件

一.验证步骤

定义验证表单类的时候，对文件类型的字段，需要采用 FileField 这个类型，即wtforms.FileField
验证器需要从 flask_wtf.file 中导入。 flask_wtf.file.FileRequired 和 flask_wtf.file.FileAllowed
在视图函数中，需要使用 from werkzeug.datastructures import CombinedMultiDict 来把 request.form 与 request.files 来进行合并。
最后使用表单验证对象.validate()进行验证。

二.实例

html:

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta http-equiv="X-UA-Compatible" content="IE=edge">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>上传文件页面</title>
</head>
<body>
    <form action="" method="post" enctype="multipart/form-data">
        <table>
            <tr>
                <td>我的头像:</td>
                <td><input type="file" name="pic"></td>
            </tr>
            <tr>
                <td></td>
                <td><input type="submit" value="提交"></td>
            </tr>
        </table>
    </form>
</body>
</html>

form.py:

from wtforms import Form,FileField
from flask_wtf.file import FileRequired,FileAllowed

class uploadForm(Form):
    pic = FileField(validators=[FileRequired(),FileAllowed(['jpg','png','gif'])])

main.py:

#coding=utf-8

from flask import Flask,render_template,request
import os
from werkzeug.utils import secure_filename
from werkzeug.datastructures import CombinedMultiDict
from uploadForm import uploadForm

app = Flask(__name__)

UPLOAD_PATH = os.path.join(os.path.dirname(__file__),'imgs')

@app.route('/')
def index():
    return 'Hello~'

@app.route('/upload/',methods=['GET','POST'])
def upload():
    if request.method == 'GET':
        return render_template('upload.html')
    else:
        form = uploadForm(CombinedMultiDict([request.files,request.form]))
        if form.validate():
            fpic = form.pic.data
            filename = secure_filename(fpic.filename)
            fpic.save(os.path.join(UPLOAD_PATH,filename))
            return '上传文件成功!'
        return f'上传失败！失败原因:{form.errors}'

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