Create JobIncompleteError

This commit is contained in:
Ken Tsang
2017-10-11 15:50:20 +01:00
parent dcf0d22d7b
commit 92508d3b96
2 changed files with 35 additions and 1 deletions

View File

@@ -7,6 +7,22 @@ from app.authentication.auth import AuthError
from app.errors import InvalidRequest
class JobIncompleteError(Exception):
def __init__(self, message):
self.message = message
def to_dict_v2(self):
return {
'status_code': 500,
"errors": [
{
"error": 'JobIncompleteError',
"message": self.message
}
]
}
class TooManyRequestsError(InvalidRequest):
status_code = 429
message_template = 'Exceeded send limits ({}) for today'
@@ -49,6 +65,10 @@ def register_errors(blueprint):
current_app.logger.exception(error)
return jsonify(json.loads(error.message)), 400
@blueprint.errorhandler(JobIncompleteError)
def job_incomplete_error(error):
return jsonify(error.to_dict_v2()), 500
@blueprint.errorhandler(NoResultFound)
@blueprint.errorhandler(DataError)
def no_result_found(e):