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

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


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


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,
            )

        @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,
            )