Better handle missing error.message

This commit is contained in:
Joel Collins 2019-01-31 16:48:19 +00:00
parent 04bed170a0
commit 63336757d4

View file

@ -2,8 +2,6 @@ from flask import jsonify
from werkzeug.exceptions import default_exceptions
from werkzeug.exceptions import HTTPException
from pprint import pprint
class JSONExceptionHandler(object):
def __init__(self, app=None):
@ -11,7 +9,13 @@ class JSONExceptionHandler(object):
self.init_app(app)
def std_handler(self, error):
message = error.description if isinstance(error, HTTPException) else error.message
if isinstance(error, HTTPException):
message = error.description
elif hasattr(error, 'message'):
message = error.message
else:
message = str(error)
status_code = error.code if isinstance(error, HTTPException) else 500
response = {