Tools/idle/StackViewer.py

   1 import os
   2 import sys
   3 import linecache
   4
   5 from TreeWidget import TreeNode, TreeItem, ScrolledCanvas
   6 from ObjectBrowser import ObjectTreeItem, make_objecttreeitem
   7 from OldStackViewer import StackViewer, NamespaceViewer
   8
   9 def StackBrowser(root, flist=None, tb=None, top=None):
  10     if top is None:
  11         from Tkinter import Toplevel
  12         top = Toplevel(root)
  13     sc = ScrolledCanvas(top, bg="white", highlightthickness=0)
  14     sc.frame.pack(expand=1, fill="both")
  15     item = StackTreeItem(flist, tb)
  16     node = TreeNode(sc.canvas, None, item)
  17     node.expand()
  18
  19 class StackTreeItem(TreeItem):
  20
  21     def __init__(self, flist=None, tb=None):
  22         self.flist = flist
  23         self.stack = get_stack(tb)
  24         self.text = get_exception()
  25
  26     def GetText(self):
  27         return self.text
  28
  29     def GetSubList(self):
  30         sublist = []
  31         for info in self.stack:
  32             item = FrameTreeItem(info, self.flist)
  33             sublist.append(item)
  34         return sublist
  35
  36 class FrameTreeItem(TreeItem):
  37
  38     def __init__(self, info, flist):
  39         self.info = info
  40         self.flist = flist
  41
  42     def GetText(self):
  43         frame, lineno = self.info
  44         try:
  45             modname = frame.f_globals["__name__"]
  46         except:
  47             modname = "?"
  48         code = frame.f_code
  49         filename = code.co_filename
  50         funcname = code.co_name
  51         sourceline = linecache.getline(filename, lineno)
  52         sourceline = sourceline.strip()
  53         if funcname in ("?", "", None):
  54             item = "%s, line %d: %s" % (modname, lineno, sourceline)
  55         else:
  56             item = "%s.%s(...), line %d: %s" % (modname, funcname,
  57                                              lineno, sourceline)
  58 ##        if i == index:
  59 ##            item = "> " + item
  60         return item
  61
  62     def GetSubList(self):
  63         frame, lineno = self.info
  64         sublist = []
  65         if frame.f_globals is not frame.f_locals:
  66             item = VariablesTreeItem("<locals>", frame.f_locals, self.flist)
  67             sublist.append(item)
  68         item = VariablesTreeItem("<globals>", frame.f_globals, self.flist)
  69         sublist.append(item)
  70         return sublist
  71
  72     def OnDoubleClick(self):
  73         if self.flist:
  74             frame, lineno = self.info
  75             filename = frame.f_code.co_filename
  76             if os.path.isfile(filename):
  77                 self.flist.gotofileline(filename, lineno)
  78
  79 class VariablesTreeItem(ObjectTreeItem):
  80
  81     def GetText(self):
  82         return self.labeltext
  83
  84     def GetLabelText(self):
  85         return None
  86
  87     def IsExpandable(self):
  88         return len(self.object) > 0
  89
  90     def keys(self):
  91         return self.object.keys()
  92
  93     def GetSubList(self):
  94         sublist = []
  95         for key in self.keys():
  96             try:
  97                 value = self.object[key]
  98             except KeyError:
  99                 continue
 100             def setfunction(value, key=key, object=self.object):
 101                 object[key] = value
 102             item = make_objecttreeitem(key + " =", value, setfunction)
 103             sublist.append(item)
 104         return sublist
 105
 106 def get_stack(t=None, f=None):
 107     if t is None:
 108         t = sys.last_traceback
 109     stack = []
 110     if t and t.tb_frame is f:
 111         t = t.tb_next
 112     while f is not None:
 113         stack.append((f, f.f_lineno))
 114         if f is self.botframe:
 115             break
 116         f = f.f_back
 117     stack.reverse()
 118     while t is not None:
 119         stack.append((t.tb_frame, t.tb_lineno))
 120         t = t.tb_next
 121     return stack
 122
 123 def get_exception(type=None, value=None):
 124     if type is None:
 125         type = sys.last_type
 126         value = sys.last_value
 127     if hasattr(type, "__name__"):
 128         type = type.__name__
 129     s = str(type)
 130     if value is not None:
 131         s = s + ": " + str(value)
 132     return s
 133
 134 def _test():
 135     try:
 136         import testcode
 137         reload(testcode)
 138     except:
 139         sys.last_type, sys.last_value, sys.last_traceback = sys.exc_info()
 140     from Tkinter import Tk
 141     root = Tk()
 142     StackBrowser(None, top=root)
 143     root.mainloop()
 144
 145 if __name__ == "__main__":
 146     _test()