reprlib32.py 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. """Redo the builtin repr() (representation) but with limits on most sizes."""
  2. __all__ = ["Repr", "repr", "recursive_repr"]
  3. import __builtin__ as builtins
  4. from itertools import islice
  5. try:
  6. from thread import get_ident
  7. except ImportError:
  8. from _dummy_thread32 import get_ident
  9. def recursive_repr(fillvalue='...'):
  10. 'Decorator to make a repr function return fillvalue for a recursive call'
  11. def decorating_function(user_function):
  12. repr_running = set()
  13. def wrapper(self):
  14. key = id(self), get_ident()
  15. if key in repr_running:
  16. return fillvalue
  17. repr_running.add(key)
  18. try:
  19. result = user_function(self)
  20. finally:
  21. repr_running.discard(key)
  22. return result
  23. # Can't use functools.wraps() here because of bootstrap issues
  24. wrapper.__module__ = getattr(user_function, '__module__')
  25. wrapper.__doc__ = getattr(user_function, '__doc__')
  26. wrapper.__name__ = getattr(user_function, '__name__')
  27. wrapper.__annotations__ = getattr(user_function, '__annotations__', {})
  28. return wrapper
  29. return decorating_function
  30. class Repr:
  31. def __init__(self):
  32. self.maxlevel = 6
  33. self.maxtuple = 6
  34. self.maxlist = 6
  35. self.maxarray = 5
  36. self.maxdict = 4
  37. self.maxset = 6
  38. self.maxfrozenset = 6
  39. self.maxdeque = 6
  40. self.maxstring = 30
  41. self.maxlong = 40
  42. self.maxother = 30
  43. def repr(self, x):
  44. return self.repr1(x, self.maxlevel)
  45. def repr1(self, x, level):
  46. typename = type(x).__name__
  47. if ' ' in typename:
  48. parts = typename.split()
  49. typename = '_'.join(parts)
  50. if hasattr(self, 'repr_' + typename):
  51. return getattr(self, 'repr_' + typename)(x, level)
  52. else:
  53. return self.repr_instance(x, level)
  54. def _repr_iterable(self, x, level, left, right, maxiter, trail=''):
  55. n = len(x)
  56. if level <= 0 and n:
  57. s = '...'
  58. else:
  59. newlevel = level - 1
  60. repr1 = self.repr1
  61. pieces = [repr1(elem, newlevel) for elem in islice(x, maxiter)]
  62. if n > maxiter: pieces.append('...')
  63. s = ', '.join(pieces)
  64. if n == 1 and trail: right = trail + right
  65. return '%s%s%s' % (left, s, right)
  66. def repr_tuple(self, x, level):
  67. return self._repr_iterable(x, level, '(', ')', self.maxtuple, ',')
  68. def repr_list(self, x, level):
  69. return self._repr_iterable(x, level, '[', ']', self.maxlist)
  70. def repr_array(self, x, level):
  71. header = "array('%s', [" % x.typecode
  72. return self._repr_iterable(x, level, header, '])', self.maxarray)
  73. def repr_set(self, x, level):
  74. x = _possibly_sorted(x)
  75. return self._repr_iterable(x, level, 'set([', '])', self.maxset)
  76. def repr_frozenset(self, x, level):
  77. x = _possibly_sorted(x)
  78. return self._repr_iterable(x, level, 'frozenset([', '])',
  79. self.maxfrozenset)
  80. def repr_deque(self, x, level):
  81. return self._repr_iterable(x, level, 'deque([', '])', self.maxdeque)
  82. def repr_dict(self, x, level):
  83. n = len(x)
  84. if n == 0: return '{}'
  85. if level <= 0: return '{...}'
  86. newlevel = level - 1
  87. repr1 = self.repr1
  88. pieces = []
  89. for key in islice(_possibly_sorted(x), self.maxdict):
  90. keyrepr = repr1(key, newlevel)
  91. valrepr = repr1(x[key], newlevel)
  92. pieces.append('%s: %s' % (keyrepr, valrepr))
  93. if n > self.maxdict: pieces.append('...')
  94. s = ', '.join(pieces)
  95. return '{%s}' % (s,)
  96. def repr_str(self, x, level):
  97. s = builtins.repr(x[:self.maxstring])
  98. if len(s) > self.maxstring:
  99. i = max(0, (self.maxstring-3)//2)
  100. j = max(0, self.maxstring-3-i)
  101. s = builtins.repr(x[:i] + x[len(x)-j:])
  102. s = s[:i] + '...' + s[len(s)-j:]
  103. return s
  104. def repr_int(self, x, level):
  105. s = builtins.repr(x) # XXX Hope this isn't too slow...
  106. if len(s) > self.maxlong:
  107. i = max(0, (self.maxlong-3)//2)
  108. j = max(0, self.maxlong-3-i)
  109. s = s[:i] + '...' + s[len(s)-j:]
  110. return s
  111. def repr_instance(self, x, level):
  112. try:
  113. s = builtins.repr(x)
  114. # Bugs in x.__repr__() can cause arbitrary
  115. # exceptions -- then make up something
  116. except Exception:
  117. return '<%s instance at %x>' % (x.__class__.__name__, id(x))
  118. if len(s) > self.maxother:
  119. i = max(0, (self.maxother-3)//2)
  120. j = max(0, self.maxother-3-i)
  121. s = s[:i] + '...' + s[len(s)-j:]
  122. return s
  123. def _possibly_sorted(x):
  124. # Since not all sequences of items can be sorted and comparison
  125. # functions may raise arbitrary exceptions, return an unsorted
  126. # sequence in that case.
  127. try:
  128. return sorted(x)
  129. except Exception:
  130. return list(x)
  131. aRepr = Repr()
  132. repr = aRepr.repr