Lib/lib-old/tb.py

   1 # Print tracebacks, with a dump of local variables.
   2 # Also an interactive stack trace browser.
   3 # Note -- this module is obsolete -- use pdb.pm() instead.
   4
   5 import sys
   6 import os
   7 from stat import *
   8 import string
   9 import linecache
  10
  11 def br(): browser(sys.last_traceback)
  12
  13 def tb(): printtb(sys.last_traceback)
  14
  15 def browser(tb):
  16         if not tb:
  17                 print 'No traceback.'
  18                 return
  19         tblist = []
  20         while tb:
  21                 tblist.append(tb)
  22                 tb = tb.tb_next
  23         ptr = len(tblist)-1
  24         tb = tblist[ptr]
  25         while 1:
  26                 if tb <> tblist[ptr]:
  27                         tb = tblist[ptr]
  28                         print `ptr` + ':',
  29                         printtbheader(tb)
  30                 try:
  31                         line = raw_input('TB: ')
  32                 except KeyboardInterrupt:
  33                         print '\n[Interrupted]'
  34                         break
  35                 except EOFError:
  36                         print '\n[EOF]'
  37                         break
  38                 cmd = string.strip(line)
  39                 if cmd:
  40                         if cmd == 'quit':
  41                                 break
  42                         elif cmd == 'list':
  43                                 browserlist(tb)
  44                         elif cmd == 'up':
  45                                 if ptr-1 >= 0: ptr = ptr-1
  46                                 else: print 'Bottom of stack.'
  47                         elif cmd == 'down':
  48                                 if ptr+1 < len(tblist): ptr = ptr+1
  49                                 else: print 'Top of stack.'
  50                         elif cmd == 'locals':
  51                                 printsymbols(tb.tb_frame.f_locals)
  52                         elif cmd == 'globals':
  53                                 printsymbols(tb.tb_frame.f_globals)
  54                         elif cmd in ('?', 'help'):
  55                                 browserhelp()
  56                         else:
  57                                 browserexec(tb, cmd)
  58
  59 def browserlist(tb):
  60         filename = tb.tb_frame.f_code.co_filename
  61         lineno = tb.tb_lineno
  62         last = lineno
  63         first = max(1, last-10)
  64         for i in range(first, last+1):
  65                 if i == lineno: prefix = '***' + string.rjust(`i`, 4) + ':'
  66                 else: prefix = string.rjust(`i`, 7) + ':'
  67                 line = linecache.getline(filename, i)
  68                 if line[-1:] == '\n': line = line[:-1]
  69                 print prefix + line
  70
  71 def browserexec(tb, cmd):
  72         locals = tb.tb_frame.f_locals
  73         globals = tb.tb_frame.f_globals
  74         try:
  75                 exec cmd+'\n' in globals, locals
  76         except:
  77                 t, v = sys.exc_info()[:2]
  78                 print '*** Exception:',
  79                 if type(t) == type(''):
  80                         print t,
  81                 else:
  82                         print t.__name__,
  83                 if v <> None:
  84                         print ':', v,
  85                 print
  86                 print 'Type help to get help.'
  87
  88 def browserhelp():
  89         print
  90         print '    This is the traceback browser.  Commands are:'
  91         print '        up      : move one level up in the call stack'
  92         print '        down    : move one level down in the call stack'
  93         print '        locals  : print all local variables at this level'
  94         print '        globals : print all global variables at this level'
  95         print '        list    : list source code around the failure'
  96         print '        help    : print help (what you are reading now)'
  97         print '        quit    : back to command interpreter'
  98         print '    Typing any other 1-line statement will execute it'
  99         print '    using the current level\'s symbol tables'
 100         print
 101
 102 def printtb(tb):
 103         while tb:
 104                 print1tb(tb)
 105                 tb = tb.tb_next
 106
 107 def print1tb(tb):
 108         printtbheader(tb)
 109         if tb.tb_frame.f_locals is not tb.tb_frame.f_globals:
 110                 printsymbols(tb.tb_frame.f_locals)
 111
 112 def printtbheader(tb):
 113         filename = tb.tb_frame.f_code.co_filename
 114         lineno = tb.tb_lineno
 115         info = '"' + filename + '"(' + `lineno` + ')'
 116         line = linecache.getline(filename, lineno)
 117         if line:
 118                 info = info + ': ' + string.strip(line)
 119         print info
 120
 121 def printsymbols(d):
 122         keys = d.keys()
 123         keys.sort()
 124         for name in keys:
 125                 print '  ' + string.ljust(name, 12) + ':',
 126                 printobject(d[name], 4)
 127                 print
 128
 129 def printobject(v, maxlevel):
 130         if v == None:
 131                 print 'None',
 132         elif type(v) in (type(0), type(0.0)):
 133                 print v,
 134         elif type(v) == type(''):
 135                 if len(v) > 20:
 136                         print `v[:17] + '...'`,
 137                 else:
 138                         print `v`,
 139         elif type(v) == type(()):
 140                 print '(',
 141                 printlist(v, maxlevel)
 142                 print ')',
 143         elif type(v) == type([]):
 144                 print '[',
 145                 printlist(v, maxlevel)
 146                 print ']',
 147         elif type(v) == type({}):
 148                 print '{',
 149                 printdict(v, maxlevel)
 150                 print '}',
 151         else:
 152                 print v,
 153
 154 def printlist(v, maxlevel):
 155         n = len(v)
 156         if n == 0: return
 157         if maxlevel <= 0:
 158                 print '...',
 159                 return
 160         for i in range(min(6, n)):
 161                 printobject(v[i], maxlevel-1)
 162                 if i+1 < n: print ',',
 163         if n > 6: print '...',
 164
 165 def printdict(v, maxlevel):
 166         keys = v.keys()
 167         n = len(keys)
 168         if n == 0: return
 169         if maxlevel <= 0:
 170                 print '...',
 171                 return
 172         keys.sort()
 173         for i in range(min(6, n)):
 174                 key = keys[i]
 175                 print `key` + ':',
 176                 printobject(v[key], maxlevel-1)
 177                 if i+1 < n: print ',',
 178         if n > 6: print '...',