iridium.pprint

"""Utilities for pretty printing."""

import pprint
from typing import Any, Dict, Generic, TypeVar

import wrapt

pprint_args: Dict[str, Any] = {
    "indent": 2,
    "depth": 4,
}
"""Globally used pprint settings."""


def pp(obj) -> str:
    """Pretty print object (using global pprint_args)."""
    return pprint.pformat(obj, **pprint_args)


T = TypeVar("T")


class NoPrint(wrapt.ObjectProxy, Generic[T]):
    """Wrap any class to avoid printing it."""

    def __repr__(self) -> str:
        return f"{type(self.__wrapped__).__name__}(...)"


class PrettyRepr(wrapt.ObjectProxy, Generic[T]):
    """Wrap any class to apply pretty-printing to it."""

    def __repr__(self) -> str:
        return pp(self.__wrapped__)

def pp(obj) -> str:
    """Pretty print object (using global pprint_args)."""
    return pprint.pformat(obj, **pprint_args)