Source code for xmm.pipeline.writers.pprint

from pprint import pprint

from .base import Writer


class PPrintWriter(Writer):
    """A data writer class that prints() for testing purposes."""

    supports_nested = True
    params = []

    def __init__(self, **kwargs):
        """
        Create a new Pretty-print writer.

        :param kwargs: Options to be passed down to :func:`pprint.pprint`

        """
        self.kwargs = kwargs

    def write(self, dataset, i):
        """Pretty-print the dataset as-is."""
        pprint(dataset, **self.kwargs)