checktheroads-hyperglass/hyperglass/exceptions/_common.py

"""Custom exceptions for hyperglass."""

# Standard Library
import json as _json
from typing import Any, Dict, List, Union, Literal, Optional

# Third Party
from pydantic import ValidationError

# Project
from hyperglass.log import log
from hyperglass.util import get_fmt_keys, repr_from_attrs
from hyperglass.constants import STATUS_CODE_MAP

ErrorLevel = Literal["danger", "warning"]


class HyperglassError(Exception):
    """hyperglass base exception."""

    def __init__(
        self,
        message: str = "",
        level: ErrorLevel = "warning",
        keywords: Optional[List[str]] = None,
    ) -> None:
        """Initialize the hyperglass base exception class."""
        self._message = message
        self._level = level
        self._keywords = keywords or []
        if self._level == "warning":
            log.error(str(self))
        elif self._level == "danger":
            log.critical(str(self))
        else:
            log.info(str(self))

    def __str__(self) -> str:
        """Return the instance's error message."""
        return self._message

    def __repr__(self) -> str:
        """Return the instance's severity & error message in a string."""
        return repr_from_attrs(self, ("_message", "level", "keywords"), strip="_")

    def dict(self) -> Dict[str, Union[str, List[str]]]:
        """Return the instance's attributes as a dictionary."""
        return {
            "message": self._message,
            "level": self._level,
            "keywords": self._keywords,
        }

    def json(self) -> str:
        """Return the instance's attributes as a JSON object."""
        return _json.dumps(self.__dict__())

    @staticmethod
    def _safe_format(template: str, **kwargs: Dict[str, str]) -> str:
        """Safely format a string template from keyword arguments."""

        keys = get_fmt_keys(template)
        for key in keys:
            if key not in kwargs:
                kwargs.pop(key)
            else:
                kwargs[key] = str(kwargs[key])
        return template.format(**kwargs)

    def _parse_pydantic_errors(*errors: Dict[str, Any]) -> str:
        errs = ("\n",)

        for err in errors:
            loc = " → ".join(str(loc) for loc in err["loc"])
            errs += (f'Field: {loc}\n  Error: {err["msg"]}\n',)

        return "\n".join(errs)

    @property
    def message(self) -> str:
        """Return the instance's `message` attribute."""
        return self._message

    @property
    def level(self) -> str:
        """Return the instance's `level` attribute."""
        return self._level

    @property
    def keywords(self) -> List[str]:
        """Return the instance's `keywords` attribute."""
        return self._keywords

    @property
    def status_code(self) -> int:
        """Return HTTP status code based on level level."""
        return STATUS_CODE_MAP.get(self._level, 500)


class PublicHyperglassError(HyperglassError):
    """Base exception class for user-facing errors.

    Error text should be defined in
    `hyperglass.configuration.params.messages` and associated with the
    exception class at start time.
    """

    _level = "warning"
    _message_template = "Something went wrong."
    _original_template_name: str = ""

    def __init_subclass__(
        cls, *, template: Optional[str] = None, level: Optional[ErrorLevel] = None
    ) -> None:
        """Override error attributes from subclass."""

        if template is not None:
            cls._message_template = template
            cls._original_template_name = template
        if level is not None:
            cls._level = level

    def __init__(self, **kwargs: str) -> None:
        """Format error message with keyword arguments."""
        # Project
        from hyperglass.state import use_state

        if "error" in kwargs:
            error = kwargs.pop("error")
            error = self._safe_format(str(error), **kwargs)
            kwargs["error"] = error

        template = self._message_template

        (messages := use_state("params").messages)
        if messages.has(self._original_template_name):
            template = messages[self._original_template_name]
        if "error" in kwargs and "({error})" not in template:
            template += " ({error})"
        self._message = self._safe_format(template, **kwargs)
        self._keywords = list(kwargs.values())
        super().__init__(message=self._message, level=self._level, keywords=self._keywords)


class PrivateHyperglassError(HyperglassError):
    """Base exception class for internal system errors.

    Error text is dynamic based on the exception being caught.
    """

    _level = "warning"

    def _parse_validation_error(self, err: ValidationError) -> str:
        errors = err.errors()
        parsed = {
            k: ", ".join(str(loc) for t in errors for loc in t["loc"] if t["type"] == k)
            for k in {e["type"] for e in errors}
        }
        return ", ".join(parsed.values())

    def __init_subclass__(cls, *, level: Optional[ErrorLevel] = None) -> None:
        """Override error attributes from subclass."""
        if level is not None:
            cls._level = level

    def __init__(self, message: str, **kwargs: Any) -> None:
        """Format error message with keyword arguments."""
        if "error" in kwargs:
            error = kwargs.pop("error")
            error = self._safe_format(str(error), **kwargs)
            kwargs["error"] = error

        if isinstance(message, ValidationError):
            message = self._parse_validation_error(message)

        self._message = self._safe_format(message, **kwargs)
        self._keywords = list(kwargs.values())
        super().__init__(message=self._message, level=self._level, keywords=self._keywords)