cpython/Mac/Tools/IDE/Wtraceback.py

import traceback
import sys
import W
import os
import types
from Carbon import List


class TraceBack:

    def __init__(self, title = "Traceback"):
        app = W.getapplication()  # checks if W is properly initialized
        self.title = title
        self.w = None
        self.closed = 1
        self.start = 0
        self.lastwindowtitle = ""
        self.bounds = (360, 298)

    def traceback(self, start = 0, lastwindowtitle = ""):
        try:
            self.lastwindowtitle = lastwindowtitle
            self.start = start
            self.type, self.value, self.tb = sys.exc_info()
            if self.type is not SyntaxError:
                self.show()
                if type(self.type) == types.ClassType:
                    errortext = self.type.__name__
                else:
                    errortext = str(self.type)
                value = str(self.value)
                if self.value and value:
                    errortext = errortext + ": " + value
                self.w.text.set(errortext)
                self.buildtblist()
                self.w.list.set(self.textlist)
                self.w.list.setselection([len(self.textlist) - 1])
                self.w.wid.SelectWindow()
                self.closed = 0
            else:
                self.syntaxerror()
        except:
            traceback.print_exc()

    def syntaxerror(self):
        try:
            value, (filename, lineno, charno, line) = self.value
        except:
            filename = ""
            lineno = None
            value = self.value
        if not filename and self.lastwindowtitle:
            filename = self.lastwindowtitle
        elif not filename:
            filename = "<unknown>"
        if filename and os.path.exists(filename):
            filename = os.path.split(filename)[1]
        if lineno and charno is not None:
            charno = charno - 1
            text = str(value) + '\rFile: "' + str(filename) + '", line ' + str(lineno) + '\r\r' + line[:charno] + "\xa5" + line[charno:-1]
        else:
            text = str(value) + '\rFile: "' + str(filename) + '"'
        self.syntaxdialog = W.ModalDialog((360, 120), "Syntax Error")
        self.syntaxdialog.text = W.TextBox((10, 10, -10, -40), text)
        self.syntaxdialog.cancel = W.Button((-190, -32, 80, 16), "Cancel", self.syntaxclose)
        self.syntaxdialog.edit = W.Button((-100, -32, 80, 16), "Edit", self.syntaxedit)
        self.syntaxdialog.setdefaultbutton(self.syntaxdialog.edit)
        self.syntaxdialog.bind("cmd.", self.syntaxdialog.cancel.push)
        self.syntaxdialog.open()

    def syntaxclose(self):
        self.syntaxdialog.close()
        del self.syntaxdialog

    def syntaxedit(self):
        try:
            value, (filename, lineno, charno, line) = self.value
        except:
            filename = ""
            lineno = None
        if not filename and self.lastwindowtitle:
            filename = self.lastwindowtitle
        elif not filename:
            filename = "<unknown>"
        self.syntaxclose()
        if lineno:
            if charno is None:
                charno = 1
            W.getapplication().openscript(filename, lineno, charno - 1)
        else:
            W.getapplication().openscript(filename)

    def show(self):
        if self.closed:
            self.setupwidgets()
            self.w.open()
        else:
            self.w.wid.ShowWindow()
            self.w.wid.SelectWindow()

    def hide(self):
        if self.closed:
            return
        self.w.close()

    def close(self):
        self.bounds = self.w.getbounds()
        self.closed = 1
        self.type, self.value, self.tb = None, None, None
        self.tblist = None

    def activate(self, onoff):
        if onoff:
            if self.closed:
                self.traceback()
            self.closed = 0
            self.checkbuttons()

    def setupwidgets(self):
        self.w = W.Window(self.bounds, self.title, minsize = (316, 168))
        self.w.text = W.TextBox((10, 10, -10, 30))
        self.w.tbtitle = W.TextBox((10, 40, -10, 10), "Traceback (innermost last):")
        self.w.list = W.TwoLineList((10, 60, -10, -40), callback = self.listhit)

        self.w.editbutton = W.Button((10, -30, 60, 16), "Edit", self.edit)
        self.w.editbutton.enable(0)

        self.w.browselocalsbutton = W.Button((80, -30, 100, 16), "Browse locals\xc9", self.browselocals)
        self.w.browselocalsbutton.enable(0)

        self.w.postmortembutton = W.Button((190, -30, 100, 16), "Post mortem\xc9", self.postmortem)

        self.w.setdefaultbutton(self.w.editbutton)
        self.w.bind("cmdb", self.w.browselocalsbutton.push)
        self.w.bind("<close>", self.close)
        self.w.bind("<activate>", self.activate)

    def buildtblist(self):
        tb = self.tb
        for i in range(self.start):
            if tb.tb_next is None:
                break
            tb = tb.tb_next
        self.tblist = traceback.extract_tb(tb)
        self.textlist = []
        for filename, lineno, func, line in self.tblist:
            tbline = ""
            if os.path.exists(filename):
                filename = os.path.split(filename)[1]
            tbline = 'File "%s", line %r, in %r' % (filename, lineno, func)
            if line:
                tbline = tbline + '\r      ' + line
            self.textlist.append(tbline[:255])

    def edit(self):
        sel = self.w.list.getselection()
        for i in sel:
            filename, lineno, func, line = self.tblist[i]
            W.getapplication().openscript(filename, lineno)

    def browselocals(self):
        sel = self.w.list.getselection()
        for i in sel:
            tb = self.tb
            for j in range(i + self.start):
                tb = tb.tb_next
            self.browse(tb.tb_frame.f_locals)

    def browse(self, object):
        import PyBrowser
        PyBrowser.Browser(object)

    def postmortem(self):
        import PyDebugger
        PyDebugger.postmortem(self.type, self.value, self.tb)

    def listhit(self, isdbl):
        if isdbl:
            self.w.editbutton.push()
        else:
            self.checkbuttons()

    def checkbuttons(self):
        havefile = len(self.w.list.getselection()) > 0
        self.w.editbutton.enable(havefile)
        self.w.browselocalsbutton.enable(havefile)
        self.w.setdefaultbutton(havefile and self.w.editbutton or self.w.postmortembutton)