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

推荐阅读更多精彩内容

  • 19年6月更新:该篇翻译已经被移到 TangYefei's Blog,相较于简书阅读体验会更好,如果喜欢请点个 ❤...
    tangyefei阅读 33,605评论 22 257
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 121,117评论 16 134
  • 本文首发于Gevin的博客 原文链接:Flask 入门指南 未经 Gevin 授权,禁止转载 1. 初识Flask...
    Gevin阅读 15,614评论 9 234
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 157,697评论 24 688
  • 望着我的小麦草,他已经长出了小方盒,细嫩而挺拔,葱翠而轻盈,心底透出一片清澈的镜面。宛如湖水泛着清晰的倒影...
    Doreen乖乖美美哒阅读 149评论 0 0