| """Redo the builtin repr() (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.maxtuple, ',') | |
| 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 = _possibly_sorted(x) | |
| return self._repr_iterable(x, level, 'set([', '])', self.maxset) | |
| def repr_frozenset(self, x, level): | |
| x = _possibly_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(_possibly_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 Exception: | |
| 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 | |
| def _possibly_sorted(x): | |
| # Since not all sequences of items can be sorted and comparison | |
| # functions may raise arbitrary exceptions, return an unsorted | |
| # sequence in that case. | |
| try: | |
| return sorted(x) | |
| except Exception: | |
| return list(x) | |
| aRepr = Repr() | |
| repr = aRepr.repr |