views.py 文件源码

def upload_avatar():
    form = UploadAvatarForm()
    if form.validate_on_submit():
        avatar = request.files['avatar']
        filename = secure_filename(avatar.filename)
        UPLOAD_FOLDER = current_app.config['UPLOAD_FOLDER']
        ALLOWED_EXTENTIONS = set(['jpg', 'png', 'jpeg', 'gif'])
        flag = '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENTIONS
        if not flag:
            flash('Error file types.')
            return redirect(url_for('.user', username=current_user.username))
        avatar.save(os.path.join(UPLOAD_FOLDER, filename))
        current_user.user_avatar = os.path.join(UPLOAD_FOLDER, filename)

        db.session.add(current_user)
        flash('Your avatar has been updated.')
        return redirect(url_for('.user', username=current_user.username))
    return render_template('upload_avatar.html', form=form)