Permalink
Join GitHub today
GitHub is home to over 36 million developers working together to host and review code, manage projects, and build software together.
Sign up#! /usr/bin/env python3 | |
""" | |
The Python Debugger Pdb | |
======================= | |
To use the debugger in its simplest form: | |
>>>import pdb | |
>>>pdb.run('<a statement>') | |
The debugger's prompt is '(Pdb) '. This will stop in the first | |
function call in <a statement>. | |
Alternatively, if a statement terminated with an unhandled exception, | |
you can use pdb's post-mortem facility to inspect the contents of the | |
traceback: | |
>>><a statement> | |
<exception traceback> | |
>>>import pdb | |
>>>pdb.pm() | |
The commands recognized by the debugger are listed in the next | |
section. Most can be abbreviated as indicated; e.g., h(elp) means | |
that 'help' can be typed as 'h' or 'help' (but not as 'he' or 'hel', | |
nor as 'H' or 'Help' or 'HELP'). Optional arguments are enclosed in | |
square brackets. Alternatives in the command syntax are separated | |
by a vertical bar (|). | |
A blank line repeats the previous command literally, except for | |
'list', where it lists the next 11 lines. | |
Commands that the debugger doesn't recognize are assumed to be Python | |
statements and are executed in the context of the program being | |
debugged. Python statements can also be prefixed with an exclamation | |
point ('!'). This is a powerful way to inspect the program being | |
debugged; it is even possible to change variables or call functions. | |
When an exception occurs in such a statement, the exception name is | |
printed but the debugger's state is not changed. | |
The debugger supports aliases, which can save typing. And aliases can | |
have parameters (see the alias help entry) which allows one a certain | |
level of adaptability to the context under examination. | |
Multiple commands may be entered on a single line, separated by the | |
pair ';;'. No intelligence is applied to separating the commands; the | |
input is split at the first ';;', even if it is in the middle of a | |
quoted string. | |
If a file ".pdbrc" exists in your home directory or in the current | |
directory, it is read in and executed as if it had been typed at the | |
debugger prompt. This is particularly useful for aliases. If both | |
files exist, the one in the home directory is read first and aliases | |
defined there can be overridden by the local file. This behavior can be | |
disabled by passing the "readrc=False" argument to the Pdb constructor. | |
Aside from aliases, the debugger is not directly programmable; but it | |
is implemented as a class from which you can derive your own debugger | |
class, which you can make as fancy as you like. | |
Debugger commands | |
================= | |
""" | |
#NOTE: the actual command documentation is collected from docstrings of the | |
# commands and is appended to __doc__ after the class has been defined. | |
import os | |
import re | |
import sys | |
import cmd | |
import bdb | |
import dis | |
import code | |
import glob | |
import pprint | |
import signal | |
import inspect | |
import traceback | |
import linecache | |
classRestart(Exception): | |
"""Causes a debugger to be restarted for the debugged python program.""" | |
pass | |
__all__= ["run","pm","Pdb","runeval","runctx","runcall","set_trace", | |
"post_mortem","help"] | |
deffind_function(funcname,filename): | |
cre= re.compile(r'def\s+%s\s*[(]'% re.escape(funcname)) | |
try: | |
fp=open(filename) | |
exceptOSError: | |
returnNone | |
# consumer of this info expects the first line to be 1 | |
with fp: | |
for lineno, lineinenumerate(fp,start=1): | |
if cre.match(line): | |
return funcname, filename, lineno | |
returnNone | |
defgetsourcelines(obj): | |
lines, lineno= inspect.findsource(obj) | |
if inspect.isframe(obj)and obj.f_globalsis obj.f_locals: | |
# must be a module frame: do not try to cut a block out of it | |
return lines,1 | |
elif inspect.ismodule(obj): | |
return lines,1 | |
return inspect.getblock(lines[lineno:]), lineno+1 | |
deflasti2lineno(code,lasti): | |
linestarts=list(dis.findlinestarts(code)) | |
linestarts.reverse() | |
for i, linenoin linestarts: | |
if lasti>= i: | |
return lineno | |
return0 | |
class_rstr(str): | |
"""String that doesn't quote its repr.""" | |
def__repr__(self): | |
returnself | |
# Interaction prompt line will separate file and call info from code | |
# text using value of line_prefix string. A newline and arrow may | |
# be to your liking. You can set it once pdb is imported using the | |
# command "pdb.line_prefix = '\n% '". | |
# line_prefix = ': ' # Use this to get the old situation back | |
line_prefix='\n->'# Probably a better default | |
classPdb(bdb.Bdb,cmd.Cmd): | |
_previous_sigint_handler=None | |
def__init__(self,completekey='tab',stdin=None,stdout=None,skip=None, | |
nosigint=False,readrc=True): | |
bdb.Bdb.__init__(self,skip=skip) | |
cmd.Cmd.__init__(self, completekey, stdin, stdout) | |
if stdout: | |
self.use_rawinput=0 | |
self.prompt='(Pdb)' | |
self.aliases= {} | |
self.displaying= {} | |
self.mainpyfile='' | |
self._wait_for_mainpyfile=False | |
self.tb_lineno= {} | |
# Try to load readline if it exists | |
try: | |
import readline | |
# remove some common file name delimiters | |
readline.set_completer_delims('\t\n`@#$%^&*()=+[{]}\\|;:\'",<>?') | |
exceptImportError: | |
pass | |
self.allow_kbdint=False | |
self.nosigint= nosigint | |
# Read $HOME/.pdbrc and ./.pdbrc | |
self.rcLines= [] | |
if readrc: | |
if'HOME'in os.environ: | |
envHome= os.environ['HOME'] | |
try: | |
withopen(os.path.join(envHome,".pdbrc"))as rcFile: | |
self.rcLines.extend(rcFile) | |
exceptOSError: | |
pass | |
try: | |
withopen(".pdbrc")as rcFile: | |
self.rcLines.extend(rcFile) | |
exceptOSError: | |
pass | |
self.commands= {}# associates a command list to breakpoint numbers | |
self.commands_doprompt= {}# for each bp num, tells if the prompt | |
# must be disp. after execing the cmd list | |
self.commands_silent= {}# for each bp num, tells if the stack trace | |
# must be disp. after execing the cmd list | |
self.commands_defining=False# True while in the process of defining | |
# a command list | |
self.commands_bnum=None# The breakpoint number for which we are | |
# defining a list | |
defsigint_handler(self,signum,frame): | |
ifself.allow_kbdint: | |
raiseKeyboardInterrupt | |
self.message("\nProgram interrupted. (Use 'cont' to resume).") | |
self.set_step() | |
self.set_trace(frame) | |
defreset(self): | |
bdb.Bdb.reset(self) | |
self.forget() | |
defforget(self): | |
self.lineno=None | |
self.stack= [] | |
self.curindex=0 | |
self.curframe=None | |
self.tb_lineno.clear() | |
defsetup(self,f,tb): | |
self.forget() | |
self.stack,self.curindex=self.get_stack(f, tb) | |
while tb: | |
# when setting up post-mortem debugging with a traceback, save all | |
# the original line numbers to be displayed along the current line | |
# numbers (which can be different, e.g. due to finally clauses) | |
lineno= lasti2lineno(tb.tb_frame.f_code, tb.tb_lasti) | |
self.tb_lineno[tb.tb_frame]= lineno | |
tb= tb.tb_next | |
self.curframe=self.stack[self.curindex][0] | |
# The f_locals dictionary is updated from the actual frame | |
# locals whenever the .f_locals accessor is called, so we | |
# cache it here to ensure that modifications are not overwritten. | |
self.curframe_locals=self.curframe.f_locals | |
returnself.execRcLines() | |
# Can be executed earlier than 'setup' if desired | |
defexecRcLines(self): | |
ifnotself.rcLines: | |
return | |
# local copy because of recursion | |
rcLines=self.rcLines | |
rcLines.reverse() | |
# execute every line only once | |
self.rcLines= [] | |
while rcLines: | |
line= rcLines.pop().strip() | |
if lineand line[0]!='#': | |
ifself.onecmd(line): | |
# if onecmd returns True, the command wants to exit | |
# from the interaction, save leftover rc lines | |
# to execute before next interaction | |
self.rcLines+=reversed(rcLines) | |
returnTrue | |
# Override Bdb methods | |
defuser_call(self,frame,argument_list): | |
"""This method is called when there is the remote possibility | |
that we ever need to stop in this function.""" | |
ifself._wait_for_mainpyfile: | |
return | |
ifself.stop_here(frame): | |
self.message('--Call--') | |
self.interaction(frame,None) | |
defuser_line(self,frame): | |
"""This function is called when we stop or break at this line.""" | |
ifself._wait_for_mainpyfile: | |
if (self.mainpyfile!=self.canonic(frame.f_code.co_filename) | |
or frame.f_lineno<=0): | |
return | |
self._wait_for_mainpyfile=False | |
ifself.bp_commands(frame): | |
self.interaction(frame,None) | |
defbp_commands(self,frame): | |
"""Call every command that was set for the current active breakpoint | |
(if there is one). | |
Returns True if the normal interaction function must be called, | |
False otherwise.""" | |
# self.currentbp is set in bdb in Bdb.break_here if a breakpoint was hit | |
ifgetattr(self,"currentbp",False)and \ | |
self.currentbpinself.commands: | |
currentbp=self.currentbp | |
self.currentbp=0 | |
lastcmd_back=self.lastcmd | |
self.setup(frame,None) | |
for lineinself.commands[currentbp]: | |
self.onecmd(line) | |
self.lastcmd= lastcmd_back | |
ifnotself.commands_silent[currentbp]: | |
self.print_stack_entry(self.stack[self.curindex]) | |
ifself.commands_doprompt[currentbp]: | |
self._cmdloop() | |
self.forget() | |
return | |
return1 | |
defuser_return(self,frame,return_value): | |
"""This function is called when a return trap is set here.""" | |
ifself._wait_for_mainpyfile: | |
return | |
frame.f_locals['__return__']= return_value | |
self.message('--Return--') | |
self.interaction(frame,None) | |
defuser_exception(self,frame,exc_info): | |
"""This function is called if an exception occurs, | |
but only if we are to stop at or just below this level.""" | |
ifself._wait_for_mainpyfile: | |
return | |
exc_type, exc_value, exc_traceback= exc_info | |
frame.f_locals['__exception__']= exc_type, exc_value | |
# An 'Internal StopIteration' exception is an exception debug event | |
# issued by the interpreter when handling a subgenerator run with | |
# 'yield from' or a generator controlled by a for loop. No exception has | |
# actually occurred in this case. The debugger uses this debug event to | |
# stop when the debuggee is returning from such generators. | |
prefix='Internal'if (not exc_traceback | |
and exc_typeisStopIteration)else'' | |
self.message('%s%s'% (prefix, | |
traceback.format_exception_only(exc_type, exc_value)[-1].strip())) | |
self.interaction(frame, exc_traceback) | |
# General interaction function | |
def_cmdloop(self): | |
whileTrue: | |
try: | |
# keyboard interrupts allow for an easy way to cancel | |
# the current command, so allow them during interactive input | |
self.allow_kbdint=True | |
self.cmdloop() | |
self.allow_kbdint=False | |
break | |
exceptKeyboardInterrupt: | |
self.message('--KeyboardInterrupt--') | |
# Called before loop, handles display expressions | |
defpreloop(self): | |
displaying=self.displaying.get(self.curframe) | |
if displaying: | |
for expr, oldvaluein displaying.items(): | |
newvalue=self._getval_except(expr) | |
# check for identity first; this prevents custom __eq__ to | |
# be called at every loop, and also prevents instances whose | |
# fields are changed to be displayed | |
if newvalueisnot oldvalueand newvalue!= oldvalue: | |
displaying[expr]= newvalue | |
self.message('display%s:%r [old:%r]'% | |
(expr, newvalue, oldvalue)) | |
definteraction(self,frame,traceback): | |
# Restore the previous signal handler at the Pdb prompt. | |
if Pdb._previous_sigint_handler: | |
signal.signal(signal.SIGINT, Pdb._previous_sigint_handler) | |
Pdb._previous_sigint_handler=None | |
ifself.setup(frame, traceback): | |
# no interaction desired at this time (happens if .pdbrc contains | |
# a command like "continue") | |
self.forget() | |
return | |
self.print_stack_entry(self.stack[self.curindex]) | |
self._cmdloop() | |
self.forget() | |
defdisplayhook(self,obj): | |
"""Custom displayhook for the exec in default(), which prevents | |
assignment of the _ variable in the builtins. | |
""" | |
# reproduce the behavior of the standard displayhook, not printing None | |
if objisnotNone: | |
self.message(repr(obj)) | |
defdefault(self,line): | |
if line[:1]=='!': line= line[1:] | |
locals=self.curframe_locals | |
globals=self.curframe.f_globals | |
try: | |
code=compile(line+'\n','<stdin>','single') | |
save_stdout= sys.stdout | |
save_stdin= sys.stdin | |
save_displayhook= sys.displayhook | |
try: | |
sys.stdin=self.stdin | |
sys.stdout=self.stdout | |
sys.displayhook=self.displayhook | |
exec(code,globals,locals) | |
finally: | |
sys.stdout= save_stdout | |
sys.stdin= save_stdin | |
sys.displayhook= save_displayhook | |
except: | |
exc_info= sys.exc_info()[:2] | |
self.error(traceback.format_exception_only(*exc_info)[-1].strip()) | |
defprecmd(self,line): | |
"""Handle alias expansion and ';;' separator.""" | |
ifnot line.strip(): | |
return line | |
args= line.split() | |
while args[0]inself.aliases: | |
line=self.aliases[args[0]] | |
ii=1 | |
for tmpArgin args[1:]: | |
line= line.replace("%"+str(ii), | |
tmpArg) | |
ii+=1 | |
line= line.replace("%*",''.join(args[1:])) | |
args= line.split() | |
# split into ';;' separated commands | |
# unless it's an alias command | |
if args[0]!='alias': | |
marker= line.find(';;') | |
if marker>=0: | |
# queue up everything after marker | |
next= line[marker+2:].lstrip() | |
self.cmdqueue.append(next) | |
line= line[:marker].rstrip() | |
return line | |
defonecmd(self,line): | |
"""Interpret the argument as though it had been typed in response | |
to the prompt. | |
Checks whether this line is typed at the normal prompt or in | |
a breakpoint command list definition. | |
""" | |
ifnotself.commands_defining: | |
return cmd.Cmd.onecmd(self, line) | |
else: | |
returnself.handle_command_def(line) | |
defhandle_command_def(self,line): | |
"""Handles one command line during command list definition.""" | |
cmd, arg, line=self.parseline(line) | |
ifnot cmd: | |
return | |
if cmd=='silent': | |
self.commands_silent[self.commands_bnum]=True | |
return# continue to handle other cmd def in the cmd list | |
elif cmd=='end': | |
self.cmdqueue= [] | |
return1# end of cmd list | |
cmdlist=self.commands[self.commands_bnum] | |
if arg: | |
cmdlist.append(cmd+''+arg) | |
else: | |
cmdlist.append(cmd) | |
# Determine if we must stop | |
try: | |
func=getattr(self,'do_'+ cmd) | |
exceptAttributeError: | |
func=self.default | |
# one of the resuming commands | |
if func.__name__inself.commands_resuming: | |
self.commands_doprompt[self.commands_bnum]=False | |
self.cmdqueue= [] | |
return1 | |
return | |
# interface abstraction functions | |
defmessage(self,msg): | |
print(msg,file=self.stdout) | |
deferror(self,msg): | |
print('***', msg,file=self.stdout) | |
# Generic completion functions. Individual complete_foo methods can be | |
# assigned below to one of these functions. | |
def_complete_location(self,text,line,begidx,endidx): | |
# Complete a file/module/function location for break/tbreak/clear. | |
if line.strip().endswith((':',',')): | |
# Here comes a line number or a condition which we can't complete. | |
return [] | |
# First, try to find matching functions (i.e. expressions). | |
try: | |
ret=self._complete_expression(text, line, begidx, endidx) | |
exceptException: | |
ret= [] | |
# Then, try to complete file names as well. | |
globs= glob.glob(text+'*') | |
for fnin globs: | |
if os.path.isdir(fn): | |
ret.append(fn+'/') | |
elif os.path.isfile(fn)and fn.lower().endswith(('.py','.pyw')): | |
ret.append(fn+':') | |
return ret | |
def_complete_bpnumber(self,text,line,begidx,endidx): | |
# Complete a breakpoint number. (This would be more helpful if we could | |
# display additional info along with the completions, such as file/line | |
# of the breakpoint.) | |
return [str(i)for i, bpinenumerate(bdb.Breakpoint.bpbynumber) | |
if bpisnotNoneandstr(i).startswith(text)] | |
def_complete_expression(self,text,line,begidx,endidx): | |
# Complete an arbitrary expression. | |
ifnotself.curframe: | |
return [] | |
# Collect globals and locals. It is usually not really sensible to also | |
# complete builtins, and they clutter the namespace quite heavily, so we | |
# leave them out. | |
ns= {**self.curframe.f_globals,**self.curframe_locals} | |
if'.'in text: | |
# Walk an attribute chain up to the last part, similar to what | |
# rlcompleter does. This will bail if any of the parts are not | |
# simple attribute access, which is what we want. | |
dotted= text.split('.') | |
try: | |
obj= ns[dotted[0]] | |
for partin dotted[1:-1]: | |
obj=getattr(obj, part) | |
except (KeyError,AttributeError): | |
return [] | |
prefix='.'.join(dotted[:-1])+'.' | |
return [prefix+ nfor nindir(obj)if n.startswith(dotted[-1])] | |
else: | |
# Complete a simple name. | |
return [nfor nin ns.keys()if n.startswith(text)] | |
# Command definitions, called by cmdloop() | |
# The argument is the remaining string on the command line | |
# Return true to exit from the command loop | |
defdo_commands(self,arg): | |
"""commands [bpnumber] | |
(com) ... | |
(com) end | |
(Pdb) | |
Specify a list of commands for breakpoint number bpnumber. | |
The commands themselves are entered on the following lines. | |
Type a line containing just 'end' to terminate the commands. | |
The commands are executed when the breakpoint is hit. | |
To remove all commands from a breakpoint, type commands and | |
follow it immediately with end; that is, give no commands. | |
With no bpnumber argument, commands refers to the last | |
breakpoint set. | |
You can use breakpoint commands to start your program up | |
again. Simply use the continue command, or step, or any other | |
command that resumes execution. | |
Specifying any command resuming execution (currently continue, | |
step, next, return, jump, quit and their abbreviations) | |
terminates the command list (as if that command was | |
immediately followed by end). This is because any time you | |
resume execution (even with a simple next or step), you may | |
encounter another breakpoint -- which could have its own | |
command list, leading to ambiguities about which list to | |
execute. | |
If you use the 'silent' command in the command list, the usual | |
message about stopping at a breakpoint is not printed. This | |
may be desirable for breakpoints that are to print a specific | |
message and then continue. If none of the other commands | |
print anything, you will see no sign that the breakpoint was | |
reached. | |
""" | |
ifnot arg: | |
bnum=len(bdb.Breakpoint.bpbynumber)-1 | |
else: | |
try: | |
bnum=int(arg) | |
except: | |
self.error("Usage: commands [bnum]\n ...\n end") | |
return | |
self.commands_bnum= bnum | |
# Save old definitions for the case of a keyboard interrupt. | |
if bnuminself.commands: | |
old_command_defs= (self.commands[bnum], | |
self.commands_doprompt[bnum], | |
self.commands_silent[bnum]) | |
else: | |
old_command_defs=None | |
self.commands[bnum]= [] | |
self.commands_doprompt[bnum]=True | |
self.commands_silent[bnum]=False | |
prompt_back=self.prompt | |
self.prompt='(com)' | |
self.commands_defining=True | |
try: | |
self.cmdloop() | |
exceptKeyboardInterrupt: | |
# Restore old definitions. | |
if old_command_defs: | |
self.commands[bnum]= old_command_defs[0] | |
self.commands_doprompt[bnum]= old_command_defs[1] | |
self.commands_silent[bnum]= old_command_defs[2] | |
else: | |
delself.commands[bnum] | |
delself.commands_doprompt[bnum] | |
delself.commands_silent[bnum] | |
self.error('command definition aborted, old commands restored') | |
finally: | |
self.commands_defining=False | |
self.prompt= prompt_back | |
complete_commands= _complete_bpnumber | |
defdo_break(self,arg,temporary=0): | |
"""b(reak) [ ([filename:]lineno | function) [, condition] ] | |
Without argument, list all breaks. | |
With a line number argument, set a break at this line in the | |
current file. With a function name, set a break at the first | |
executable line of that function. If a second argument is | |
present, it is a string specifying an expression which must | |
evaluate to true before the breakpoint is honored. | |
The line number may be prefixed with a filename and a colon, | |
to specify a breakpoint in another file (probably one that | |
hasn't been loaded yet). The file is searched for on | |
sys.path; the .py suffix may be omitted. | |
""" | |
ifnot arg: | |
ifself.breaks:# There's at least one | |
self.message("Num Type Disp Enb Where") | |
for bpin bdb.Breakpoint.bpbynumber: | |
if bp: | |
self.message(bp.bpformat()) | |
return | |
# parse arguments; comma has lowest precedence | |
# and cannot occur in filename | |
filename=None | |
lineno=None | |
cond=None | |
comma= arg.find(',') | |
if comma>0: | |
# parse stuff after comma: "condition" | |
cond= arg[comma+1:].lstrip() | |
arg= arg[:comma].rstrip() | |
# parse stuff before comma: [filename:]lineno | function | |
colon= arg.rfind(':') | |
funcname=None | |
if colon>=0: | |
filename= arg[:colon].rstrip() | |
f=self.lookupmodule(filename) | |
ifnot f: | |
self.error('%r not found from sys.path'% filename) | |
return | |
else: | |
filename= f | |
arg= arg[colon+1:].lstrip() | |
try: | |
lineno=int(arg) | |
exceptValueError: | |
self.error('Bad lineno:%s'% arg) | |
return | |
else: | |
# no colon; can be lineno or function | |
try: | |
lineno=int(arg) | |
exceptValueError: | |
try: | |
func=eval(arg, | |
self.curframe.f_globals, | |
self.curframe_locals) | |
except: | |
func= arg | |
try: | |
ifhasattr(func,'__func__'): | |
func= func.__func__ | |
code= func.__code__ | |
#use co_name to identify the bkpt (function names | |
#could be aliased, but co_name is invariant) | |
funcname= code.co_name | |
lineno= code.co_firstlineno | |
filename= code.co_filename | |
except: | |
# last thing to try | |
(ok, filename, ln)=self.lineinfo(arg) | |
ifnot ok: | |
self.error('The specified object%r is not a function' | |
'or was not found along sys.path.'% arg) | |
return | |
funcname= ok# ok contains a function name | |
lineno=int(ln) | |
ifnot filename: | |
filename=self.defaultFile() | |
# Check for reasonable breakpoint | |
line=self.checkline(filename, lineno) | |
if line: | |
# now set the break point | |
err=self.set_break(filename, line, temporary, cond, funcname) | |
if err: | |
self.error(err) | |
else: | |
bp=self.get_breaks(filename, line)[-1] | |
self.message("Breakpoint%d at%s:%d"% | |
(bp.number, bp.file, bp.line)) | |
# To be overridden in derived debuggers | |
defdefaultFile(self): | |
"""Produce a reasonable default.""" | |
filename=self.curframe.f_code.co_filename | |
if filename=='<string>'andself.mainpyfile: | |
filename=self.mainpyfile | |
return filename | |
do_b= do_break | |
complete_break= _complete_location | |
complete_b= _complete_location | |
defdo_tbreak(self,arg): | |
"""tbreak [ ([filename:]lineno | function) [, condition] ] | |
Same arguments as break, but sets a temporary breakpoint: it | |
is automatically deleted when first hit. | |
""" | |
self.do_break(arg,1) | |
complete_tbreak= _complete_location | |
deflineinfo(self,identifier): | |
failed= (None,None,None) | |
# Input is identifier, may be in single quotes | |
idstring= identifier.split("'") | |
iflen(idstring)==1: | |
# not in single quotes | |
id= idstring[0].strip() | |
eliflen(idstring)==3: | |
# quoted | |
id= idstring[1].strip() | |
else: | |
return failed | |
ifid=='':return failed | |
parts=id.split('.') | |
# Protection for derived debuggers | |
if parts[0]=='self': | |
del parts[0] | |
iflen(parts)==0: | |
return failed | |
# Best first guess at file to look at | |
fname=self.defaultFile() | |
iflen(parts)==1: | |
item= parts[0] | |
else: | |
# More than one part. | |
# First is module, second is method/class | |
f=self.lookupmodule(parts[0]) | |
if f: | |
fname= f | |
item= parts[1] | |
answer= find_function(item, fname) | |
return answeror failed | |
defcheckline(self,filename,lineno): | |
"""Check whether specified line seems to be executable. | |
Return `lineno` if it is, 0 if not (e.g. a docstring, comment, blank | |
line or EOF). Warning: testing is not comprehensive. | |
""" | |
# this method should be callable before starting debugging, so default | |
# to "no globals" if there is no current frame | |
globs=self.curframe.f_globalsifhasattr(self,'curframe')elseNone | |
line= linecache.getline(filename, lineno, globs) | |
ifnot line: | |
self.message('End of file') | |
return0 | |
line= line.strip() | |
# Don't allow setting breakpoint at a blank line | |
if (not lineor (line[0]=='#')or | |
(line[:3]=='"""')or line[:3]=="'''"): | |
self.error('Blank or comment') | |
return0 | |
return lineno | |
defdo_enable(self,arg): | |
"""enable bpnumber [bpnumber ...] | |
Enables the breakpoints given as a space separated list of | |
breakpoint numbers. | |
""" | |
args= arg.split() | |
for iin args: | |
try: | |
bp=self.get_bpbynumber(i) | |
exceptValueErroras err: | |
self.error(err) | |
else: | |
bp.enable() | |
self.message('Enabled%s'% bp) | |
complete_enable= _complete_bpnumber | |
defdo_disable(self,arg): | |
"""disable bpnumber [bpnumber ...] | |
Disables the breakpoints given as a space separated list of | |
breakpoint numbers. Disabling a breakpoint means it cannot | |
cause the program to stop execution, but unlike clearing a | |
breakpoint, it remains in the list of breakpoints and can be | |
(re-)enabled. | |
""" | |
args= arg.split() | |
for iin args: | |
try: | |
bp=self.get_bpbynumber(i) | |
exceptValueErroras err: | |
self.error(err) | |
else: | |
bp.disable() | |
self.message('Disabled%s'% bp) | |
complete_disable= _complete_bpnumber | |
defdo_condition(self,arg): | |
"""condition bpnumber [condition] | |
Set a new condition for the breakpoint, an expression which | |
must evaluate to true before the breakpoint is honored. If | |
condition is absent, any existing condition is removed; i.e., | |
the breakpoint is made unconditional. | |
""" | |
args= arg.split('',1) | |
try: | |
cond= args[1] | |
exceptIndexError: | |
cond=None | |
try: | |
bp=self.get_bpbynumber(args[0].strip()) | |
exceptIndexError: | |
self.error('Breakpoint number expected') | |
exceptValueErroras err: | |
self.error(err) | |
else: | |
bp.cond= cond | |
ifnot cond: | |
self.message('Breakpoint%d is now unconditional.'% bp.number) | |
else: | |
self.message('New condition set for breakpoint%d.'% bp.number) | |
complete_condition= _complete_bpnumber | |
defdo_ignore(self,arg): | |
"""ignore bpnumber [count] | |
Set the ignore count for the given breakpoint number. If | |
count is omitted, the ignore count is set to 0. A breakpoint | |
becomes active when the ignore count is zero. When non-zero, | |
the count is decremented each time the breakpoint is reached | |
and the breakpoint is not disabled and any associated | |
condition evaluates to true. | |
""" | |
args= arg.split() | |
try: | |
count=int(args[1].strip()) | |
except: | |
count=0 | |
try: | |
bp=self.get_bpbynumber(args[0].strip()) | |
exceptIndexError: | |
self.error('Breakpoint number expected') | |
exceptValueErroras err: | |
self.error(err) | |
else: | |
bp.ignore= count | |
if count>0: | |
if count>1: | |
countstr='%d crossings'% count | |
else: | |
countstr='1 crossing' | |
self.message('Will ignore next%s of breakpoint%d.'% | |
(countstr, bp.number)) | |
else: | |
self.message('Will stop next time breakpoint%d is reached.' | |
% bp.number) | |
complete_ignore= _complete_bpnumber | |
defdo_clear(self,arg): | |
"""cl(ear) filename:lineno\ncl(ear) [bpnumber [bpnumber...]] | |
With a space separated list of breakpoint numbers, clear | |
those breakpoints. Without argument, clear all breaks (but | |
first ask confirmation). With a filename:lineno argument, | |
clear all breaks at that line in that file. | |
""" | |
ifnot arg: | |
try: | |
reply=input('Clear all breaks?') | |
exceptEOFError: | |
reply='no' | |
reply= reply.strip().lower() | |
if replyin ('y','yes'): | |
bplist= [bpfor bpin bdb.Breakpoint.bpbynumberif bp] | |
self.clear_all_breaks() | |
for bpin bplist: | |
self.message('Deleted%s'% bp) | |
return | |
if':'in arg: | |
# Make sure it works for "clear C:\foo\bar.py:12" | |
i= arg.rfind(':') | |
filename= arg[:i] | |
arg= arg[i+1:] | |
try: | |
lineno=int(arg) | |
exceptValueError: | |
err="Invalid line number (%s)"% arg | |
else: | |
bplist=self.get_breaks(filename, lineno) | |
err=self.clear_break(filename, lineno) | |
if err: | |
self.error(err) | |
else: | |
for bpin bplist: | |
self.message('Deleted%s'% bp) | |
return | |
numberlist= arg.split() | |
for iin numberlist: | |
try: | |
bp=self.get_bpbynumber(i) | |
exceptValueErroras err: | |
self.error(err) | |
else: | |
self.clear_bpbynumber(i) | |
self.message('Deleted%s'% bp) | |
do_cl= do_clear# 'c' is already an abbreviation for 'continue' | |
complete_clear= _complete_location | |
complete_cl= _complete_location | |
defdo_where(self,arg): | |
"""w(here) | |
Print a stack trace, with the most recent frame at the bottom. | |
An arrow indicates the "current frame", which determines the | |
context of most commands. 'bt' is an alias for this command. | |
""" | |
self.print_stack_trace() | |
do_w= do_where | |
do_bt= do_where | |
def_select_frame(self,number): | |
assert0<= number<len(self.stack) | |
self.curindex= number | |
self.curframe=self.stack[self.curindex][0] | |
self.curframe_locals=self.curframe.f_locals | |
self.print_stack_entry(self.stack[self.curindex]) | |
self.lineno=None | |
defdo_up(self,arg): | |
"""u(p) [count] | |
Move the current frame count (default one) levels up in the | |
stack trace (to an older frame). | |
""" | |
ifself.curindex==0: | |
self.error('Oldest frame') | |
return | |
try: | |
count=int(argor1) | |
exceptValueError: | |
self.error('Invalid frame count (%s)'% arg) | |
return | |
if count<0: | |
newframe=0 | |
else: | |
newframe=max(0,self.curindex- count) | |
self._select_frame(newframe) | |
do_u= do_up | |
defdo_down(self,arg): | |
"""d(own) [count] | |
Move the current frame count (default one) levels down in the | |
stack trace (to a newer frame). | |
""" | |
ifself.curindex+1==len(self.stack): | |
self.error('Newest frame') | |
return | |
try: | |
count=int(argor1) | |
exceptValueError: | |
self.error('Invalid frame count (%s)'% arg) | |
return | |
if count<0: | |
newframe=len(self.stack)-1 | |
else: | |
newframe=min(len(self.stack)-1,self.curindex+ count) | |
self._select_frame(newframe) | |
do_d= do_down | |
defdo_until(self,arg): | |
"""unt(il) [lineno] | |
Without argument, continue execution until the line with a | |
number greater than the current one is reached. With a line | |
number, continue execution until a line with a number greater | |
or equal to that is reached. In both cases, also stop when | |
the current frame returns. | |
""" | |
if arg: | |
try: | |
lineno=int(arg) | |
exceptValueError: | |
self.error('Error in argument:%r'% arg) | |
return | |
if lineno<=self.curframe.f_lineno: | |
self.error('"until" line number is smaller than current' | |
'line number') | |
return | |
else: | |
lineno=None | |
self.set_until(self.curframe, lineno) | |
return1 | |
do_unt= do_until | |
defdo_step(self,arg): | |
"""s(tep) | |
Execute the current line, stop at the first possible occasion | |
(either in a function that is called or in the current | |
function). | |
""" | |
self.set_step() | |
return1 | |
do_s= do_step | |
defdo_next(self,arg): | |
"""n(ext) | |
Continue execution until the next line in the current function | |
is reached or it returns. | |
""" | |
self.set_next(self.curframe) | |
return1 | |
do_n= do_next | |
defdo_run(self,arg): | |
"""run [args...] | |
Restart the debugged python program. If a string is supplied | |
it is split with "shlex", and the result is used as the new | |
sys.argv. History, breakpoints, actions and debugger options | |
are preserved. "restart" is an alias for "run". | |
""" | |
if arg: | |
import shlex | |
argv0= sys.argv[0:1] | |
sys.argv= shlex.split(arg) | |
sys.argv[:0]= argv0 | |
# this is caught in the main debugger loop | |
raise Restart | |
do_restart= do_run | |
defdo_return(self,arg): | |
"""r(eturn) | |
Continue execution until the current function returns. | |
""" | |
self.set_return(self.curframe) | |
return1 | |
do_r= do_return | |
defdo_continue(self,arg): | |
"""c(ont(inue)) | |
Continue execution, only stop when a breakpoint is encountered. | |
""" | |
ifnotself.nosigint: | |
try: | |
Pdb._previous_sigint_handler= \ | |
signal.signal(signal.SIGINT,self.sigint_handler) | |
exceptValueError: | |
# ValueError happens when do_continue() is invoked from | |
# a non-main thread in which case we just continue without | |
# SIGINT set. Would printing a message here (once) make | |
# sense? | |
pass | |
self.set_continue() | |
return1 | |
do_c= do_cont= do_continue | |
defdo_jump(self,arg): | |
"""j(ump) lineno | |
Set the next line that will be executed. Only available in | |
the bottom-most frame. This lets you jump back and execute | |
code again, or jump forward to skip code that you don't want | |
to run. | |
It should be noted that not all jumps are allowed -- for | |
instance it is not possible to jump into the middle of a | |
for loop or out of a finally clause. | |
""" | |
ifself.curindex+1!=len(self.stack): | |
self.error('You can only jump within the bottom frame') | |
return | |
try: | |
arg=int(arg) | |
exceptValueError: | |
self.error("The 'jump' command requires a line number") | |
else: | |
try: | |
# Do the jump, fix up our copy of the stack, and display the | |
# new position | |
self.curframe.f_lineno= arg | |
self.stack[self.curindex]=self.stack[self.curindex][0], arg | |
self.print_stack_entry(self.stack[self.curindex]) | |
exceptValueErroras e: | |
self.error('Jump failed:%s'% e) | |
do_j= do_jump | |
defdo_debug(self,arg): | |
"""debug code | |
Enter a recursive debugger that steps through the code | |
argument (which is an arbitrary expression or statement to be | |
executed in the current environment). | |
""" | |
sys.settrace(None) | |
globals=self.curframe.f_globals | |
locals=self.curframe_locals | |
p= Pdb(self.completekey,self.stdin,self.stdout) | |
p.prompt="(%s)"%self.prompt.strip() | |
self.message("ENTERING RECURSIVE DEBUGGER") | |
try: | |
sys.call_tracing(p.run, (arg,globals,locals)) | |
exceptException: | |
exc_info= sys.exc_info()[:2] | |
self.error(traceback.format_exception_only(*exc_info)[-1].strip()) | |
self.message("LEAVING RECURSIVE DEBUGGER") | |
sys.settrace(self.trace_dispatch) | |
self.lastcmd= p.lastcmd | |
complete_debug= _complete_expression | |
defdo_quit(self,arg): | |
"""q(uit)\nexit | |
Quit from the debugger. The program being executed is aborted. | |
""" | |
self._user_requested_quit=True | |
self.set_quit() | |
return1 | |
do_q= do_quit | |
do_exit= do_quit | |
defdo_EOF(self,arg): | |
"""EOF | |
Handles the receipt of EOF as a command. | |
""" | |
self.message('') | |
self._user_requested_quit=True | |
self.set_quit() | |
return1 | |
defdo_args(self,arg): | |
"""a(rgs) | |
Print the argument list of the current function. | |
""" | |
co=self.curframe.f_code | |
dict=self.curframe_locals | |
n= co.co_argcount+ co.co_posonlyargcount+ co.co_kwonlyargcount | |
if co.co_flags& inspect.CO_VARARGS: n= n+1 | |
if co.co_flags& inspect.CO_VARKEYWORDS: n= n+1 | |
for iinrange(n): | |
name= co.co_varnames[i] | |
if nameindict: | |
self.message('%s =%r'% (name, dict[name])) | |
else: | |
self.message('%s = *** undefined ***'% (name,)) | |
do_a= do_args | |
defdo_retval(self,arg): | |
"""retval | |
Print the return value for the last return of a function. | |
""" | |
if'__return__'inself.curframe_locals: | |
self.message(repr(self.curframe_locals['__return__'])) | |
else: | |
self.error('Not yet returned!') | |
do_rv= do_retval | |
def_getval(self,arg): | |
try: | |
returneval(arg,self.curframe.f_globals,self.curframe_locals) | |
except: | |
exc_info= sys.exc_info()[:2] | |
self.error(traceback.format_exception_only(*exc_info)[-1].strip()) | |
raise | |
def_getval_except(self,arg,frame=None): | |
try: | |
if frameisNone: | |
returneval(arg,self.curframe.f_globals,self.curframe_locals) | |
else: | |
returneval(arg, frame.f_globals, frame.f_locals) | |
except: | |
exc_info= sys.exc_info()[:2] | |
err= traceback.format_exception_only(*exc_info)[-1].strip() | |
return _rstr('** raised%s **'% err) | |
defdo_p(self,arg): | |
"""p expression | |
Print the value of the expression. | |
""" | |
try: | |
self.message(repr(self._getval(arg))) | |
except: | |
pass | |
defdo_pp(self,arg): | |
"""pp expression | |
Pretty-print the value of the expression. | |
""" | |
try: | |
self.message(pprint.pformat(self._getval(arg))) | |
except: | |
pass | |
complete_print= _complete_expression | |
complete_p= _complete_expression | |
complete_pp= _complete_expression | |
defdo_list(self,arg): | |
"""l(ist) [first [,last] | .] | |
List source code for the current file. Without arguments, | |
list 11 lines around the current line or continue the previous | |
listing. With . as argument, list 11 lines around the current | |
line. With one argument, list 11 lines starting at that line. | |
With two arguments, list the given range; if the second | |
argument is less than the first, it is a count. | |
The current line in the current frame is indicated by "->". | |
If an exception is being debugged, the line where the | |
exception was originally raised or propagated is indicated by | |
">>", if it differs from the current line. | |
""" | |
self.lastcmd='list' | |
last=None | |
if argand arg!='.': | |
try: | |
if','in arg: | |
first, last= arg.split(',') | |
first=int(first.strip()) | |
last=int(last.strip()) | |
if last< first: | |
# assume it's a count | |
last= first+ last | |
else: | |
first=int(arg.strip()) | |
first=max(1, first-5) | |
exceptValueError: | |
self.error('Error in argument:%r'% arg) | |
return | |
elifself.linenoisNoneor arg=='.': | |
first=max(1,self.curframe.f_lineno-5) | |
else: | |
first=self.lineno+1 | |
if lastisNone: | |
last= first+10 | |
filename=self.curframe.f_code.co_filename | |
breaklist=self.get_file_breaks(filename) | |
try: | |
lines= linecache.getlines(filename,self.curframe.f_globals) | |
self._print_lines(lines[first-1:last], first, breaklist, | |
self.curframe) | |
self.lineno=min(last,len(lines)) | |
iflen(lines)< last: | |
self.message('[EOF]') | |
exceptKeyboardInterrupt: | |
pass | |
do_l= do_list | |
defdo_longlist(self,arg): | |
"""longlist | ll | |
List the whole source code for the current function or frame. | |
""" | |
filename=self.curframe.f_code.co_filename | |
breaklist=self.get_file_breaks(filename) | |
try: | |
lines, lineno= getsourcelines(self.curframe) | |
exceptOSErroras err: | |
self.error(err) | |
return | |
self._print_lines(lines, lineno, breaklist,self.curframe) | |
do_ll= do_longlist | |
defdo_source(self,arg): | |
"""source expression | |
Try to get source code for the given object and display it. | |
""" | |
try: | |
obj=self._getval(arg) | |
except: | |
return | |
try: | |
lines, lineno= getsourcelines(obj) | |
except (OSError,TypeError)as err: | |
self.error(err) | |
return | |
self._print_lines(lines, lineno) | |
complete_source= _complete_expression | |
def_print_lines(self,lines,start,breaks=(),frame=None): | |
"""Print a range of lines.""" | |
if frame: | |
current_lineno= frame.f_lineno | |
exc_lineno=self.tb_lineno.get(frame,-1) | |
else: | |
current_lineno= exc_lineno=-1 | |
for lineno, lineinenumerate(lines, start): | |
s=str(lineno).rjust(3) | |
iflen(s)<4: | |
s+='' | |
if linenoin breaks: | |
s+='B' | |
else: | |
s+='' | |
if lineno== current_lineno: | |
s+='->' | |
elif lineno== exc_lineno: | |
s+='>>' | |
self.message(s+'\t'+ line.rstrip()) | |
defdo_whatis(self,arg): | |
"""whatis arg | |
Print the type of the argument. | |
""" | |
try: | |
value=self._getval(arg) | |
except: | |
# _getval() already printed the error | |
return | |
code=None | |
# Is it a function? | |
try: | |
code= value.__code__ | |
exceptException: | |
pass | |
if code: | |
self.message('Function%s'% code.co_name) | |
return | |
# Is it an instance method? | |
try: | |
code= value.__func__.__code__ | |
exceptException: | |
pass | |
if code: | |
self.message('Method%s'% code.co_name) | |
return | |
# Is it a class? | |
if value.__class__istype: | |
self.message('Class%s.%s'% (value.__module__, value.__qualname__)) | |
return | |
# None of the above... | |
self.message(type(value)) | |
complete_whatis= _complete_expression | |
defdo_display(self,arg): | |
"""display [expression] | |
Display the value of the expression if it changed, each time execution | |
stops in the current frame. | |
Without expression, list all display expressions for the current frame. | |
""" | |
ifnot arg: | |
self.message('Currently displaying:') | |
for iteminself.displaying.get(self.curframe, {}).items(): | |
self.message('%s:%r'% item) | |
else: | |
val=self._getval_except(arg) | |
self.displaying.setdefault(self.curframe, {})[arg]= val | |
self.message('display%s:%r'% (arg, val)) | |
complete_display= _complete_expression | |
defdo_undisplay(self,arg): | |
"""undisplay [expression] | |
Do not display the expression any more in the current frame. | |
Without expression, clear all display expressions for the current frame. | |
""" | |
if arg: | |
try: | |
delself.displaying.get(self.curframe, {})[arg] | |
exceptKeyError: | |
self.error('not displaying%s'% arg) | |
else: | |
self.displaying.pop(self.curframe,None) | |
defcomplete_undisplay(self,text,line,begidx,endidx): | |
return [efor einself.displaying.get(self.curframe, {}) | |
if e.startswith(text)] | |
defdo_interact(self,arg): | |
"""interact | |
Start an interactive interpreter whose global namespace | |
contains all the (global and local) names found in the current scope. | |
""" | |
ns= {**self.curframe.f_globals,**self.curframe_locals} | |
code.interact("*interactive*",local=ns) | |
defdo_alias(self,arg): | |
"""alias [name [command [parameter parameter ...] ]] | |
Create an alias called 'name' that executes 'command'. The | |
command must *not* be enclosed in quotes. Replaceable | |
parameters can be indicated by %1, %2, and so on, while %* is | |
replaced by all the parameters. If no command is given, the | |
current alias for name is shown. If no name is given, all | |
aliases are listed. | |
Aliases may be nested and can contain anything that can be | |
legally typed at the pdb prompt. Note! You *can* override | |
internal pdb commands with aliases! Those internal commands | |
are then hidden until the alias is removed. Aliasing is | |
recursively applied to the first word of the command line; all | |
other words in the line are left alone. | |
As an example, here are two useful aliases (especially when | |
placed in the .pdbrc file): | |
# Print instance variables (usage "pi classInst") | |
alias pi for k in %1.__dict__.keys(): print("%1.",k,"=",%1.__dict__[k]) | |
# Print instance variables in self | |
alias ps pi self | |
""" | |
args= arg.split() | |
iflen(args)==0: | |
keys=sorted(self.aliases.keys()) | |
for aliasin keys: | |
self.message("%s =%s"% (alias,self.aliases[alias])) | |
return | |
if args[0]inself.aliasesandlen(args)==1: | |
self.message("%s =%s"% (args[0],self.aliases[args[0]])) | |
else: | |
self.aliases[args[0]]=''.join(args[1:]) | |
defdo_unalias(self,arg): | |
"""unalias name | |
Delete the specified alias. | |
""" | |
args= arg.split() | |
iflen(args)==0:return | |
if args[0]inself.aliases: | |
delself.aliases[args[0]] | |
defcomplete_unalias(self,text,line,begidx,endidx): | |
return [afor ainself.aliasesif a.startswith(text)] | |
# List of all the commands making the program resume execution. | |
commands_resuming= ['do_continue','do_step','do_next','do_return', | |
'do_quit','do_jump'] | |
# Print a traceback starting at the top stack frame. | |
# The most recently entered frame is printed last; | |
# this is different from dbx and gdb, but consistent with | |
# the Python interpreter's stack trace. | |
# It is also consistent with the up/down commands (which are | |
# compatible with dbx and gdb: up moves towards 'main()' | |
# and down moves towards the most recent stack frame). | |
defprint_stack_trace(self): | |
try: | |
for frame_linenoinself.stack: | |
self.print_stack_entry(frame_lineno) | |
exceptKeyboardInterrupt: | |
pass | |
defprint_stack_entry(self,frame_lineno,prompt_prefix=line_prefix): | |
frame, lineno= frame_lineno | |
if frameisself.curframe: | |
prefix='>' | |
else: | |
prefix='' | |
self.message(prefix+ | |
self.format_stack_entry(frame_lineno, prompt_prefix)) | |
# Provide help | |
defdo_help(self,arg): | |
"""h(elp) | |
Without argument, print the list of available commands. | |
With a command name as argument, print help about that command. | |
"help pdb" shows the full pdb documentation. | |
"help exec" gives help on the ! command. | |
""" | |
ifnot arg: | |
return cmd.Cmd.do_help(self, arg) | |
try: | |
try: | |
topic=getattr(self,'help_'+ arg) | |
return topic() | |
exceptAttributeError: | |
command=getattr(self,'do_'+ arg) | |
exceptAttributeError: | |
self.error('No help for%r'% arg) | |
else: | |
if sys.flags.optimize>=2: | |
self.error('No help for%r; please do not run Python with -OO' | |
'if you need command help'% arg) | |
return | |
self.message(command.__doc__.rstrip()) | |
do_h= do_help | |
defhelp_exec(self): | |
"""(!) statement | |
Execute the (one-line) statement in the context of the current | |
stack frame. The exclamation point can be omitted unless the | |
first word of the statement resembles a debugger command. To | |
assign to a global variable you must always prefix the command | |
with a 'global' command, e.g.: | |
(Pdb) global list_options; list_options = ['-l'] | |
(Pdb) | |
""" | |
self.message((self.help_exec.__doc__or'').strip()) | |
defhelp_pdb(self): | |
help() | |
# other helper functions | |
deflookupmodule(self,filename): | |
"""Helper function for break/clear parsing -- may be overridden. | |
lookupmodule() translates (possibly incomplete) file or module name | |
into an absolute file name. | |
""" | |
if os.path.isabs(filename)and os.path.exists(filename): | |
return filename | |
f= os.path.join(sys.path[0], filename) | |
if os.path.exists(f)andself.canonic(f)==self.mainpyfile: | |
return f | |
root, ext= os.path.splitext(filename) | |
if ext=='': | |
filename= filename+'.py' | |
if os.path.isabs(filename): | |
return filename | |
for dirnamein sys.path: | |
while os.path.islink(dirname): | |
dirname= os.readlink(dirname) | |
fullname= os.path.join(dirname, filename) | |
if os.path.exists(fullname): | |
return fullname | |
returnNone | |
def_runmodule(self,module_name): | |
self._wait_for_mainpyfile=True | |
self._user_requested_quit=False | |
import runpy | |
mod_name, mod_spec, code= runpy._get_module_details(module_name) | |
self.mainpyfile=self.canonic(code.co_filename) | |
import __main__ | |
__main__.__dict__.clear() | |
__main__.__dict__.update({ | |
"__name__":"__main__", | |
"__file__":self.mainpyfile, | |
"__package__": mod_spec.parent, | |
"__loader__": mod_spec.loader, | |
"__spec__": mod_spec, | |
"__builtins__":__builtins__, | |
}) | |
self.run(code) | |
def_runscript(self,filename): | |
# The script has to run in __main__ namespace (or imports from | |
# __main__ will break). | |
# | |
# So we clear up the __main__ and set several special variables | |
# (this gets rid of pdb's globals and cleans old variables on restarts). | |
import __main__ | |
__main__.__dict__.clear() | |
__main__.__dict__.update({"__name__" :"__main__", | |
"__file__" : filename, | |
"__builtins__":__builtins__, | |
}) | |
# When bdb sets tracing, a number of call and line events happens | |
# BEFORE debugger even reaches user's code (and the exact sequence of | |
# events depends on python version). So we take special measures to | |
# avoid stopping before we reach the main script (see user_line and | |
# user_call for details). | |
self._wait_for_mainpyfile=True | |
self.mainpyfile=self.canonic(filename) | |
self._user_requested_quit=False | |
withopen(filename,"rb")as fp: | |
statement="exec(compile(%r,%r, 'exec'))"% \ | |
(fp.read(),self.mainpyfile) | |
self.run(statement) | |
# Collect all command help into docstring, if not run with -OO | |
if__doc__isnotNone: | |
# unfortunately we can't guess this order from the class definition | |
_help_order= [ | |
'help','where','down','up','break','tbreak','clear','disable', | |
'enable','ignore','condition','commands','step','next','until', | |
'jump','return','retval','run','continue','list','longlist', | |
'args','p','pp','whatis','source','display','undisplay', | |
'interact','alias','unalias','debug','quit', | |
] | |
for _commandin _help_order: | |
__doc__+=getattr(Pdb,'do_'+ _command).__doc__.strip()+'\n\n' | |
__doc__+= Pdb.help_exec.__doc__ | |
del _help_order, _command | |
# Simplified interface | |
defrun(statement,globals=None,locals=None): | |
Pdb().run(statement,globals,locals) | |
defruneval(expression,globals=None,locals=None): | |
return Pdb().runeval(expression,globals,locals) | |
defrunctx(statement,globals,locals): | |
# B/W compatibility | |
run(statement,globals,locals) | |
defruncall(*args,**kwds): | |
return Pdb().runcall(*args,**kwds) | |
defset_trace(*,header=None): | |
pdb= Pdb() | |
if headerisnotNone: | |
pdb.message(header) | |
pdb.set_trace(sys._getframe().f_back) | |
# Post-Mortem interface | |
defpost_mortem(t=None): | |
# handling the default | |
if tisNone: | |
# sys.exc_info() returns (type, value, traceback) if an exception is | |
# being handled, otherwise it returns None | |
t= sys.exc_info()[2] | |
if tisNone: | |
raiseValueError("A valid traceback must be passed if no" | |
"exception is being handled") | |
p= Pdb() | |
p.reset() | |
p.interaction(None, t) | |
defpm(): | |
post_mortem(sys.last_traceback) | |
# Main program for testing | |
TESTCMD='import x; x.main()' | |
deftest(): | |
run(TESTCMD) | |
# print help | |
defhelp(): | |
import pydoc | |
pydoc.pager(__doc__) | |
_usage="""\ | |
usage: pdb.py [-c command] ... [-m module | pyfile] [arg] ... | |
Debug the Python program given by pyfile. Alternatively, | |
an executable module or package to debug can be specified using | |
the -m switch. | |
Initial commands are read from .pdbrc files in your home directory | |
and in the current directory, if they exist. Commands supplied with | |
-c are executed after commands from .pdbrc files. | |
To let the script run until an exception occurs, use "-c continue". | |
To let the script run up to a given line X in the debugged file, use | |
"-c 'until X'".""" | |
defmain(): | |
import getopt | |
opts, args= getopt.getopt(sys.argv[1:],'mhc:', ['--help','--command=']) | |
ifnot args: | |
print(_usage) | |
sys.exit(2) | |
commands= [] | |
run_as_module=False | |
for opt, optargin opts: | |
if optin ['-h','--help']: | |
print(_usage) | |
sys.exit() | |
elif optin ['-c','--command']: | |
commands.append(optarg) | |
elif optin ['-m']: | |
run_as_module=True | |
mainpyfile= args[0]# Get script filename | |
ifnot run_as_moduleandnot os.path.exists(mainpyfile): | |
print('Error:', mainpyfile,'does not exist') | |
sys.exit(1) | |
sys.argv[:]= args# Hide "pdb.py" and pdb options from argument list | |
# Replace pdb's dir with script's dir in front of module search path. | |
ifnot run_as_module: | |
sys.path[0]= os.path.dirname(mainpyfile) | |
# Note on saving/restoring sys.argv: it's a good idea when sys.argv was | |
# modified by the script being debugged. It's a bad idea when it was | |
# changed by the user from the command line. There is a "restart" command | |
# which allows explicit specification of command line arguments. | |
pdb= Pdb() | |
pdb.rcLines.extend(commands) | |
whileTrue: | |
try: | |
if run_as_module: | |
pdb._runmodule(mainpyfile) | |
else: | |
pdb._runscript(mainpyfile) | |
if pdb._user_requested_quit: | |
break | |
print("The program finished and will be restarted") | |
except Restart: | |
print("Restarting", mainpyfile,"with arguments:") | |
print("\t"+"".join(args)) | |
exceptSystemExit: | |
# In most cases SystemExit does not warrant a post-mortem session. | |
print("The program exited via sys.exit(). Exit status:",end='') | |
print(sys.exc_info()[1]) | |
exceptSyntaxError: | |
traceback.print_exc() | |
sys.exit(1) | |
except: | |
traceback.print_exc() | |
print("Uncaught exception. Entering post mortem debugging") | |
print("Running 'cont' or 'step' will restart the program") | |
t= sys.exc_info()[2] | |
pdb.interaction(None, t) | |
print("Post mortem debugger finished. The"+ mainpyfile+ | |
" will be restarted") | |
# When invoked as main program, invoke the debugger on a script | |
if__name__=='__main__': | |
import pdb | |
pdb.main() |