Python code coverage for Lib/idlelib/StackViewer.py

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