我正在尝试编写一些包装函数,该函数将首先检查请求的大小,如果请求的大小大于我的预期,则会向客户端返回特定内容。我之前尝试使用request.content_length
,当请求是POST
请求时可以工作,但无法用于GET
请求。有没有办法解决这个问题?
flask.request.headers.get('Content-Length')
所以你可以这样做:
import functools
def check_content_length_of_request_decorator(max_content_length):
def wrapper(fn):
@functools.wraps(fn)
def decorated_view(*args, **kwargs):
if int(flask.request.headers.get('Content-Length') or 0) > max_content_length:
return flask.abort(400, description='Too much content in request')
else:
return fn(*args, **kwargs)
return decorated_view
return wrapper
@app.route('/my_route', methods=['GET', 'POST'])
@check_content_length_of_request_decorator(1000)
def my_route():
return 'This request has a valid content length!'