Spaces:
Running
Running
| import bdb | |
| import os | |
| from tkinter import * | |
| from tkinter.ttk import Frame, Scrollbar | |
| from idlelib import macosx | |
| from idlelib.scrolledlist import ScrolledList | |
| from idlelib.window import ListedToplevel | |
| class Idb(bdb.Bdb): | |
| def __init__(self, gui): | |
| self.gui = gui # An instance of Debugger or proxy of remote. | |
| bdb.Bdb.__init__(self) | |
| def user_line(self, frame): | |
| if self.in_rpc_code(frame): | |
| self.set_step() | |
| return | |
| message = self.__frame2message(frame) | |
| try: | |
| self.gui.interaction(message, frame) | |
| except TclError: # When closing debugger window with [x] in 3.x | |
| pass | |
| def user_exception(self, frame, info): | |
| if self.in_rpc_code(frame): | |
| self.set_step() | |
| return | |
| message = self.__frame2message(frame) | |
| self.gui.interaction(message, frame, info) | |
| def in_rpc_code(self, frame): | |
| if frame.f_code.co_filename.count('rpc.py'): | |
| return True | |
| else: | |
| prev_frame = frame.f_back | |
| prev_name = prev_frame.f_code.co_filename | |
| if 'idlelib' in prev_name and 'debugger' in prev_name: | |
| # catch both idlelib/debugger.py and idlelib/debugger_r.py | |
| # on both Posix and Windows | |
| return False | |
| return self.in_rpc_code(prev_frame) | |
| 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 # If passed, a proxy of remote instance. | |
| self.frame = None | |
| self.make_gui() | |
| self.interacting = 0 | |
| self.nesting_level = 0 | |
| def run(self, *args): | |
| # Deal with the scenario where we've already got a program running | |
| # in the debugger and we want to start another. If that is the case, | |
| # our second 'run' was invoked from an event dispatched not from | |
| # the main event loop, but from the nested event loop in 'interaction' | |
| # below. So our stack looks something like this: | |
| # outer main event loop | |
| # run() | |
| # <running program with traces> | |
| # callback to debugger's interaction() | |
| # nested event loop | |
| # run() for second command | |
| # | |
| # This kind of nesting of event loops causes all kinds of problems | |
| # (see e.g. issue #24455) especially when dealing with running as a | |
| # subprocess, where there's all kinds of extra stuff happening in | |
| # there - insert a traceback.print_stack() to check it out. | |
| # | |
| # By this point, we've already called restart_subprocess() in | |
| # ScriptBinding. However, we also need to unwind the stack back to | |
| # that outer event loop. To accomplish this, we: | |
| # - return immediately from the nested run() | |
| # - abort_loop ensures the nested event loop will terminate | |
| # - the debugger's interaction routine completes normally | |
| # - the restart_subprocess() will have taken care of stopping | |
| # the running program, which will also let the outer run complete | |
| # | |
| # That leaves us back at the outer main event loop, at which point our | |
| # after event can fire, and we'll come back to this routine with a | |
| # clean stack. | |
| if self.nesting_level > 0: | |
| self.abort_loop() | |
| self.root.after(100, lambda: self.run(*args)) | |
| return | |
| try: | |
| self.interacting = 1 | |
| return self.idb.run(*args) | |
| finally: | |
| self.interacting = 0 | |
| def close(self, event=None): | |
| try: | |
| self.quit() | |
| except Exception: | |
| pass | |
| if self.interacting: | |
| self.top.bell() | |
| return | |
| if self.stackviewer: | |
| self.stackviewer.close(); self.stackviewer = None | |
| # Clean up pyshell if user clicked debugger control close widget. | |
| # (Causes a harmless extra cycle through close_debugger() if user | |
| # toggled debugger from pyshell Debug menu) | |
| self.pyshell.close_debugger() | |
| # Now close the debugger control window.... | |
| 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.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.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.wakeup() | |
| # Nested main loop: Tkinter's main loop is not reentrant, so use | |
| # Tcl's vwait facility, which reenters the event loop until an | |
| # event handler sets the variable we're waiting on | |
| self.nesting_level += 1 | |
| self.root.tk.call('vwait', '::idledebugwait') | |
| self.nesting_level -= 1 | |
| # | |
| 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.abort_loop() | |
| def step(self): | |
| self.idb.set_step() | |
| self.abort_loop() | |
| def next(self): | |
| self.idb.set_next(self.frame) | |
| self.abort_loop() | |
| def ret(self): | |
| self.idb.set_return(self.frame) | |
| self.abort_loop() | |
| def quit(self): | |
| self.idb.set_quit() | |
| self.abort_loop() | |
| def abort_loop(self): | |
| self.root.tk.call('set', '::idledebugwait', '1') | |
| stackviewer = None | |
| def show_stack(self): | |
| if not self.stackviewer and self.vstack.get(): | |
| self.stackviewer = sv = 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, stackitem): | |
| self.frame = stackitem[0] # lineno is stackitem[1] | |
| self.show_variables() | |
| localsviewer = None | |
| globalsviewer = None | |
| def show_locals(self): | |
| lv = self.localsviewer | |
| if self.vlocals.get(): | |
| if not lv: | |
| self.localsviewer = 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 = 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 | |
| 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, filename, lineno): | |
| self.idb.set_break(filename, lineno) | |
| def clear_breakpoint_here(self, filename, lineno): | |
| self.idb.clear_break(filename, lineno) | |
| def clear_file_breaks(self, filename): | |
| self.idb.clear_all_file_breaks(filename) | |
| def load_breakpoints(self): | |
| "Load PyShellEditorWindow breakpoints into subprocess debugger" | |
| for editwin in self.pyshell.flist.inversedict: | |
| filename = editwin.io.filename | |
| try: | |
| for lineno in editwin.breakpoints: | |
| self.set_breakpoint_here(filename, lineno) | |
| except AttributeError: | |
| continue | |
| class StackViewer(ScrolledList): | |
| def __init__(self, master, flist, gui): | |
| if macosx.isAquaTk(): | |
| # At least on with the stock AquaTk version on OSX 10.4 you'll | |
| # get a shaking GUI that eventually kills IDLE if the width | |
| # argument is specified. | |
| ScrolledList.__init__(self, master) | |
| else: | |
| ScrolledList.__init__(self, master, width=80) | |
| self.flist = flist | |
| self.gui = gui | |
| self.stack = [] | |
| def load_stack(self, stack, index=None): | |
| self.stack = stack | |
| self.clear() | |
| for i in range(len(stack)): | |
| frame, lineno = stack[i] | |
| try: | |
| modname = frame.f_globals["__name__"] | |
| except: | |
| modname = "?" | |
| code = frame.f_code | |
| filename = code.co_filename | |
| funcname = code.co_name | |
| import linecache | |
| sourceline = linecache.getline(filename, lineno) | |
| sourceline = sourceline.strip() | |
| if funcname in ("?", "", None): | |
| item = "%s, line %d: %s" % (modname, lineno, sourceline) | |
| else: | |
| item = "%s.%s(), line %d: %s" % (modname, funcname, | |
| lineno, sourceline) | |
| if i == index: | |
| item = "> " + item | |
| self.append(item) | |
| if index is not None: | |
| self.select(index) | |
| def popup_event(self, event): | |
| "override base method" | |
| if self.stack: | |
| return ScrolledList.popup_event(self, event) | |
| def fill_menu(self): | |
| "override base method" | |
| menu = self.menu | |
| menu.add_command(label="Go to source line", | |
| command=self.goto_source_line) | |
| menu.add_command(label="Show stack frame", | |
| command=self.show_stack_frame) | |
| def on_select(self, index): | |
| "override base method" | |
| if 0 <= index < len(self.stack): | |
| self.gui.show_frame(self.stack[index]) | |
| def on_double(self, index): | |
| "override base method" | |
| self.show_source(index) | |
| def goto_source_line(self): | |
| index = self.listbox.index("active") | |
| self.show_source(index) | |
| def show_stack_frame(self): | |
| index = self.listbox.index("active") | |
| if 0 <= index < len(self.stack): | |
| self.gui.show_frame(self.stack[index]) | |
| def show_source(self, index): | |
| if not (0 <= index < len(self.stack)): | |
| return | |
| frame, lineno = self.stack[index] | |
| code = frame.f_code | |
| filename = code.co_filename | |
| if os.path.isfile(filename): | |
| edit = self.flist.open(filename) | |
| if edit: | |
| edit.gotoline(lineno) | |
| class NamespaceViewer: | |
| def __init__(self, master, title, dict=None): | |
| width = 0 | |
| height = 40 | |
| if dict: | |
| height = 20*len(dict) # XXX 20 == observed height of Entry widget | |
| self.master = master | |
| self.title = title | |
| import reprlib | |
| self.repr = reprlib.Repr() | |
| self.repr.maxstring = 60 | |
| self.repr.maxother = 60 | |
| self.frame = frame = Frame(master) | |
| self.frame.pack(expand=1, fill="both") | |
| self.label = Label(frame, text=title, borderwidth=2, relief="groove") | |
| self.label.pack(fill="x") | |
| self.vbar = vbar = Scrollbar(frame, name="vbar") | |
| vbar.pack(side="right", fill="y") | |
| self.canvas = canvas = Canvas(frame, | |
| height=min(300, max(40, height)), | |
| scrollregion=(0, 0, width, height)) | |
| canvas.pack(side="left", fill="both", expand=1) | |
| vbar["command"] = canvas.yview | |
| canvas["yscrollcommand"] = vbar.set | |
| self.subframe = subframe = Frame(canvas) | |
| self.sfid = canvas.create_window(0, 0, window=subframe, anchor="nw") | |
| self.load_dict(dict) | |
| dict = -1 | |
| def load_dict(self, dict, force=0, rpc_client=None): | |
| if dict is self.dict and not force: | |
| return | |
| subframe = self.subframe | |
| frame = self.frame | |
| for c in list(subframe.children.values()): | |
| c.destroy() | |
| self.dict = None | |
| if not dict: | |
| l = Label(subframe, text="None") | |
| l.grid(row=0, column=0) | |
| else: | |
| #names = sorted(dict) | |
| ### | |
| # Because of (temporary) limitations on the dict_keys type (not yet | |
| # public or pickleable), have the subprocess to send a list of | |
| # keys, not a dict_keys object. sorted() will take a dict_keys | |
| # (no subprocess) or a list. | |
| # | |
| # There is also an obscure bug in sorted(dict) where the | |
| # interpreter gets into a loop requesting non-existing dict[0], | |
| # dict[1], dict[2], etc from the debugger_r.DictProxy. | |
| ### | |
| keys_list = dict.keys() | |
| names = sorted(keys_list) | |
| ### | |
| row = 0 | |
| for name in names: | |
| value = dict[name] | |
| svalue = self.repr.repr(value) # repr(value) | |
| # Strip extra quotes caused by calling repr on the (already) | |
| # repr'd value sent across the RPC interface: | |
| if rpc_client: | |
| svalue = svalue[1:-1] | |
| l = Label(subframe, text=name) | |
| l.grid(row=row, column=0, sticky="nw") | |
| l = Entry(subframe, width=0, borderwidth=0) | |
| l.insert(0, svalue) | |
| l.grid(row=row, column=1, sticky="nw") | |
| row = row+1 | |
| self.dict = dict | |
| # XXX Could we use a <Configure> callback for the following? | |
| subframe.update_idletasks() # Alas! | |
| width = subframe.winfo_reqwidth() | |
| height = subframe.winfo_reqheight() | |
| canvas = self.canvas | |
| self.canvas["scrollregion"] = (0, 0, width, height) | |
| if height > 300: | |
| canvas["height"] = 300 | |
| frame.pack(expand=1) | |
| else: | |
| canvas["height"] = height | |
| frame.pack(expand=0) | |
| def close(self): | |
| self.frame.destroy() | |
| if __name__ == "__main__": | |
| from unittest import main | |
| main('idlelib.idle_test.test_debugger', verbosity=2, exit=False) | |
| # TODO: htest? | |