使用 Flask-Uploads 拒绝超过一定数量的文件？

Question

Flask 推荐使用 Flask-Uploads 模块来处理上传。我想拒绝任何超过一定大小的文件。有一些解决方案：

In addition, you can also use patch_request_class to patch your app’s request_class to have a maximum size for uploads.

patch_request_class(app, 32 * 1024 * 1024)

From this SO post:

MAX_CONTENT_LENGTH is the correct way to reject file uploads larger than you want,

app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024

# save to disk first, then check filesize
request.files['file'].save('/tmp/foo')
size = os.stat('/tmp/foo').st_size

-或-

# save to memory, then check filesize
blob = request.files['file'].read()
size = len(blob)

我没有在官方文档中看到 MAX_CONTENT_LENGTH，它甚至没有像 SO post 那样手动检查文件大小。这两种方法最终是相同的，还是存在（big/subtle？）差异？另外，patch_request_class 是先将文件保存到磁盘以确定总上传大小，还是先保存到内存？

Answer 1

MAX_CONTENT_LENGTH是Flask本身的配置项，0.6版本引入 http://flask.pocoo.org/docs/0.10/patterns/fileuploads/#improving-uploads

By default Flask will happily accept file uploads to an unlimited amount of memory, but you can limit that by setting the MAX_CONTENT_LENGTH config key:
from flask import Flask, Request

app = Flask(__name__)
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
The code above will limited the maximum allowed payload to 16 megabytes. If a larger file is transmitted, Flask will raise an RequestEntityTooLarge exception.

This feature was added in Flask 0.6 but can be achieved in older versions as well by subclassing the request object. For more information on that consult the Werkzeug documentation on file handling.

并且来自 flask-uploads 来源： https://bitbucket.org/leafstorm/flask-uploads/src/440e06b851d24811d20f8e06a8eaf5c5bf58c241/flaskext/uploads.py?at=default

def patch_request_class(app, size=64 * 1024 * 1024):
    """
    By default, Flask will accept uploads to an arbitrary size. While Werkzeug
    switches uploads from memory to a temporary file when they hit 500 KiB,
    it's still possible for someone to overload your disk space with a
    gigantic file.

    This patches the app's request class's
    `~werkzeug.BaseRequest.max_content_length` attribute so that any upload
    larger than the given size is rejected with an HTTP error.

    .. note::

       In Flask 0.6, you can do this by setting the `MAX_CONTENT_LENGTH`
       setting, without patching the request class. To emulate this behavior,
       you can pass `None` as the size (you must pass it explicitly). That is
       the best way to call this function, as it won't break the Flask 0.6
       functionality if it exists.

    .. versionchanged:: 0.1.1

    :param app: The app to patch the request class of.
    :param size: The maximum size to accept, in bytes. The default is 64 MiB.
                 If it is `None`, the app's `MAX_CONTENT_LENGTH` configuration
                 setting will be used to patch.
    """
    if size is None:
        if isinstance(app.request_class.__dict__['max_content_length'],
                      property):
            return
        size = app.config.get('MAX_CONTENT_LENGTH')
    reqclass = app.request_class
    patched = type(reqclass.__name__, (reqclass,),
                   {'max_content_length': size})
    app.request_class = patched

所以我会说：

app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024

Answer 2

patch_request_class 只是为了向后兼容。

从 Flask 0.6 开始你应该使用 app.config['MAX_CONTENT_LENGTH'].

我最近接任了 Flask-Uploads 项目的维护者，我做的第一件事就是删除了 this commit 中的 patch_request_class，因为它只会给新项目带来混乱.任何依赖它的旧项目都应该将 Flask-Uploads==0.1.3 固定在他们的 requirements.txt 文件中。

使用 Flask-Uploads 拒绝超过一定数量的文件？

Rejecting files greater than a certain amount with Flask-Uploads?

python

file-upload

flask

flask-uploads