aboutsummaryrefslogtreecommitdiff
"""Common utilities."""
import logging
import traceback

from flask import request, jsonify, render_template

logger = logging.getLogger(__name__)


def add_trace(exc: Exception, errobj: dict) -> dict:
    """Add the traceback to the error handling object."""
    return {
        **errobj,
        "error-trace": "".join(traceback.format_exception(exc))
    }


def build_handler(description: str):
    """Generic utility to build error handlers."""
    def __handler__(exc: Exception):
        error = (exc.name if hasattr(exc, "name") else exc.__class__.__name__)
        status_code = exc.code if hasattr(exc, "code") else 500
        content_type = request.content_type
        if bool(content_type) and content_type.lower() == "application/json":
            return (
                jsonify(add_trace(exc,
                                  {
                                      "requested-uri": request.url,
                                      "error": error,
                                      "error_description": description
                                  })),
                status_code)

        return (render_template(f"http-error-{str(status_code)[0:-2]}xx.html",
                                error=exc,
                                page=request.url,
                                description=description,
                                trace=traceback.format_exception(exc)),
                status_code)

    return __handler__