Mercurial > repos > rliterman > csp2

annotate CSP2/CSP2_env/env-d9b9114564458d9d-741b3de822f2aaca6c6caa4325c4afce/lib/python3.8/idlelib/stackviewer.py @ 68:5028fdace37b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload commit 2e9511a184a1ca667c7be0c6321a36dc4e3d116d
author jpayne
date Tue, 18 Mar 2025 16:23:26 -0400
parents
children
rev   line source
jpayne@68 1 import linecache
jpayne@68 2 import os
jpayne@68 3 import sys
jpayne@68 4
jpayne@68 5 import tkinter as tk
jpayne@68 6
jpayne@68 7 from idlelib.debugobj import ObjectTreeItem, make_objecttreeitem
jpayne@68 8 from idlelib.tree import TreeNode, TreeItem, ScrolledCanvas
jpayne@68 9
jpayne@68 10 def StackBrowser(root, flist=None, tb=None, top=None):
jpayne@68 11 global sc, item, node # For testing.
jpayne@68 12 if top is None:
jpayne@68 13 top = tk.Toplevel(root)
jpayne@68 14 sc = ScrolledCanvas(top, bg="white", highlightthickness=0)
jpayne@68 15 sc.frame.pack(expand=1, fill="both")
jpayne@68 16 item = StackTreeItem(flist, tb)
jpayne@68 17 node = TreeNode(sc.canvas, None, item)
jpayne@68 18 node.expand()
jpayne@68 19
jpayne@68 20
jpayne@68 21 class StackTreeItem(TreeItem):
jpayne@68 22
jpayne@68 23 def __init__(self, flist=None, tb=None):
jpayne@68 24 self.flist = flist
jpayne@68 25 self.stack = self.get_stack(tb)
jpayne@68 26 self.text = self.get_exception()
jpayne@68 27
jpayne@68 28 def get_stack(self, tb):
jpayne@68 29 if tb is None:
jpayne@68 30 tb = sys.last_traceback
jpayne@68 31 stack = []
jpayne@68 32 if tb and tb.tb_frame is None:
jpayne@68 33 tb = tb.tb_next
jpayne@68 34 while tb is not None:
jpayne@68 35 stack.append((tb.tb_frame, tb.tb_lineno))
jpayne@68 36 tb = tb.tb_next
jpayne@68 37 return stack
jpayne@68 38
jpayne@68 39 def get_exception(self):
jpayne@68 40 type = sys.last_type
jpayne@68 41 value = sys.last_value
jpayne@68 42 if hasattr(type, "__name__"):
jpayne@68 43 type = type.__name__
jpayne@68 44 s = str(type)
jpayne@68 45 if value is not None:
jpayne@68 46 s = s + ": " + str(value)
jpayne@68 47 return s
jpayne@68 48
jpayne@68 49 def GetText(self):
jpayne@68 50 return self.text
jpayne@68 51
jpayne@68 52 def GetSubList(self):
jpayne@68 53 sublist = []
jpayne@68 54 for info in self.stack:
jpayne@68 55 item = FrameTreeItem(info, self.flist)
jpayne@68 56 sublist.append(item)
jpayne@68 57 return sublist
jpayne@68 58
jpayne@68 59
jpayne@68 60 class FrameTreeItem(TreeItem):
jpayne@68 61
jpayne@68 62 def __init__(self, info, flist):
jpayne@68 63 self.info = info
jpayne@68 64 self.flist = flist
jpayne@68 65
jpayne@68 66 def GetText(self):
jpayne@68 67 frame, lineno = self.info
jpayne@68 68 try:
jpayne@68 69 modname = frame.f_globals["__name__"]
jpayne@68 70 except:
jpayne@68 71 modname = "?"
jpayne@68 72 code = frame.f_code
jpayne@68 73 filename = code.co_filename
jpayne@68 74 funcname = code.co_name
jpayne@68 75 sourceline = linecache.getline(filename, lineno)
jpayne@68 76 sourceline = sourceline.strip()
jpayne@68 77 if funcname in ("?", "", None):
jpayne@68 78 item = "%s, line %d: %s" % (modname, lineno, sourceline)
jpayne@68 79 else:
jpayne@68 80 item = "%s.%s(...), line %d: %s" % (modname, funcname,
jpayne@68 81 lineno, sourceline)
jpayne@68 82 return item
jpayne@68 83
jpayne@68 84 def GetSubList(self):
jpayne@68 85 frame, lineno = self.info
jpayne@68 86 sublist = []
jpayne@68 87 if frame.f_globals is not frame.f_locals:
jpayne@68 88 item = VariablesTreeItem("<locals>", frame.f_locals, self.flist)
jpayne@68 89 sublist.append(item)
jpayne@68 90 item = VariablesTreeItem("<globals>", frame.f_globals, self.flist)
jpayne@68 91 sublist.append(item)
jpayne@68 92 return sublist
jpayne@68 93
jpayne@68 94 def OnDoubleClick(self):
jpayne@68 95 if self.flist:
jpayne@68 96 frame, lineno = self.info
jpayne@68 97 filename = frame.f_code.co_filename
jpayne@68 98 if os.path.isfile(filename):
jpayne@68 99 self.flist.gotofileline(filename, lineno)
jpayne@68 100
jpayne@68 101
jpayne@68 102 class VariablesTreeItem(ObjectTreeItem):
jpayne@68 103
jpayne@68 104 def GetText(self):
jpayne@68 105 return self.labeltext
jpayne@68 106
jpayne@68 107 def GetLabelText(self):
jpayne@68 108 return None
jpayne@68 109
jpayne@68 110 def IsExpandable(self):
jpayne@68 111 return len(self.object) > 0
jpayne@68 112
jpayne@68 113 def GetSubList(self):
jpayne@68 114 sublist = []
jpayne@68 115 for key in self.object.keys():
jpayne@68 116 try:
jpayne@68 117 value = self.object[key]
jpayne@68 118 except KeyError:
jpayne@68 119 continue
jpayne@68 120 def setfunction(value, key=key, object=self.object):
jpayne@68 121 object[key] = value
jpayne@68 122 item = make_objecttreeitem(key + " =", value, setfunction)
jpayne@68 123 sublist.append(item)
jpayne@68 124 return sublist
jpayne@68 125
jpayne@68 126
jpayne@68 127 def _stack_viewer(parent): # htest #
jpayne@68 128 from idlelib.pyshell import PyShellFileList
jpayne@68 129 top = tk.Toplevel(parent)
jpayne@68 130 top.title("Test StackViewer")
jpayne@68 131 x, y = map(int, parent.geometry().split('+')[1:])
jpayne@68 132 top.geometry("+%d+%d" % (x + 50, y + 175))
jpayne@68 133 flist = PyShellFileList(top)
jpayne@68 134 try: # to obtain a traceback object
jpayne@68 135 intentional_name_error
jpayne@68 136 except NameError:
jpayne@68 137 exc_type, exc_value, exc_tb = sys.exc_info()
jpayne@68 138 # inject stack trace to sys
jpayne@68 139 sys.last_type = exc_type
jpayne@68 140 sys.last_value = exc_value
jpayne@68 141 sys.last_traceback = exc_tb
jpayne@68 142
jpayne@68 143 StackBrowser(top, flist=flist, top=top, tb=exc_tb)
jpayne@68 144
jpayne@68 145 # restore sys to original state
jpayne@68 146 del sys.last_type
jpayne@68 147 del sys.last_value
jpayne@68 148 del sys.last_traceback
jpayne@68 149
jpayne@68 150 if __name__ == '__main__':
jpayne@68 151 from unittest import main
jpayne@68 152 main('idlelib.idle_test.test_stackviewer', verbosity=2, exit=False)
jpayne@68 153
jpayne@68 154 from idlelib.idle_test.htest import run
jpayne@68 155 run(_stack_viewer)