mirror of
				https://github.com/python/cpython.git
				synced 2025-11-04 11:49:12 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			276 lines
		
	
	
	
		
			8.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			276 lines
		
	
	
	
		
			8.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
import string
 | 
						|
import sys
 | 
						|
import os
 | 
						|
from Tkinter import *
 | 
						|
import linecache
 | 
						|
from repr import Repr
 | 
						|
from WindowList import ListedToplevel
 | 
						|
 | 
						|
from ScrolledList import ScrolledList
 | 
						|
 | 
						|
 | 
						|
class StackBrowser:
 | 
						|
 | 
						|
    def __init__(self, root, flist, stack=None):
 | 
						|
        self.top = top = ListedToplevel(root)
 | 
						|
        top.protocol("WM_DELETE_WINDOW", self.close)
 | 
						|
        top.bind("<Key-Escape>", self.close)
 | 
						|
        top.wm_title("Stack viewer")
 | 
						|
        top.wm_iconname("Stack")
 | 
						|
        # Create help label
 | 
						|
        self.helplabel = Label(top,
 | 
						|
            text="Click once to view variables; twice for source",
 | 
						|
            borderwidth=2, relief="groove")
 | 
						|
        self.helplabel.pack(fill="x")
 | 
						|
        #
 | 
						|
        self.sv = StackViewer(top, flist, self)
 | 
						|
        if stack is None:
 | 
						|
            stack = get_stack()
 | 
						|
        self.sv.load_stack(stack)
 | 
						|
 | 
						|
    def close(self, event=None):
 | 
						|
        self.top.destroy()
 | 
						|
 | 
						|
    localsframe = None
 | 
						|
    localsviewer = None
 | 
						|
    localsdict = None
 | 
						|
    globalsframe = None
 | 
						|
    globalsviewer = None
 | 
						|
    globalsdict = None
 | 
						|
    curframe = None
 | 
						|
 | 
						|
    def show_frame(self, (frame, lineno)):
 | 
						|
        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()
 | 
						|
 | 
						|
 | 
						|
class StackViewer(ScrolledList):
 | 
						|
 | 
						|
    def __init__(self, master, flist, browser):
 | 
						|
        ScrolledList.__init__(self, master, width=80)
 | 
						|
        self.flist = flist
 | 
						|
        self.browser = browser
 | 
						|
        self.stack = []
 | 
						|
 | 
						|
    def load_stack(self, stack, index=None):
 | 
						|
        self.stack = stack
 | 
						|
        self.clear()
 | 
						|
##        if len(stack) > 10:
 | 
						|
##            l["height"] = 10
 | 
						|
##            self.topframe.pack(expand=1)
 | 
						|
##        else:
 | 
						|
##            l["height"] = len(stack)
 | 
						|
##            self.topframe.pack(expand=0)
 | 
						|
        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
 | 
						|
            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)
 | 
						|
            if i == index:
 | 
						|
                item = "> " + item
 | 
						|
            self.append(item)
 | 
						|
        if index is not None:
 | 
						|
            self.select(index)
 | 
						|
 | 
						|
    def popup_event(self, event):
 | 
						|
        if self.stack:
 | 
						|
            return ScrolledList.popup_event(self, event)
 | 
						|
 | 
						|
    def fill_menu(self):
 | 
						|
        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):
 | 
						|
        if 0 <= index < len(self.stack):
 | 
						|
            self.browser.show_frame(self.stack[index])
 | 
						|
 | 
						|
    def on_double(self, index):
 | 
						|
        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.browser.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)
 | 
						|
 | 
						|
 | 
						|
def get_stack(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
 | 
						|
 | 
						|
 | 
						|
def getexception(type=None, value=None):
 | 
						|
    if type is None:
 | 
						|
        type = sys.last_type
 | 
						|
        value = sys.last_value
 | 
						|
    if hasattr(type, "__name__"):
 | 
						|
        type = type.__name__
 | 
						|
    s = str(type)
 | 
						|
    if value is not None:
 | 
						|
        s = s + ": " + str(value)
 | 
						|
    return s
 | 
						|
 | 
						|
 | 
						|
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
 | 
						|
        self.repr = 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):
 | 
						|
        if dict is self.dict and not force:
 | 
						|
            return
 | 
						|
        subframe = self.subframe
 | 
						|
        frame = self.frame
 | 
						|
        for c in subframe.children.values():
 | 
						|
            c.destroy()
 | 
						|
        self.dict = None
 | 
						|
        if not dict:
 | 
						|
            l = Label(subframe, text="None")
 | 
						|
            l.grid(row=0, column=0)
 | 
						|
        else:
 | 
						|
            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
 | 
						|
        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()
 |