← Back to branch summary

~ubuntu-branches/ubuntu/karmic/pypy/karmic

« back to all changes in this revision

Viewing changes to lib-python/2.4.1/idlelib/StackViewer.py

  • Committer: Bazaar Package Importer
  • Author(s): Alexandre Fayolle
  • Date: 2007-04-13 09:33:09 UTC
  • Revision ID: james.westby@ubuntu.com-20070413093309-yoojh4jcoocu2krz
Tags: upstream-1.0.0
Import upstream version 1.0.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib-python/2.4.1/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()