Axione-IPE-Viewer/webapp/eligibility_api/elig_api_exceptions.py

64 lines
1.6 KiB
Python
Raw Normal View History

2022-11-16 16:10:30 +01:00
from flask import Flask, jsonify
class ApiParamException(Exception):
"""
Exception thrown if API misused
"""
def __init__(self, description):
self.code = 400
self.name = "Bad API parameter"
self.description = description
2023-03-03 14:16:34 +01:00
class NetwoApiErrorException(Exception):
"""
Exception thrown if Netwo API
"""
def __init__(self, description: str, netwo_status_code: int):
self.netwo_status_code = netwo_status_code
self.name = "Error contacting Netwo API"
self.description = description
2022-11-16 16:10:30 +01:00
class FlaskExceptions:
"""
Manages flask custom exceptions
"""
def __init__(self, flask_app: Flask):
self.flask_app = flask_app
def add_exceptions(self):
"""
declares custom exceptions to flask app
"""
@self.flask_app.errorhandler(ApiParamException)
def handle_exception(e):
return (
jsonify(
{
"code": e.code,
"name": e.name,
"description": e.description,
}
),
400,
2023-02-28 22:31:35 +01:00
)
2023-03-03 14:16:34 +01:00
@self.flask_app.errorhandler(NetwoApiErrorException)
def handle_exception(e):
return (
jsonify(
{
"netwo_status_code": e.netwo_status_code,
"name": e.name,
"description": e.description,
}
),
500,
)