烧瓶上传定义允许哪些文件 [英] Flask-uploads define which files are allowed

问题描述

我正在使用称为"flask-uploads"的烧瓶扩展名，并且我只想允许上传真实图片(jpg，png等).如何限制上传文件的类型?

I am working with the flask extension called "flask-uploads" and I only want to allow real pictures to be uploaded (jpg, png so on). How can I restrict the type of uploaded file?

@app.route("/post", methods=["GET", "POST"])
@login_required
def post():
    # HERE I'M LINKING TO ANOTHER FILE, THIS IS NOT THE ERROR
    post = posts.Post(session["user_id"])

    if request.method == "POST" and 'photo' in request.files:
        # save photo in img folder
        try:
            file = photos.save(request.files["photo"])

        # I THINK THE ERROR IS IN HERE
        except UploadNotAllowed:
            return redirect(url_for("index"))
        else:
            path = 'static/' + str(file)
            post.upload(path)

        return redirect(url_for("index"))

        else:
            following = post.loadgroups()
            return render_template("post.html", groups = following)

这是flask-uploads的配置:

This is the configuration of flask-uploads:

from flask_uploads import UploadSet, IMAGES, configure_uploads, UploadNotAllowed

# configure flask_upload API
photos = UploadSet("photos", IMAGES)
app.config["UPLOADED_PHOTOS_DEST"] = "static/img"
configure_uploads(app, photos)

推荐答案

photos = UploadSet("photos", IMAGES)

此行控制只能上载图像文件. 但是，如果您想更具体一点，可以

This line controls only image file can be uploaded. However, if you want to be more specific you can

photos= UploadSet('photos', ('png', 'jpg'))

这篇关于烧瓶上传定义允许哪些文件的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！