home *** CD-ROM | disk | FTP | other *** search
/ PC World 2001 April / PCWorld_2001-04_cd.bin / Software / TemaCD / webclean / !!!python!!! / BeOpen-Python-2.0.exe / STACKVIEWER.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  3.8 KB  |  136 lines
  1. import string
  2. from Tkinter import *
  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, stack=None):
  10.     top = Toplevel(root)
  11.     sc = ScrolledCanvas(top, bg="white", highlightthickness=0)
  12.     sc.frame.pack(expand=1, fill="both")
  13.     item = StackTreeItem(flist)
  14.     node = TreeNode(sc.canvas, None, item)
  15.     node.expand()
  16.  
  17. class StackTreeItem(TreeItem):
  18.  
  19.     def __init__(self, flist=None):
  20.         self.flist = flist
  21.         self.stack = get_stack()
  22.         self.text = get_exception()
  23.  
  24.     def GetText(self):
  25.         return self.text
  26.  
  27.     def GetSubList(self):
  28.         sublist = []
  29.         for info in self.stack:
  30.             item = FrameTreeItem(info, self.flist)
  31.             sublist.append(item)
  32.         return sublist
  33.  
  34. class FrameTreeItem(TreeItem):
  35.  
  36.     def __init__(self, info, flist):
  37.         self.info = info
  38.         self.flist = flist
  39.  
  40.     def GetText(self):
  41.         frame, lineno = self.info
  42.         try:
  43.             modname = frame.f_globals["__name__"]
  44.         except:
  45.             modname = "?"
  46.         code = frame.f_code
  47.         filename = code.co_filename
  48.         funcname = code.co_name
  49.         sourceline = linecache.getline(filename, lineno)
  50.         sourceline = string.strip(sourceline)
  51.         if funcname in ("?", "", None):
  52.             item = "%s, line %d: %s" % (modname, lineno, sourceline)
  53.         else:
  54.             item = "%s.%s(...), line %d: %s" % (modname, funcname,
  55.                                              lineno, sourceline)
  56. ##        if i == index:
  57. ##            item = "> " + item
  58.         return item
  59.  
  60.     def GetSubList(self):
  61.         frame, lineno = self.info
  62.         sublist = []
  63.         if frame.f_globals is not frame.f_locals:
  64.             item = VariablesTreeItem("<locals>", frame.f_locals, self.flist)
  65.             sublist.append(item)
  66.         item = VariablesTreeItem("<globals>", frame.f_globals, self.flist)
  67.         sublist.append(item)
  68.         return sublist
  69.  
  70.     def OnDoubleClick(self):
  71.         if self.flist:
  72.             frame, lineno = self.info
  73.             filename = frame.f_code.co_filename
  74.             edit = self.flist.open(filename)
  75.             edit.gotoline(lineno)
  76.  
  77. class VariablesTreeItem(ObjectTreeItem):
  78.  
  79.     def GetText(self):
  80.         return self.labeltext
  81.  
  82.     def GetLabelText(self):
  83.         return None
  84.  
  85.     def IsExpandable(self):
  86.         return len(self.object) > 0
  87.  
  88.     def keys(self):
  89.         return self.object.keys()
  90.  
  91.     def GetSubList(self):
  92.         sublist = []
  93.         for key in self.keys():
  94.             try:
  95.                 value = self.object[key]
  96.             except KeyError:
  97.                 continue
  98.             def setfunction(value, key=key, object=self.object):
  99.                 object[key] = value
  100.             item = make_objecttreeitem(key + " =", value, setfunction)
  101.             sublist.append(item)
  102.         return sublist
  103.         
  104. def get_stack(t=None, f=None):
  105.     if t is None:
  106.         t = sys.last_traceback
  107.     stack = []
  108.     if t and t.tb_frame is f:
  109.         t = t.tb_next
  110.     while f is not None:
  111.         stack.append((f, f.f_lineno))
  112.         if f is self.botframe:
  113.             break
  114.         f = f.f_back
  115.     stack.reverse()
  116.     while t is not None:
  117.         stack.append((t.tb_frame, t.tb_lineno))
  118.         t = t.tb_next
  119.     return stack
  120.  
  121. def get_exception(type=None, value=None):
  122.     if type is None:
  123.         type = sys.last_type
  124.         value = sys.last_value
  125.     if hasattr(type, "__name__"):
  126.         type = type.__name__
  127.     s = str(type)
  128.     if value is not None:
  129.         s = s + ": " + str(value)
  130.     return s
  131.  
  132. if __name__ == "__main__":
  133.     root = Tk()
  134.     root.withdraw()
  135.     StackBrowser(root)
  136.