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