diff options
Diffstat (limited to 'Tools/idle/StackViewer.py')
| -rw-r--r-- | Tools/idle/StackViewer.py | 288 |
1 files changed, 288 insertions, 0 deletions
diff --git a/Tools/idle/StackViewer.py b/Tools/idle/StackViewer.py new file mode 100644 index 0000000..a59668a --- /dev/null +++ b/Tools/idle/StackViewer.py @@ -0,0 +1,288 @@ +import string +import sys +import os +from Tkinter import * +import linecache +from repr import Repr + + +class StackViewer: + + def __init__(self, root=None, flist=None): + self.flist = flist + # Create root and/or toplevel window + if not root: + import Tkinter + root = Tkinter._default_root + if not root: + root = top = Tk() + else: + top = Toplevel(root) + self.root = root + self.top = top + top.wm_title("Stack viewer") + # Create top frame, with scrollbar and listbox + self.topframe = Frame(top) + self.topframe.pack(fill="both", expand=1) + self.vbar = Scrollbar(self.topframe, name="vbar") + self.vbar.pack(side="right", fill="y") + self.listbox = Listbox(self.topframe, exportselection=0, + takefocus=1, width=60) + self.listbox.pack(expand=1, fill="both") + # Tie listbox and scrollbar together + self.vbar["command"] = self.listbox.yview + self.listbox["yscrollcommand"] = self.vbar.set + # Bind events to the list box + self.listbox.bind("<ButtonRelease-1>", self.click_event) + self.listbox.bind("<Double-ButtonRelease-1>", self.double_click_event) + self.listbox.bind("<ButtonPress-3>", self.popup_event) + self.listbox.bind("<Key-Up>", self.up_event) + self.listbox.bind("<Key-Down>", self.down_event) + # Load the stack + linecache.checkcache() + stack = getstack() + self.load_stack(stack) + + def load_stack(self, stack): + self.stack = stack + l = self.listbox + l.delete(0, END) + if len(stack) > 10: + l["height"] = 10 + self.topframe.pack(expand=1) + else: + l["height"] = len(stack) + self.topframe.pack(expand=0) + for frame, lineno in stack: + try: + modname = frame.f_globals["__name__"] + except: + modname = "?" + code = frame.f_code + filename = code.co_filename + funcname = code.co_name + sourceline = linecache.getline(filename, lineno) + sourceline = string.strip(sourceline) + if funcname in ("?", "", None): + item = "%s, line %d: %s" % (modname, lineno, sourceline) + else: + item = "%s.%s(), line %d: %s" % (modname, funcname, + lineno, sourceline) + l.insert(END, item) + l.focus_set() + l.selection_clear(0, "end") + l.activate("end") + l.see("end") + + rmenu = None + + def click_event(self, event): + self.listbox.activate("@%d,%d" % (event.x, event.y)) + self.show_stack_frame() + return "break" + + def popup_event(self, event): + if not self.rmenu: + self.make_menu() + rmenu = self.rmenu + self.event = event + self.listbox.activate("@%d,%d" % (event.x, event.y)) + rmenu.tk_popup(event.x_root, event.y_root) + + def make_menu(self): + rmenu = Menu(self.top, tearoff=0) + rmenu.add_command(label="Go to source line", + command=self.goto_source_line) + rmenu.add_command(label="Show stack frame", + command=self.show_stack_frame) + self.rmenu = rmenu + + def goto_source_line(self): + index = self.listbox.index("active") + self.show_source(index) + + def show_stack_frame(self): + index = self.listbox.index("active") + self.show_frame(index) + + def double_click_event(self, event): + index = self.listbox.index("active") + self.show_source(index) + return "break" + + def up_event(self, event): + index = self.listbox.index("active") - 1 + if index < 0: + self.top.bell() + return "break" + self.show_frame(index) + return "break" + + def down_event(self, event): + index = self.listbox.index("active") + 1 + if index >= len(self.stack): + self.top.bell() + return "break" + self.show_frame(index) + return "break" + + def show_source(self, index): + if not 0 <= index < len(self.stack): + self.top.bell() + return + frame, lineno = self.stack[index] + code = frame.f_code + filename = code.co_filename + if not self.flist: + self.top.bell() + return + if not os.path.exists(filename): + self.top.bell() + return + edit = self.flist.open(filename) + edit.gotoline(lineno) + + localsframe = None + localsviewer = None + localsdict = None + globalsframe = None + globalsviewer = None + globalsdict = None + curframe = None + + def show_frame(self, index): + if not 0 <= index < len(self.stack): + self.top.bell() + return + self.listbox.selection_clear(0, "end") + self.listbox.selection_set(index) + self.listbox.activate(index) + self.listbox.see(index) + self.listbox.focus_set() + frame, lineno = self.stack[index] + if frame is self.curframe: + return + self.curframe = None + if frame.f_globals is not self.globalsdict: + self.show_globals(frame) + self.show_locals(frame) + self.curframe = frame + + def show_globals(self, frame): + title = "Global Variables" + if frame.f_globals.has_key("__name__"): + try: + name = str(frame.f_globals["__name__"]) + "" + except: + name = "" + if name: + title = title + " in module " + name + self.globalsdict = None + if self.globalsviewer: + self.globalsviewer.close() + self.globalsviewer = None + if not self.globalsframe: + self.globalsframe = Frame(self.top) + self.globalsdict = frame.f_globals + self.globalsviewer = NamespaceViewer( + self.globalsframe, + title, + self.globalsdict) + self.globalsframe.pack(fill="both", side="bottom") + + def show_locals(self, frame): + self.localsdict = None + if self.localsviewer: + self.localsviewer.close() + self.localsviewer = None + if frame.f_locals is not frame.f_globals: + title = "Local Variables" + code = frame.f_code + funcname = code.co_name + if funcname not in ("?", "", None): + title = title + " in " + funcname + if not self.localsframe: + self.localsframe = Frame(self.top) + self.localsdict = frame.f_locals + self.localsviewer = NamespaceViewer( + self.localsframe, + title, + self.localsdict) + self.localsframe.pack(fill="both", side="top") + else: + if self.localsframe: + self.localsframe.forget() + + +def getstack(t=None, f=None): + if t is None: + t = sys.last_traceback + stack = [] + if t and t.tb_frame is f: + t = t.tb_next + while f is not None: + stack.append((f, f.f_lineno)) + if f is self.botframe: + break + f = f.f_back + stack.reverse() + while t is not None: + stack.append((t.tb_frame, t.tb_lineno)) + t = t.tb_next + return stack + + +class NamespaceViewer: + + def __init__(self, frame, title, dict): + width = 0 + height = 20*len(dict) # XXX 20 == observed height of Entry widget + self.frame = frame + self.title = title + self.dict = dict + self.repr = Repr() + self.repr.maxstring = 60 + self.repr.maxother = 60 + 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") + names = dict.keys() + names.sort() + row = 0 + for name in names: + value = dict[name] + svalue = self.repr.repr(value) # repr(value) + l = Label(subframe, text=name) + l.grid(row=row, column=0, sticky="nw") +## l = Label(subframe, text=svalue, justify="l", wraplength=300) + l = Entry(subframe, width=0, borderwidth=0) + l.insert(0, svalue) +## l["state"] = "disabled" + l.grid(row=row, column=1, sticky="nw") + row = row+1 + frame.update_idletasks() # Alas! + width = subframe.winfo_reqwidth() + height = subframe.winfo_reqheight() + 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): + for c in self.subframe, self.label, self.vbar, self.canvas: + try: + c.destroy() + except: + pass |