| import os |
| import bdb |
| import types |
| import traceback |
| from Tkinter import * |
| from WindowList import ListedToplevel |
| |
| import StackViewer |
| |
| |
| class Idb(bdb.Bdb): |
| |
| def __init__(self, gui): |
| self.gui = gui |
| bdb.Bdb.__init__(self) |
| |
| def user_line(self, frame): |
| # get the currently executing function |
| co_filename = frame.f_code.co_filename |
| co_name = frame.f_code.co_name |
| try: |
| func = frame.f_locals[co_name] |
| if getattr(func, "DebuggerStepThrough", 0): |
| print "XXXX DEBUGGER STEPPING THROUGH" |
| self.set_step() |
| return |
| except: |
| pass |
| if co_filename in ('rpc.py', '<string>'): |
| self.set_step() |
| return |
| if co_filename.endswith('threading.py'): |
| self.set_step() |
| return |
| message = self.__frame2message(frame) |
| self.gui.interaction(message, frame) |
| |
| def user_exception(self, frame, info): |
| message = self.__frame2message(frame) |
| self.gui.interaction(message, frame, info) |
| |
| def __frame2message(self, frame): |
| code = frame.f_code |
| filename = code.co_filename |
| lineno = frame.f_lineno |
| basename = os.path.basename(filename) |
| message = "%s:%s" % (basename, lineno) |
| if code.co_name != "?": |
| message = "%s: %s()" % (message, code.co_name) |
| return message |
| |
| |
| class Debugger: |
| |
| vstack = vsource = vlocals = vglobals = None |
| |
| def __init__(self, pyshell, idb=None): |
| if idb is None: |
| idb = Idb(self) |
| self.pyshell = pyshell |
| self.idb = idb |
| self.frame = None |
| self.make_gui() |
| self.interacting = 0 |
| |
| def run(self, *args): |
| try: |
| self.interacting = 1 |
| return self.idb.run(*args) |
| finally: |
| self.interacting = 0 |
| |
| def close(self, event=None): |
| if self.interacting: |
| self.top.bell() |
| return |
| if self.stackviewer: |
| self.stackviewer.close(); self.stackviewer = None |
| self.pyshell.close_debugger() |
| self.top.destroy() |
| |
| def make_gui(self): |
| pyshell = self.pyshell |
| self.flist = pyshell.flist |
| self.root = root = pyshell.root |
| self.top = top =ListedToplevel(root) |
| self.top.wm_title("Debug Control") |
| self.top.wm_iconname("Debug") |
| top.wm_protocol("WM_DELETE_WINDOW", self.close) |
| self.top.bind("<Escape>", self.close) |
| # |
| self.bframe = bframe = Frame(top) |
| self.bframe.pack(anchor="w") |
| self.buttons = bl = [] |
| # |
| self.bcont = b = Button(bframe, text="Go", command=self.cont) |
| bl.append(b) |
| self.bstep = b = Button(bframe, text="Step", command=self.step) |
| bl.append(b) |
| self.bnext = b = Button(bframe, text="Over", command=self.next) |
| bl.append(b) |
| self.bret = b = Button(bframe, text="Out", command=self.ret) |
| bl.append(b) |
| self.bret = b = Button(bframe, text="Quit", command=self.quit) |
| bl.append(b) |
| # |
| for b in bl: |
| b.configure(state="disabled") |
| b.pack(side="left") |
| # |
| self.cframe = cframe = Frame(bframe) |
| self.cframe.pack(side="left") |
| # |
| if not self.vstack: |
| self.__class__.vstack = BooleanVar(top) |
| self.vstack.set(1) |
| self.bstack = Checkbutton(cframe, |
| text="Stack", command=self.show_stack, variable=self.vstack) |
| self.bstack.grid(row=0, column=0) |
| if not self.vsource: |
| self.__class__.vsource = BooleanVar(top) |
| ##self.vsource.set(1) |
| self.bsource = Checkbutton(cframe, |
| text="Source", command=self.show_source, variable=self.vsource) |
| self.bsource.grid(row=0, column=1) |
| if not self.vlocals: |
| self.__class__.vlocals = BooleanVar(top) |
| self.vlocals.set(1) |
| self.blocals = Checkbutton(cframe, |
| text="Locals", command=self.show_locals, variable=self.vlocals) |
| self.blocals.grid(row=1, column=0) |
| if not self.vglobals: |
| self.__class__.vglobals = BooleanVar(top) |
| ##self.vglobals.set(1) |
| self.bglobals = Checkbutton(cframe, |
| text="Globals", command=self.show_globals, variable=self.vglobals) |
| self.bglobals.grid(row=1, column=1) |
| # |
| self.status = Label(top, anchor="w") |
| self.status.pack(anchor="w") |
| self.error = Label(top, anchor="w") |
| self.error.pack(anchor="w", fill="x") |
| self.errorbg = self.error.cget("background") |
| # |
| self.fstack = Frame(top, height=1) |
| self.fstack.pack(expand=1, fill="both") |
| self.flocals = Frame(top) |
| self.flocals.pack(expand=1, fill="both") |
| self.fglobals = Frame(top, height=1) |
| self.fglobals.pack(expand=1, fill="both") |
| # |
| if self.vstack.get(): |
| self.show_stack() |
| if self.vlocals.get(): |
| self.show_locals() |
| if self.vglobals.get(): |
| self.show_globals() |
| |
| |
| def interaction(self, message, frame, info=None): |
| self.frame = frame |
| self.status.configure(text=message) |
| # |
| if info: |
| type, value, tb = info |
| try: |
| m1 = type.__name__ |
| except AttributeError: |
| m1 = "%s" % str(type) |
| if value is not None: |
| try: |
| m1 = "%s: %s" % (m1, str(value)) |
| except: |
| pass |
| bg = "yellow" |
| else: |
| m1 = "" |
| tb = None |
| bg = self.errorbg |
| self.error.configure(text=m1, background=bg) |
| # |
| sv = self.stackviewer |
| if sv: |
| stack, i = self.idb.get_stack(self.frame, tb) |
| sv.load_stack(stack, i) |
| # |
| self.show_variables(1) |
| # |
| if self.vsource.get(): |
| self.sync_source_line() |
| # |
| for b in self.buttons: |
| b.configure(state="normal") |
| # |
| self.top.tkraise() |
| self.root.mainloop() |
| # |
| for b in self.buttons: |
| b.configure(state="disabled") |
| self.status.configure(text="") |
| self.error.configure(text="", background=self.errorbg) |
| self.frame = None |
| |
| def sync_source_line(self): |
| frame = self.frame |
| if not frame: |
| return |
| filename, lineno = self.__frame2fileline(frame) |
| if filename[:1] + filename[-1:] != "<>" and os.path.exists(filename): |
| self.flist.gotofileline(filename, lineno) |
| |
| def __frame2fileline(self, frame): |
| code = frame.f_code |
| filename = code.co_filename |
| lineno = frame.f_lineno |
| return filename, lineno |
| |
| def cont(self): |
| self.idb.set_continue() |
| self.root.quit() |
| |
| def step(self): |
| self.idb.set_step() |
| self.root.quit() |
| |
| def next(self): |
| self.idb.set_next(self.frame) |
| self.root.quit() |
| |
| def ret(self): |
| self.idb.set_return(self.frame) |
| self.root.quit() |
| |
| def quit(self): |
| self.idb.set_quit() |
| self.root.quit() |
| |
| stackviewer = None |
| |
| def show_stack(self): |
| if not self.stackviewer and self.vstack.get(): |
| self.stackviewer = sv = StackViewer.StackViewer( |
| self.fstack, self.flist, self) |
| if self.frame: |
| stack, i = self.idb.get_stack(self.frame, None) |
| sv.load_stack(stack, i) |
| else: |
| sv = self.stackviewer |
| if sv and not self.vstack.get(): |
| self.stackviewer = None |
| sv.close() |
| self.fstack['height'] = 1 |
| |
| def show_source(self): |
| if self.vsource.get(): |
| self.sync_source_line() |
| |
| def show_frame(self, (frame, lineno)): |
| # Called from OldStackViewer |
| self.frame = frame |
| self.show_variables() |
| |
| localsviewer = None |
| globalsviewer = None |
| |
| def show_locals(self): |
| lv = self.localsviewer |
| if self.vlocals.get(): |
| if not lv: |
| self.localsviewer = StackViewer.NamespaceViewer( |
| self.flocals, "Locals") |
| else: |
| if lv: |
| self.localsviewer = None |
| lv.close() |
| self.flocals['height'] = 1 |
| self.show_variables() |
| |
| def show_globals(self): |
| gv = self.globalsviewer |
| if self.vglobals.get(): |
| if not gv: |
| self.globalsviewer = StackViewer.NamespaceViewer( |
| self.fglobals, "Globals") |
| else: |
| if gv: |
| self.globalsviewer = None |
| gv.close() |
| self.fglobals['height'] = 1 |
| self.show_variables() |
| |
| def show_variables(self, force=0): |
| lv = self.localsviewer |
| gv = self.globalsviewer |
| frame = self.frame |
| if not frame: |
| ldict = gdict = None |
| else: |
| ldict = frame.f_locals |
| gdict = frame.f_globals |
| if lv and gv and ldict is gdict: |
| ldict = None |
| # Calls OldStackviewer.NamespaceViewer.load_dict(): |
| if lv: |
| lv.load_dict(ldict, force, self.pyshell.interp.rpcclt) |
| if gv: |
| gv.load_dict(gdict, force, self.pyshell.interp.rpcclt) |
| |
| def set_breakpoint_here(self, edit): |
| text = edit.text |
| filename = edit.io.filename |
| if not filename: |
| text.bell() |
| return |
| lineno = int(float(text.index("insert"))) |
| msg = self.idb.set_break(filename, lineno) |
| if msg: |
| text.bell() |
| return |
| text.tag_add("BREAK", "insert linestart", "insert lineend +1char") |
| |
| def clear_breakpoint_here(self, edit): |
| text = edit.text |
| filename = edit.io.filename |
| if not filename: |
| text.bell() |
| return |
| lineno = int(float(text.index("insert"))) |
| msg = self.idb.clear_break(filename, lineno) |
| if msg: |
| text.bell() |
| return |
| text.tag_remove("BREAK", "insert linestart",\ |
| "insert lineend +1char") |
| |
| def clear_file_breaks(self, edit): |
| text = edit.text |
| filename = edit.io.filename |
| if not filename: |
| text.bell() |
| return |
| msg = self.idb.clear_all_file_breaks(filename) |
| if msg: |
| text.bell() |
| return |
| text.tag_delete("BREAK") |