Tools/idle/Debugger.py

   1 import os
   2 import bdb
   3 import traceback
   4 from Tkinter import *
   5 from WindowList import ListedToplevel
   6
   7 import StackViewer
   8
   9
  10 class Debugger(bdb.Bdb):
  11
  12     interacting = 0
  13
  14     vstack = vsource = vlocals = vglobals = None
  15
  16     def __init__(self, pyshell):
  17         bdb.Bdb.__init__(self)
  18         self.pyshell = pyshell
  19         self.make_gui()
  20
  21     def canonic(self, filename):
  22         # Canonicalize filename -- called by Bdb
  23         return os.path.normcase(os.path.abspath(filename))
  24
  25     def close(self, event=None):
  26         if self.interacting:
  27             self.top.bell()
  28             return
  29         if self.stackviewer:
  30             self.stackviewer.close(); self.stackviewer = None
  31         self.pyshell.close_debugger()
  32         self.top.destroy()
  33
  34     def run(self, *args):
  35         try:
  36             self.interacting = 1
  37             return apply(bdb.Bdb.run, (self,) + args)
  38         finally:
  39             self.interacting = 0
  40
  41     def user_line(self, frame):
  42         self.interaction(frame)
  43
  44     def user_return(self, frame, rv):
  45         # XXX show rv?
  46         ##self.interaction(frame)
  47         pass
  48
  49     def user_exception(self, frame, info):
  50         self.interaction(frame, info)
  51
  52     def make_gui(self):
  53         pyshell = self.pyshell
  54         self.flist = pyshell.flist
  55         self.root = root = pyshell.root
  56         self.top = top =ListedToplevel(root)
  57         self.top.wm_title("Debug Control")
  58         self.top.wm_iconname("Debug")
  59         top.wm_protocol("WM_DELETE_WINDOW", self.close)
  60         self.top.bind("<Escape>", self.close)
  61         #
  62         self.bframe = bframe = Frame(top)
  63         self.bframe.pack(anchor="w")
  64         self.buttons = bl = []
  65         #
  66         self.bcont = b = Button(bframe, text="Go", command=self.cont)
  67         bl.append(b)
  68         self.bstep = b = Button(bframe, text="Step", command=self.step)
  69         bl.append(b)
  70         self.bnext = b = Button(bframe, text="Over", command=self.next)
  71         bl.append(b)
  72         self.bret = b = Button(bframe, text="Out", command=self.ret)
  73         bl.append(b)
  74         self.bret = b = Button(bframe, text="Quit", command=self.quit)
  75         bl.append(b)
  76         #
  77         for b in bl:
  78             b.configure(state="disabled")
  79             b.pack(side="left")
  80         #
  81         self.cframe = cframe = Frame(bframe)
  82         self.cframe.pack(side="left")
  83         #
  84         if not self.vstack:
  85             self.__class__.vstack = BooleanVar(top)
  86             self.vstack.set(1)
  87         self.bstack = Checkbutton(cframe,
  88             text="Stack", command=self.show_stack, variable=self.vstack)
  89         self.bstack.grid(row=0, column=0)
  90         if not self.vsource:
  91             self.__class__.vsource = BooleanVar(top)
  92             ##self.vsource.set(1)
  93         self.bsource = Checkbutton(cframe,
  94             text="Source", command=self.show_source, variable=self.vsource)
  95         self.bsource.grid(row=0, column=1)
  96         if not self.vlocals:
  97             self.__class__.vlocals = BooleanVar(top)
  98             self.vlocals.set(1)
  99         self.blocals = Checkbutton(cframe,
 100             text="Locals", command=self.show_locals, variable=self.vlocals)
 101         self.blocals.grid(row=1, column=0)
 102         if not self.vglobals:
 103             self.__class__.vglobals = BooleanVar(top)
 104             ##self.vglobals.set(1)
 105         self.bglobals = Checkbutton(cframe,
 106             text="Globals", command=self.show_globals, variable=self.vglobals)
 107         self.bglobals.grid(row=1, column=1)
 108         #
 109         self.status = Label(top, anchor="w")
 110         self.status.pack(anchor="w")
 111         self.error = Label(top, anchor="w")
 112         self.error.pack(anchor="w", fill="x")
 113         self.errorbg = self.error.cget("background")
 114         #
 115         self.fstack = Frame(top, height=1)
 116         self.fstack.pack(expand=1, fill="both")
 117         self.flocals = Frame(top)
 118         self.flocals.pack(expand=1, fill="both")
 119         self.fglobals = Frame(top, height=1)
 120         self.fglobals.pack(expand=1, fill="both")
 121         #
 122         if self.vstack.get():
 123             self.show_stack()
 124         if self.vlocals.get():
 125             self.show_locals()
 126         if self.vglobals.get():
 127             self.show_globals()
 128
 129     frame = None
 130
 131     def interaction(self, frame, info=None):
 132         self.frame = frame
 133         code = frame.f_code
 134         file = code.co_filename
 135         base = os.path.basename(file)
 136         lineno = frame.f_lineno
 137         #
 138         message = "%s:%s" % (base, lineno)
 139         if code.co_name != "?":
 140             message = "%s: %s()" % (message, code.co_name)
 141         self.status.configure(text=message)
 142         #
 143         if info:
 144             type, value, tb = info
 145             try:
 146                 m1 = type.__name__
 147             except AttributeError:
 148                 m1 = "%s" % str(type)
 149             if value is not None:
 150                 try:
 151                     m1 = "%s: %s" % (m1, str(value))
 152                 except:
 153                     pass
 154             bg = "yellow"
 155         else:
 156             m1 = ""
 157             tb = None
 158             bg = self.errorbg
 159         self.error.configure(text=m1, background=bg)
 160         #
 161         sv = self.stackviewer
 162         if sv:
 163             stack, i = self.get_stack(self.frame, tb)
 164             sv.load_stack(stack, i)
 165         #
 166         self.show_variables(1)
 167         #
 168         if self.vsource.get():
 169             self.sync_source_line()
 170         #
 171         for b in self.buttons:
 172             b.configure(state="normal")
 173         #
 174         self.top.tkraise()
 175         self.root.mainloop()
 176         #
 177         for b in self.buttons:
 178             b.configure(state="disabled")
 179         self.status.configure(text="")
 180         self.error.configure(text="", background=self.errorbg)
 181         self.frame = None
 182
 183     def sync_source_line(self):
 184         frame = self.frame
 185         if not frame:
 186             return
 187         code = frame.f_code
 188         file = code.co_filename
 189         lineno = frame.f_lineno
 190         if file[:1] + file[-1:] != "<>" and os.path.exists(file):
 191             edit = self.flist.open(file)
 192             if edit:
 193                 edit.gotoline(lineno)
 194
 195     def cont(self):
 196         self.set_continue()
 197         self.root.quit()
 198
 199     def step(self):
 200         self.set_step()
 201         self.root.quit()
 202
 203     def next(self):
 204         self.set_next(self.frame)
 205         self.root.quit()
 206
 207     def ret(self):
 208         self.set_return(self.frame)
 209         self.root.quit()
 210
 211     def quit(self):
 212         self.set_quit()
 213         self.root.quit()
 214
 215     stackviewer = None
 216
 217     def show_stack(self):
 218         if not self.stackviewer and self.vstack.get():
 219             self.stackviewer = sv = StackViewer.StackViewer(
 220                 self.fstack, self.flist, self)
 221             if self.frame:
 222                 stack, i = self.get_stack(self.frame, None)
 223                 sv.load_stack(stack, i)
 224         else:
 225             sv = self.stackviewer
 226             if sv and not self.vstack.get():
 227                 self.stackviewer = None
 228                 sv.close()
 229             self.fstack['height'] = 1
 230
 231     def show_source(self):
 232         if self.vsource.get():
 233             self.sync_source_line()
 234
 235     def show_frame(self, (frame, lineno)):
 236         self.frame = frame
 237         self.show_variables()
 238
 239     localsviewer = None
 240     globalsviewer = None
 241
 242     def show_locals(self):
 243         lv = self.localsviewer
 244         if self.vlocals.get():
 245             if not lv:
 246                 self.localsviewer = StackViewer.NamespaceViewer(
 247                     self.flocals, "Locals")
 248         else:
 249             if lv:
 250                 self.localsviewer = None
 251                 lv.close()
 252                 self.flocals['height'] = 1
 253         self.show_variables()
 254
 255     def show_globals(self):
 256         gv = self.globalsviewer
 257         if self.vglobals.get():
 258             if not gv:
 259                 self.globalsviewer = StackViewer.NamespaceViewer(
 260                     self.fglobals, "Globals")
 261         else:
 262             if gv:
 263                 self.globalsviewer = None
 264                 gv.close()
 265                 self.fglobals['height'] = 1
 266         self.show_variables()
 267
 268     def show_variables(self, force=0):
 269         lv = self.localsviewer
 270         gv = self.globalsviewer
 271         frame = self.frame
 272         if not frame:
 273             ldict = gdict = None
 274         else:
 275             ldict = frame.f_locals
 276             gdict = frame.f_globals
 277             if lv and gv and ldict is gdict:
 278                 ldict = None
 279         if lv:
 280             lv.load_dict(ldict, force)
 281         if gv:
 282             gv.load_dict(gdict, force)
 283
 284     def set_breakpoint_here(self, edit):
 285         text = edit.text
 286         filename = edit.io.filename
 287         if not filename:
 288             text.bell()
 289             return
 290         lineno = int(float(text.index("insert")))
 291         msg = self.set_break(filename, lineno)
 292         if msg:
 293             text.bell()
 294             return
 295         text.tag_add("BREAK", "insert linestart", "insert lineend +1char")
 296
 297     # A literal copy of Bdb.set_break() without the print statement at the end
 298     def set_break(self, filename, lineno, temporary=0, cond = None):
 299         import linecache # Import as late as possible
 300         line = linecache.getline(filename, lineno)
 301         if not line:
 302                 return 'That line does not exist!'
 303         if not self.breaks.has_key(filename):
 304                 self.breaks[filename] = []
 305         list = self.breaks[filename]
 306         if not lineno in list:
 307                 list.append(lineno)
 308         bp = bdb.Breakpoint(filename, lineno, temporary, cond)