python-2.5.2/win32/Lib/bdb.py
changeset 0 ae805ac0140d
equal deleted inserted replaced
-1:000000000000 0:ae805ac0140d
       
     1 """Debugger basics"""
       
     2 
       
     3 import sys
       
     4 import os
       
     5 import types
       
     6 
       
     7 __all__ = ["BdbQuit","Bdb","Breakpoint"]
       
     8 
       
     9 class BdbQuit(Exception):
       
    10     """Exception to give up completely"""
       
    11 
       
    12 
       
    13 class Bdb:
       
    14 
       
    15     """Generic Python debugger base class.
       
    16 
       
    17     This class takes care of details of the trace facility;
       
    18     a derived class should implement user interaction.
       
    19     The standard debugger class (pdb.Pdb) is an example.
       
    20     """
       
    21 
       
    22     def __init__(self):
       
    23         self.breaks = {}
       
    24         self.fncache = {}
       
    25 
       
    26     def canonic(self, filename):
       
    27         if filename == "<" + filename[1:-1] + ">":
       
    28             return filename
       
    29         canonic = self.fncache.get(filename)
       
    30         if not canonic:
       
    31             canonic = os.path.abspath(filename)
       
    32             canonic = os.path.normcase(canonic)
       
    33             self.fncache[filename] = canonic
       
    34         return canonic
       
    35 
       
    36     def reset(self):
       
    37         import linecache
       
    38         linecache.checkcache()
       
    39         self.botframe = None
       
    40         self.stopframe = None
       
    41         self.returnframe = None
       
    42         self.quitting = 0
       
    43 
       
    44     def trace_dispatch(self, frame, event, arg):
       
    45         if self.quitting:
       
    46             return # None
       
    47         if event == 'line':
       
    48             return self.dispatch_line(frame)
       
    49         if event == 'call':
       
    50             return self.dispatch_call(frame, arg)
       
    51         if event == 'return':
       
    52             return self.dispatch_return(frame, arg)
       
    53         if event == 'exception':
       
    54             return self.dispatch_exception(frame, arg)
       
    55         if event == 'c_call':
       
    56             return self.trace_dispatch
       
    57         if event == 'c_exception':
       
    58             return self.trace_dispatch
       
    59         if event == 'c_return':
       
    60             return self.trace_dispatch
       
    61         print 'bdb.Bdb.dispatch: unknown debugging event:', repr(event)
       
    62         return self.trace_dispatch
       
    63 
       
    64     def dispatch_line(self, frame):
       
    65         if self.stop_here(frame) or self.break_here(frame):
       
    66             self.user_line(frame)
       
    67             if self.quitting: raise BdbQuit
       
    68         return self.trace_dispatch
       
    69 
       
    70     def dispatch_call(self, frame, arg):
       
    71         # XXX 'arg' is no longer used
       
    72         if self.botframe is None:
       
    73             # First call of dispatch since reset()
       
    74             self.botframe = frame.f_back # (CT) Note that this may also be None!
       
    75             return self.trace_dispatch
       
    76         if not (self.stop_here(frame) or self.break_anywhere(frame)):
       
    77             # No need to trace this function
       
    78             return # None
       
    79         self.user_call(frame, arg)
       
    80         if self.quitting: raise BdbQuit
       
    81         return self.trace_dispatch
       
    82 
       
    83     def dispatch_return(self, frame, arg):
       
    84         if self.stop_here(frame) or frame == self.returnframe:
       
    85             self.user_return(frame, arg)
       
    86             if self.quitting: raise BdbQuit
       
    87         return self.trace_dispatch
       
    88 
       
    89     def dispatch_exception(self, frame, arg):
       
    90         if self.stop_here(frame):
       
    91             self.user_exception(frame, arg)
       
    92             if self.quitting: raise BdbQuit
       
    93         return self.trace_dispatch
       
    94 
       
    95     # Normally derived classes don't override the following
       
    96     # methods, but they may if they want to redefine the
       
    97     # definition of stopping and breakpoints.
       
    98 
       
    99     def stop_here(self, frame):
       
   100         # (CT) stopframe may now also be None, see dispatch_call.
       
   101         # (CT) the former test for None is therefore removed from here.
       
   102         if frame is self.stopframe:
       
   103             return True
       
   104         while frame is not None and frame is not self.stopframe:
       
   105             if frame is self.botframe:
       
   106                 return True
       
   107             frame = frame.f_back
       
   108         return False
       
   109 
       
   110     def break_here(self, frame):
       
   111         filename = self.canonic(frame.f_code.co_filename)
       
   112         if not filename in self.breaks:
       
   113             return False
       
   114         lineno = frame.f_lineno
       
   115         if not lineno in self.breaks[filename]:
       
   116             # The line itself has no breakpoint, but maybe the line is the
       
   117             # first line of a function with breakpoint set by function name.
       
   118             lineno = frame.f_code.co_firstlineno
       
   119             if not lineno in self.breaks[filename]:
       
   120                 return False
       
   121 
       
   122         # flag says ok to delete temp. bp
       
   123         (bp, flag) = effective(filename, lineno, frame)
       
   124         if bp:
       
   125             self.currentbp = bp.number
       
   126             if (flag and bp.temporary):
       
   127                 self.do_clear(str(bp.number))
       
   128             return True
       
   129         else:
       
   130             return False
       
   131 
       
   132     def do_clear(self, arg):
       
   133         raise NotImplementedError, "subclass of bdb must implement do_clear()"
       
   134 
       
   135     def break_anywhere(self, frame):
       
   136         return self.breaks.has_key(
       
   137             self.canonic(frame.f_code.co_filename))
       
   138 
       
   139     # Derived classes should override the user_* methods
       
   140     # to gain control.
       
   141 
       
   142     def user_call(self, frame, argument_list):
       
   143         """This method is called when there is the remote possibility
       
   144         that we ever need to stop in this function."""
       
   145         pass
       
   146 
       
   147     def user_line(self, frame):
       
   148         """This method is called when we stop or break at this line."""
       
   149         pass
       
   150 
       
   151     def user_return(self, frame, return_value):
       
   152         """This method is called when a return trap is set here."""
       
   153         pass
       
   154 
       
   155     def user_exception(self, frame, (exc_type, exc_value, exc_traceback)):
       
   156         """This method is called if an exception occurs,
       
   157         but only if we are to stop at or just below this level."""
       
   158         pass
       
   159 
       
   160     # Derived classes and clients can call the following methods
       
   161     # to affect the stepping state.
       
   162 
       
   163     def set_step(self):
       
   164         """Stop after one line of code."""
       
   165         self.stopframe = None
       
   166         self.returnframe = None
       
   167         self.quitting = 0
       
   168 
       
   169     def set_next(self, frame):
       
   170         """Stop on the next line in or below the given frame."""
       
   171         self.stopframe = frame
       
   172         self.returnframe = None
       
   173         self.quitting = 0
       
   174 
       
   175     def set_return(self, frame):
       
   176         """Stop when returning from the given frame."""
       
   177         self.stopframe = frame.f_back
       
   178         self.returnframe = frame
       
   179         self.quitting = 0
       
   180 
       
   181     def set_trace(self, frame=None):
       
   182         """Start debugging from `frame`.
       
   183 
       
   184         If frame is not specified, debugging starts from caller's frame.
       
   185         """
       
   186         if frame is None:
       
   187             frame = sys._getframe().f_back
       
   188         self.reset()
       
   189         while frame:
       
   190             frame.f_trace = self.trace_dispatch
       
   191             self.botframe = frame
       
   192             frame = frame.f_back
       
   193         self.set_step()
       
   194         sys.settrace(self.trace_dispatch)
       
   195 
       
   196     def set_continue(self):
       
   197         # Don't stop except at breakpoints or when finished
       
   198         self.stopframe = self.botframe
       
   199         self.returnframe = None
       
   200         self.quitting = 0
       
   201         if not self.breaks:
       
   202             # no breakpoints; run without debugger overhead
       
   203             sys.settrace(None)
       
   204             frame = sys._getframe().f_back
       
   205             while frame and frame is not self.botframe:
       
   206                 del frame.f_trace
       
   207                 frame = frame.f_back
       
   208 
       
   209     def set_quit(self):
       
   210         self.stopframe = self.botframe
       
   211         self.returnframe = None
       
   212         self.quitting = 1
       
   213         sys.settrace(None)
       
   214 
       
   215     # Derived classes and clients can call the following methods
       
   216     # to manipulate breakpoints.  These methods return an
       
   217     # error message is something went wrong, None if all is well.
       
   218     # Set_break prints out the breakpoint line and file:lineno.
       
   219     # Call self.get_*break*() to see the breakpoints or better
       
   220     # for bp in Breakpoint.bpbynumber: if bp: bp.bpprint().
       
   221 
       
   222     def set_break(self, filename, lineno, temporary=0, cond = None,
       
   223                   funcname=None):
       
   224         filename = self.canonic(filename)
       
   225         import linecache # Import as late as possible
       
   226         line = linecache.getline(filename, lineno)
       
   227         if not line:
       
   228             return 'Line %s:%d does not exist' % (filename,
       
   229                                    lineno)
       
   230         if not filename in self.breaks:
       
   231             self.breaks[filename] = []
       
   232         list = self.breaks[filename]
       
   233         if not lineno in list:
       
   234             list.append(lineno)
       
   235         bp = Breakpoint(filename, lineno, temporary, cond, funcname)
       
   236 
       
   237     def clear_break(self, filename, lineno):
       
   238         filename = self.canonic(filename)
       
   239         if not filename in self.breaks:
       
   240             return 'There are no breakpoints in %s' % filename
       
   241         if lineno not in self.breaks[filename]:
       
   242             return 'There is no breakpoint at %s:%d' % (filename,
       
   243                                     lineno)
       
   244         # If there's only one bp in the list for that file,line
       
   245         # pair, then remove the breaks entry
       
   246         for bp in Breakpoint.bplist[filename, lineno][:]:
       
   247             bp.deleteMe()
       
   248         if not Breakpoint.bplist.has_key((filename, lineno)):
       
   249             self.breaks[filename].remove(lineno)
       
   250         if not self.breaks[filename]:
       
   251             del self.breaks[filename]
       
   252 
       
   253     def clear_bpbynumber(self, arg):
       
   254         try:
       
   255             number = int(arg)
       
   256         except:
       
   257             return 'Non-numeric breakpoint number (%s)' % arg
       
   258         try:
       
   259             bp = Breakpoint.bpbynumber[number]
       
   260         except IndexError:
       
   261             return 'Breakpoint number (%d) out of range' % number
       
   262         if not bp:
       
   263             return 'Breakpoint (%d) already deleted' % number
       
   264         self.clear_break(bp.file, bp.line)
       
   265 
       
   266     def clear_all_file_breaks(self, filename):
       
   267         filename = self.canonic(filename)
       
   268         if not filename in self.breaks:
       
   269             return 'There are no breakpoints in %s' % filename
       
   270         for line in self.breaks[filename]:
       
   271             blist = Breakpoint.bplist[filename, line]
       
   272             for bp in blist:
       
   273                 bp.deleteMe()
       
   274         del self.breaks[filename]
       
   275 
       
   276     def clear_all_breaks(self):
       
   277         if not self.breaks:
       
   278             return 'There are no breakpoints'
       
   279         for bp in Breakpoint.bpbynumber:
       
   280             if bp:
       
   281                 bp.deleteMe()
       
   282         self.breaks = {}
       
   283 
       
   284     def get_break(self, filename, lineno):
       
   285         filename = self.canonic(filename)
       
   286         return filename in self.breaks and \
       
   287             lineno in self.breaks[filename]
       
   288 
       
   289     def get_breaks(self, filename, lineno):
       
   290         filename = self.canonic(filename)
       
   291         return filename in self.breaks and \
       
   292             lineno in self.breaks[filename] and \
       
   293             Breakpoint.bplist[filename, lineno] or []
       
   294 
       
   295     def get_file_breaks(self, filename):
       
   296         filename = self.canonic(filename)
       
   297         if filename in self.breaks:
       
   298             return self.breaks[filename]
       
   299         else:
       
   300             return []
       
   301 
       
   302     def get_all_breaks(self):
       
   303         return self.breaks
       
   304 
       
   305     # Derived classes and clients can call the following method
       
   306     # to get a data structure representing a stack trace.
       
   307 
       
   308     def get_stack(self, f, t):
       
   309         stack = []
       
   310         if t and t.tb_frame is f:
       
   311             t = t.tb_next
       
   312         while f is not None:
       
   313             stack.append((f, f.f_lineno))
       
   314             if f is self.botframe:
       
   315                 break
       
   316             f = f.f_back
       
   317         stack.reverse()
       
   318         i = max(0, len(stack) - 1)
       
   319         while t is not None:
       
   320             stack.append((t.tb_frame, t.tb_lineno))
       
   321             t = t.tb_next
       
   322         return stack, i
       
   323 
       
   324     #
       
   325 
       
   326     def format_stack_entry(self, frame_lineno, lprefix=': '):
       
   327         import linecache, repr
       
   328         frame, lineno = frame_lineno
       
   329         filename = self.canonic(frame.f_code.co_filename)
       
   330         s = '%s(%r)' % (filename, lineno)
       
   331         if frame.f_code.co_name:
       
   332             s = s + frame.f_code.co_name
       
   333         else:
       
   334             s = s + "<lambda>"
       
   335         if '__args__' in frame.f_locals:
       
   336             args = frame.f_locals['__args__']
       
   337         else:
       
   338             args = None
       
   339         if args:
       
   340             s = s + repr.repr(args)
       
   341         else:
       
   342             s = s + '()'
       
   343         if '__return__' in frame.f_locals:
       
   344             rv = frame.f_locals['__return__']
       
   345             s = s + '->'
       
   346             s = s + repr.repr(rv)
       
   347         line = linecache.getline(filename, lineno)
       
   348         if line: s = s + lprefix + line.strip()
       
   349         return s
       
   350 
       
   351     # The following two methods can be called by clients to use
       
   352     # a debugger to debug a statement, given as a string.
       
   353 
       
   354     def run(self, cmd, globals=None, locals=None):
       
   355         if globals is None:
       
   356             import __main__
       
   357             globals = __main__.__dict__
       
   358         if locals is None:
       
   359             locals = globals
       
   360         self.reset()
       
   361         sys.settrace(self.trace_dispatch)
       
   362         if not isinstance(cmd, types.CodeType):
       
   363             cmd = cmd+'\n'
       
   364         try:
       
   365             try:
       
   366                 exec cmd in globals, locals
       
   367             except BdbQuit:
       
   368                 pass
       
   369         finally:
       
   370             self.quitting = 1
       
   371             sys.settrace(None)
       
   372 
       
   373     def runeval(self, expr, globals=None, locals=None):
       
   374         if globals is None:
       
   375             import __main__
       
   376             globals = __main__.__dict__
       
   377         if locals is None:
       
   378             locals = globals
       
   379         self.reset()
       
   380         sys.settrace(self.trace_dispatch)
       
   381         if not isinstance(expr, types.CodeType):
       
   382             expr = expr+'\n'
       
   383         try:
       
   384             try:
       
   385                 return eval(expr, globals, locals)
       
   386             except BdbQuit:
       
   387                 pass
       
   388         finally:
       
   389             self.quitting = 1
       
   390             sys.settrace(None)
       
   391 
       
   392     def runctx(self, cmd, globals, locals):
       
   393         # B/W compatibility
       
   394         self.run(cmd, globals, locals)
       
   395 
       
   396     # This method is more useful to debug a single function call.
       
   397 
       
   398     def runcall(self, func, *args, **kwds):
       
   399         self.reset()
       
   400         sys.settrace(self.trace_dispatch)
       
   401         res = None
       
   402         try:
       
   403             try:
       
   404                 res = func(*args, **kwds)
       
   405             except BdbQuit:
       
   406                 pass
       
   407         finally:
       
   408             self.quitting = 1
       
   409             sys.settrace(None)
       
   410         return res
       
   411 
       
   412 
       
   413 def set_trace():
       
   414     Bdb().set_trace()
       
   415 
       
   416 
       
   417 class Breakpoint:
       
   418 
       
   419     """Breakpoint class
       
   420 
       
   421     Implements temporary breakpoints, ignore counts, disabling and
       
   422     (re)-enabling, and conditionals.
       
   423 
       
   424     Breakpoints are indexed by number through bpbynumber and by
       
   425     the file,line tuple using bplist.  The former points to a
       
   426     single instance of class Breakpoint.  The latter points to a
       
   427     list of such instances since there may be more than one
       
   428     breakpoint per line.
       
   429 
       
   430     """
       
   431 
       
   432     # XXX Keeping state in the class is a mistake -- this means
       
   433     # you cannot have more than one active Bdb instance.
       
   434 
       
   435     next = 1        # Next bp to be assigned
       
   436     bplist = {}     # indexed by (file, lineno) tuple
       
   437     bpbynumber = [None] # Each entry is None or an instance of Bpt
       
   438                 # index 0 is unused, except for marking an
       
   439                 # effective break .... see effective()
       
   440 
       
   441     def __init__(self, file, line, temporary=0, cond=None, funcname=None):
       
   442         self.funcname = funcname
       
   443         # Needed if funcname is not None.
       
   444         self.func_first_executable_line = None
       
   445         self.file = file    # This better be in canonical form!
       
   446         self.line = line
       
   447         self.temporary = temporary
       
   448         self.cond = cond
       
   449         self.enabled = 1
       
   450         self.ignore = 0
       
   451         self.hits = 0
       
   452         self.number = Breakpoint.next
       
   453         Breakpoint.next = Breakpoint.next + 1
       
   454         # Build the two lists
       
   455         self.bpbynumber.append(self)
       
   456         if self.bplist.has_key((file, line)):
       
   457             self.bplist[file, line].append(self)
       
   458         else:
       
   459             self.bplist[file, line] = [self]
       
   460 
       
   461 
       
   462     def deleteMe(self):
       
   463         index = (self.file, self.line)
       
   464         self.bpbynumber[self.number] = None   # No longer in list
       
   465         self.bplist[index].remove(self)
       
   466         if not self.bplist[index]:
       
   467             # No more bp for this f:l combo
       
   468             del self.bplist[index]
       
   469 
       
   470     def enable(self):
       
   471         self.enabled = 1
       
   472 
       
   473     def disable(self):
       
   474         self.enabled = 0
       
   475 
       
   476     def bpprint(self, out=None):
       
   477         if out is None:
       
   478             out = sys.stdout
       
   479         if self.temporary:
       
   480             disp = 'del  '
       
   481         else:
       
   482             disp = 'keep '
       
   483         if self.enabled:
       
   484             disp = disp + 'yes  '
       
   485         else:
       
   486             disp = disp + 'no   '
       
   487         print >>out, '%-4dbreakpoint   %s at %s:%d' % (self.number, disp,
       
   488                                                        self.file, self.line)
       
   489         if self.cond:
       
   490             print >>out, '\tstop only if %s' % (self.cond,)
       
   491         if self.ignore:
       
   492             print >>out, '\tignore next %d hits' % (self.ignore)
       
   493         if (self.hits):
       
   494             if (self.hits > 1): ss = 's'
       
   495             else: ss = ''
       
   496             print >>out, ('\tbreakpoint already hit %d time%s' %
       
   497                           (self.hits, ss))
       
   498 
       
   499 # -----------end of Breakpoint class----------
       
   500 
       
   501 def checkfuncname(b, frame):
       
   502     """Check whether we should break here because of `b.funcname`."""
       
   503     if not b.funcname:
       
   504         # Breakpoint was set via line number.
       
   505         if b.line != frame.f_lineno:
       
   506             # Breakpoint was set at a line with a def statement and the function
       
   507             # defined is called: don't break.
       
   508             return False
       
   509         return True
       
   510 
       
   511     # Breakpoint set via function name.
       
   512 
       
   513     if frame.f_code.co_name != b.funcname:
       
   514         # It's not a function call, but rather execution of def statement.
       
   515         return False
       
   516 
       
   517     # We are in the right frame.
       
   518     if not b.func_first_executable_line:
       
   519         # The function is entered for the 1st time.
       
   520         b.func_first_executable_line = frame.f_lineno
       
   521 
       
   522     if  b.func_first_executable_line != frame.f_lineno:
       
   523         # But we are not at the first line number: don't break.
       
   524         return False
       
   525     return True
       
   526 
       
   527 # Determines if there is an effective (active) breakpoint at this
       
   528 # line of code.  Returns breakpoint number or 0 if none
       
   529 def effective(file, line, frame):
       
   530     """Determine which breakpoint for this file:line is to be acted upon.
       
   531 
       
   532     Called only if we know there is a bpt at this
       
   533     location.  Returns breakpoint that was triggered and a flag
       
   534     that indicates if it is ok to delete a temporary bp.
       
   535 
       
   536     """
       
   537     possibles = Breakpoint.bplist[file,line]
       
   538     for i in range(0, len(possibles)):
       
   539         b = possibles[i]
       
   540         if b.enabled == 0:
       
   541             continue
       
   542         if not checkfuncname(b, frame):
       
   543             continue
       
   544         # Count every hit when bp is enabled
       
   545         b.hits = b.hits + 1
       
   546         if not b.cond:
       
   547             # If unconditional, and ignoring,
       
   548             # go on to next, else break
       
   549             if b.ignore > 0:
       
   550                 b.ignore = b.ignore -1
       
   551                 continue
       
   552             else:
       
   553                 # breakpoint and marker that's ok
       
   554                 # to delete if temporary
       
   555                 return (b,1)
       
   556         else:
       
   557             # Conditional bp.
       
   558             # Ignore count applies only to those bpt hits where the
       
   559             # condition evaluates to true.
       
   560             try:
       
   561                 val = eval(b.cond, frame.f_globals,
       
   562                        frame.f_locals)
       
   563                 if val:
       
   564                     if b.ignore > 0:
       
   565                         b.ignore = b.ignore -1
       
   566                         # continue
       
   567                     else:
       
   568                         return (b,1)
       
   569                 # else:
       
   570                 #   continue
       
   571             except:
       
   572                 # if eval fails, most conservative
       
   573                 # thing is to stop on breakpoint
       
   574                 # regardless of ignore count.
       
   575                 # Don't delete temporary,
       
   576                 # as another hint to user.
       
   577                 return (b,0)
       
   578     return (None, None)
       
   579 
       
   580 # -------------------- testing --------------------
       
   581 
       
   582 class Tdb(Bdb):
       
   583     def user_call(self, frame, args):
       
   584         name = frame.f_code.co_name
       
   585         if not name: name = '???'
       
   586         print '+++ call', name, args
       
   587     def user_line(self, frame):
       
   588         import linecache
       
   589         name = frame.f_code.co_name
       
   590         if not name: name = '???'
       
   591         fn = self.canonic(frame.f_code.co_filename)
       
   592         line = linecache.getline(fn, frame.f_lineno)
       
   593         print '+++', fn, frame.f_lineno, name, ':', line.strip()
       
   594     def user_return(self, frame, retval):
       
   595         print '+++ return', retval
       
   596     def user_exception(self, frame, exc_stuff):
       
   597         print '+++ exception', exc_stuff
       
   598         self.set_continue()
       
   599 
       
   600 def foo(n):
       
   601     print 'foo(', n, ')'
       
   602     x = bar(n*10)
       
   603     print 'bar returned', x
       
   604 
       
   605 def bar(a):
       
   606     print 'bar(', a, ')'
       
   607     return a/2
       
   608 
       
   609 def test():
       
   610     t = Tdb()
       
   611     t.run('import bdb; bdb.foo(10)')
       
   612 
       
   613 # end