home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / System / pprint.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-11-09  |  9.3 KB  |  353 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. """Support to pretty-print lists, tuples, & dictionaries recursively.
  5.  
  6. Very simple, but useful, especially in debugging data structures.
  7.  
  8. Classes
  9. -------
  10.  
  11. PrettyPrinter()
  12.     Handle pretty-printing operations onto a stream using a configured
  13.     set of formatting parameters.
  14.  
  15. Functions
  16. ---------
  17.  
  18. pformat()
  19.     Format a Python object into a pretty-printed representation.
  20.  
  21. pprint()
  22.     Pretty-print a Python object to a stream [default is sys.stdout].
  23.  
  24. saferepr()
  25.     Generate a 'standard' repr()-like value, but protect against recursive
  26.     data structures.
  27.  
  28. """
  29. import sys as _sys
  30. from cStringIO import StringIO as _StringIO
  31. __all__ = [
  32.     'pprint',
  33.     'pformat',
  34.     'isreadable',
  35.     'isrecursive',
  36.     'saferepr',
  37.     'PrettyPrinter']
  38. _commajoin = ', '.join
  39. _id = id
  40. _len = len
  41. _type = type
  42.  
  43. def pprint(object, stream = None, indent = 1, width = 80, depth = None):
  44.     '''Pretty-print a Python object to a stream [default is sys.stdout].'''
  45.     printer = PrettyPrinter(stream = stream, indent = indent, width = width, depth = depth)
  46.     printer.pprint(object)
  47.  
  48.  
  49. def pformat(object, indent = 1, width = 80, depth = None):
  50.     '''Format a Python object into a pretty-printed representation.'''
  51.     return PrettyPrinter(indent = indent, width = width, depth = depth).pformat(object)
  52.  
  53.  
  54. def saferepr(object):
  55.     '''Version of repr() which can handle recursive data structures.'''
  56.     return _safe_repr(object, { }, None, 0)[0]
  57.  
  58.  
  59. def isreadable(object):
  60.     '''Determine if saferepr(object) is readable by eval().'''
  61.     return _safe_repr(object, { }, None, 0)[1]
  62.  
  63.  
  64. def isrecursive(object):
  65.     '''Determine if object requires a recursive representation.'''
  66.     return _safe_repr(object, { }, None, 0)[2]
  67.  
  68.  
  69. class PrettyPrinter:
  70.     
  71.     def __init__(self, indent = 1, width = 80, depth = None, stream = None):
  72.         '''Handle pretty printing operations onto a stream using a set of
  73.         configured parameters.
  74.  
  75.         indent
  76.             Number of spaces to indent for each level of nesting.
  77.  
  78.         width
  79.             Attempted maximum number of columns in the output.
  80.  
  81.         depth
  82.             The maximum depth to print out nested structures.
  83.  
  84.         stream
  85.             The desired output stream.  If omitted (or false), the standard
  86.             output stream available at construction will be used.
  87.  
  88.         '''
  89.         indent = int(indent)
  90.         width = int(width)
  91.         if not indent >= 0:
  92.             raise AssertionError, 'indent must be >= 0'
  93.         if not depth is None and depth > 0:
  94.             raise AssertionError, 'depth must be > 0'
  95.         if not width:
  96.             raise AssertionError, 'width must be != 0'
  97.         self._depth = depth
  98.         self._indent_per_level = indent
  99.         self._width = width
  100.         if stream is not None:
  101.             self._stream = stream
  102.         else:
  103.             self._stream = _sys.stdout
  104.  
  105.     
  106.     def pprint(self, object):
  107.         self._stream.write(self.pformat(object) + '\n')
  108.  
  109.     
  110.     def pformat(self, object):
  111.         sio = _StringIO()
  112.         self._format(object, sio, 0, 0, { }, 0)
  113.         return sio.getvalue()
  114.  
  115.     
  116.     def isrecursive(self, object):
  117.         return self.format(object, { }, 0, 0)[2]
  118.  
  119.     
  120.     def isreadable(self, object):
  121.         (s, readable, recursive) = self.format(object, { }, 0, 0)
  122.         if readable:
  123.             pass
  124.         return not recursive
  125.  
  126.     
  127.     def _format(self, object, stream, indent, allowance, context, level):
  128.         level = level + 1
  129.         objid = _id(object)
  130.         if objid in context:
  131.             stream.write(_recursion(object))
  132.             self._recursive = True
  133.             self._readable = False
  134.             return None
  135.         
  136.         rep = self._repr(object, context, level - 1)
  137.         typ = _type(object)
  138.         sepLines = _len(rep) > self._width - 1 - indent - allowance
  139.         write = stream.write
  140.         if sepLines:
  141.             r = getattr(typ, '__repr__', None)
  142.             if issubclass(typ, dict) and r is dict.__repr__:
  143.                 write('{')
  144.                 if self._indent_per_level > 1:
  145.                     write((self._indent_per_level - 1) * ' ')
  146.                 
  147.                 length = _len(object)
  148.                 if length:
  149.                     context[objid] = 1
  150.                     indent = indent + self._indent_per_level
  151.                     items = object.items()
  152.                     items.sort()
  153.                     (key, ent) = items[0]
  154.                     rep = self._repr(key, context, level)
  155.                     write(rep)
  156.                     write(': ')
  157.                     self._format(ent, stream, indent + _len(rep) + 2, allowance + 1, context, level)
  158.                     if length > 1:
  159.                         for key, ent in items[1:]:
  160.                             rep = self._repr(key, context, level)
  161.                             write(',\n%s%s: ' % (' ' * indent, rep))
  162.                             self._format(ent, stream, indent + _len(rep) + 2, allowance + 1, context, level)
  163.                         
  164.                     
  165.                     indent = indent - self._indent_per_level
  166.                     del context[objid]
  167.                 
  168.                 write('}')
  169.                 return None
  170.             
  171.             if (issubclass(typ, list) or r is list.__repr__ or issubclass(typ, tuple)) and r is tuple.__repr__:
  172.                 if issubclass(typ, list):
  173.                     write('[')
  174.                     endchar = ']'
  175.                 else:
  176.                     write('(')
  177.                     endchar = ')'
  178.                 if self._indent_per_level > 1:
  179.                     write((self._indent_per_level - 1) * ' ')
  180.                 
  181.                 length = _len(object)
  182.                 if length:
  183.                     context[objid] = 1
  184.                     indent = indent + self._indent_per_level
  185.                     self._format(object[0], stream, indent, allowance + 1, context, level)
  186.                     if length > 1:
  187.                         for ent in object[1:]:
  188.                             write(',\n' + ' ' * indent)
  189.                             self._format(ent, stream, indent, allowance + 1, context, level)
  190.                         
  191.                     
  192.                     indent = indent - self._indent_per_level
  193.                     del context[objid]
  194.                 
  195.                 if issubclass(typ, tuple) and length == 1:
  196.                     write(',')
  197.                 
  198.                 write(endchar)
  199.                 return None
  200.             
  201.         
  202.         write(rep)
  203.  
  204.     
  205.     def _repr(self, object, context, level):
  206.         (repr, readable, recursive) = self.format(object, context.copy(), self._depth, level)
  207.         if not readable:
  208.             self._readable = False
  209.         
  210.         if recursive:
  211.             self._recursive = True
  212.         
  213.         return repr
  214.  
  215.     
  216.     def format(self, object, context, maxlevels, level):
  217.         """Format object for a specific context, returning a string
  218.         and flags indicating whether the representation is 'readable'
  219.         and whether the object represents a recursive construct.
  220.         """
  221.         return _safe_repr(object, context, maxlevels, level)
  222.  
  223.  
  224.  
  225. def _safe_repr(object, context, maxlevels, level):
  226.     typ = _type(object)
  227.     if typ is str:
  228.         if 'locale' not in _sys.modules:
  229.             return (repr(object), True, False)
  230.         
  231.         if "'" in object and '"' not in object:
  232.             closure = '"'
  233.             quotes = {
  234.                 '"': '\\"' }
  235.         else:
  236.             closure = "'"
  237.             quotes = {
  238.                 "'": "\\'" }
  239.         qget = quotes.get
  240.         sio = _StringIO()
  241.         write = sio.write
  242.         for char in object:
  243.             if char.isalpha():
  244.                 write(char)
  245.                 continue
  246.             write(qget(char, repr(char)[1:-1]))
  247.         
  248.         return ('%s%s%s' % (closure, sio.getvalue(), closure), True, False)
  249.     
  250.     r = getattr(typ, '__repr__', None)
  251.     if issubclass(typ, dict) and r is dict.__repr__:
  252.         if not object:
  253.             return ('{}', True, False)
  254.         
  255.         objid = _id(object)
  256.         if maxlevels and level > maxlevels:
  257.             return ('{...}', False, objid in context)
  258.         
  259.         if objid in context:
  260.             return (_recursion(object), False, True)
  261.         
  262.         context[objid] = 1
  263.         readable = True
  264.         recursive = False
  265.         components = []
  266.         append = components.append
  267.         level += 1
  268.         saferepr = _safe_repr
  269.         for k, v in object.iteritems():
  270.             (krepr, kreadable, krecur) = saferepr(k, context, maxlevels, level)
  271.             (vrepr, vreadable, vrecur) = saferepr(v, context, maxlevels, level)
  272.             append('%s: %s' % (krepr, vrepr))
  273.             if readable and kreadable:
  274.                 pass
  275.             readable = vreadable
  276.             if krecur or vrecur:
  277.                 recursive = True
  278.                 continue
  279.         
  280.         del context[objid]
  281.         return ('{%s}' % _commajoin(components), readable, recursive)
  282.     
  283.     if (issubclass(typ, list) or r is list.__repr__ or issubclass(typ, tuple)) and r is tuple.__repr__:
  284.         if issubclass(typ, list):
  285.             if not object:
  286.                 return ('[]', True, False)
  287.             
  288.             format = '[%s]'
  289.         elif _len(object) == 1:
  290.             format = '(%s,)'
  291.         elif not object:
  292.             return ('()', True, False)
  293.         
  294.         format = '(%s)'
  295.         objid = _id(object)
  296.         if maxlevels and level > maxlevels:
  297.             return (format % '...', False, objid in context)
  298.         
  299.         if objid in context:
  300.             return (_recursion(object), False, True)
  301.         
  302.         context[objid] = 1
  303.         readable = True
  304.         recursive = False
  305.         components = []
  306.         append = components.append
  307.         level += 1
  308.         for o in object:
  309.             (orepr, oreadable, orecur) = _safe_repr(o, context, maxlevels, level)
  310.             append(orepr)
  311.             if not oreadable:
  312.                 readable = False
  313.             
  314.             if orecur:
  315.                 recursive = True
  316.                 continue
  317.         
  318.         del context[objid]
  319.         return (format % _commajoin(components), readable, recursive)
  320.     
  321.     rep = repr(object)
  322.     if rep:
  323.         pass
  324.     return (rep, not rep.startswith('<'), False)
  325.  
  326.  
  327. def _recursion(object):
  328.     return '<Recursion on %s with id=%s>' % (_type(object).__name__, _id(object))
  329.  
  330.  
  331. def _perfcheck(object = None):
  332.     import time
  333.     if object is None:
  334.         object = [
  335.             ('string', (1, 2), [
  336.                 3,
  337.                 4], {
  338.                 5: 6,
  339.                 7: 8 })] * 100000
  340.     
  341.     p = PrettyPrinter()
  342.     t1 = time.time()
  343.     _safe_repr(object, { }, None, 0)
  344.     t2 = time.time()
  345.     p.pformat(object)
  346.     t3 = time.time()
  347.     print '_safe_repr:', t2 - t1
  348.     print 'pformat:', t3 - t2
  349.  
  350. if __name__ == '__main__':
  351.     _perfcheck()
  352.  
  353.