文件上传过程中,一定要进行文件大小的校验,如果用户上传文件过大,比如:100GB,所以一直在处理这个请求,别的请求进来就无法处理了,所以要限制上传文件的大小。
flask中文件上传大小的校验
flask中,提供了校验文件大小的机制,所以在配置文件中进行配置,并重写异常返回信息即可。
设置配置文件:app.config['MAX_CONTENT_LENGTH']=1024 * 1024 * 5
自定制错误返回信息
在flask内部,会帮我们校验,如果上传文件大于设置,会返回一个 413 的异常
所以自定义这个413的异常信息
@app.errorhandler(413) def error(arg): ret = make_response('上传文件最大支持:5M') ret.status_code = 413 return ret