home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / StackViewer.py < prev    next >
Text File  |  2003-12-30  |  4KB  |  138 lines
  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()
  138.