122 lines · 3.9 KB · Python
RawDownload
"""Redo the `...` (representation) but with limits on most sizes.""" __all__ = ["Repr","repr"] import __builtin__from itertools import islice class Repr:     def __init__(self):        self.maxlevel = 6        self.maxtuple = 6        self.maxlist = 6        self.maxarray = 5        self.maxdict = 4        self.maxset = 6        self.maxfrozenset = 6        self.maxdeque = 6        self.maxstring = 30        self.maxlong = 40        self.maxother = 20     def repr(self, x):        return self.repr1(x, self.maxlevel)     def repr1(self, x, level):        typename = type(x).__name__        if ' ' in typename:            parts = typename.split()            typename = '_'.join(parts)        if hasattr(self, 'repr_' + typename):            return getattr(self, 'repr_' + typename)(x, level)        else:            s = __builtin__.repr(x)            if len(s) > self.maxother:                i = max(0, (self.maxother-3)//2)                j = max(0, self.maxother-3-i)                s = s[:i] + '...' + s[len(s)-j:]            return s     def _repr_iterable(self, x, level, left, right, maxiter, trail=''):        n = len(x)        if level <= 0 and n:            s = '...'        else:            newlevel = level - 1            repr1 = self.repr1            pieces = [repr1(elem, newlevel) for elem in islice(x, maxiter)]            if n > maxiter:  pieces.append('...')            s = ', '.join(pieces)            if n == 1 and trail:  right = trail + right        return '%s%s%s' % (left, s, right)     def repr_tuple(self, x, level):        return self._repr_iterable(x, level, '(', ')', self.maxlist, ',')     def repr_list(self, x, level):        return self._repr_iterable(x, level, '[', ']', self.maxlist)     def repr_array(self, x, level):        header = "array('%s', [" % x.typecode        return self._repr_iterable(x, level, header, '])', self.maxarray)     def repr_set(self, x, level):        x = sorted(x)        return self._repr_iterable(x, level, 'set([', '])', self.maxset)     def repr_frozenset(self, x, level):        x = sorted(x)        return self._repr_iterable(x, level, 'frozenset([', '])',                                   self.maxfrozenset)     def repr_deque(self, x, level):        return self._repr_iterable(x, level, 'deque([', '])', self.maxdeque)     def repr_dict(self, x, level):        n = len(x)        if n == 0: return '{}'        if level <= 0: return '{...}'        newlevel = level - 1        repr1 = self.repr1        pieces = []        for key in islice(sorted(x), self.maxdict):            keyrepr = repr1(key, newlevel)            valrepr = repr1(x[key], newlevel)            pieces.append('%s: %s' % (keyrepr, valrepr))        if n > self.maxdict: pieces.append('...')        s = ', '.join(pieces)        return '{%s}' % (s,)     def repr_str(self, x, level):        s = __builtin__.repr(x[:self.maxstring])        if len(s) > self.maxstring:            i = max(0, (self.maxstring-3)//2)            j = max(0, self.maxstring-3-i)            s = __builtin__.repr(x[:i] + x[len(x)-j:])            s = s[:i] + '...' + s[len(s)-j:]        return s     def repr_long(self, x, level):        s = __builtin__.repr(x) # XXX Hope this isn't too slow...        if len(s) > self.maxlong:            i = max(0, (self.maxlong-3)//2)            j = max(0, self.maxlong-3-i)            s = s[:i] + '...' + s[len(s)-j:]        return s     def repr_instance(self, x, level):        try:            s = __builtin__.repr(x)            # Bugs in x.__repr__() can cause arbitrary            # exceptions -- then make up something        except:            return '<%s instance at %x>' % (x.__class__.__name__, id(x))        if len(s) > self.maxstring:            i = max(0, (self.maxstring-3)//2)            j = max(0, self.maxstring-3-i)            s = s[:i] + '...' + s[len(s)-j:]        return s aRepr = Repr()repr = aRepr.repr