from collections import OrderedDict

from six import StringIO


class PrettyPrinter(object):
    """Cleaner pprint output.

    Heavily inspired by the Python pprint module, but more basic for now.

    """
    def pformat(self, obj):
        stream = StringIO()
        self._format(obj, stream)
        return stream.getvalue()

    def _format(self, obj, stream, indent=4, level=1):
        _repr = getattr(type(obj), '__repr__', None)
        write = stream.write

        if (
            (isinstance(obj, dict) and _repr is dict.__repr__) or
            (isinstance(obj, OrderedDict) and _repr == OrderedDict.__repr__)
        ):
            write('{\n')
            num = len(obj)

            if num > 0:
                for i, (key, value) in enumerate(obj.items()):
                    write(' ' * (indent * level))
                    write("'%s'" % key)
                    write(': ')
                    self._format(value, stream, level=level + 1)
                    if i < num - 1:
                        write(',')
                    write('\n')

                write(' ' * (indent * (level - 1)))
            write('}')

        elif isinstance(obj, list) and _repr is list.__repr__:
            write('[')
            num = len(obj)

            if num > 0:
                write('\n')
                for i, value in enumerate(obj):
                    write(' ' * (indent * level))
                    self._format(value, stream, level=level + 1)
                    if i < num - 1:
                        write(',')
                    write('\n')
                write(' ' * (indent * (level - 1)))
            write(']')
        else:
            value = repr(obj)
            if '\n' in value:
                lines = value.split('\n')
                num = len(lines)
                for i, line in enumerate(lines):
                    if i > 0:
                        write(' ' * (indent * (level - 1)))
                    write(line)
                    if i < num - 1:
                        write('\n')
            else:
                write(value)