Flask上传单个文件或多个文件

今天记下使用Flask如何长传单个或多个文件。

以上传头像为例

配置上传路径：

# 上传文件存储的目录
MEDIA_LEYU_UPLOAD = os.path.join(os.getcwd(), 'upload')

#头像上传路径
UPLOAD_AVATAR_FOLDER = os.path.join(MEDIA_LEYU_UPLOAD, 'avatar')

表单设计

from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileAllowed, FileRequired
from wtforms import SubmitField

class SetavatarForm(FlaskForm):
    avatar = FileField(
        '用户头像', validators=[FileRequired(), FileAllowed(['jpg', 'jpeg', 'png'], 'Images only!')]
    )
    submit = SubmitField('确认')

模板

<form class="form-horizontal"  method="POST"  enctype="multipart/form-data" >
                        {{ form.csrf_token }}

                        <div class="col-md-4 col-sm-4 col-xs-12 input-width-ie7">
                        {{ form.avatar(class="form-control") }}
                        </div>

                        <div class="col-md-4 col-sm-4 col-xs-12 col-sm-offset-3">
                          {{ form.submit(class="btn btn-success") }}
                        </div>
  </form>

视图函数

@user.route('/set-avatar', endpoint='SetAvatar', methods=['GET', 'POST'])
def set_avatar():
    form = SetavatarForm()
    if request.method == 'POST' and form.validate_on_submit():
        file_obj = request.files.get('avatar', '')
        #获得文件名
        file_name = file_obj.filename
        #获得文件格式
        file_postfix = file_name.split('.')[-1]
        #对文件进行重命名
        store_filename = '{}.{}'.format(str(current_user.id), file_postfix)
        #检查文件保存路径
        if not os.path.exists(constant.UPLOAD_AVATAR_FOLDER):
            os.makedirs(constant.UPLOAD_AVATAR_FOLDER)

        path = os.path.join(constant.UPLOAD_AVATAR_FOLDER, store_filename)
        #保存文件
        file_obj.save(path)
        #便于远程访问使用URI保存
        url = '{}{}'.format(request.url_root, url_for('main.AvatarUpload', filename=store_filename)[1:])
        try:
            current_user.avatar = url
            db.session.commit()
        except Exception as err:
            raise err
        return render_template('user/set_avatar.html', form=form)
    elif not form.validate_on_submit():
        flash("请上传jpg或png格式的头像")
        return render_template('user/set_avatar.html', form=form)
    else:
        return render_template('user/set_avatar.html', form=form)

大致过程如上，并未进行安全认证

如果是多文件上传要注意两点：
模板文件要配置一项：

{{ form.screen_shot(multiple="multiple") }}

这样才能够使用多文件上传
另外多文件上传后为文件列表，获取文件方法如下：

file_obj = request.files.getlist('screen_shot')
for file in file_obj:
    file_name = file.filename