OutputWindow.py 4.09 KB
Newer Older
1
from tkinter import *
2
from idlelib.EditorWindow import EditorWindow
David Scherer's avatar
David Scherer committed
3
import re
4
import tkinter.messagebox as tkMessageBox
5
from idlelib import IOBinding
David Scherer's avatar
David Scherer committed
6

7
class OutputWindow(EditorWindow):
David Scherer's avatar
David Scherer committed
8

9
    """An editor window that can serve as an output file.
David Scherer's avatar
David Scherer committed
10

11 12
    Also the future base class for the Python shell window.
    This class has no input facilities.
David Scherer's avatar
David Scherer committed
13 14
    """

15
    def __init__(self, *args):
16
        EditorWindow.__init__(self, *args)
David Scherer's avatar
David Scherer committed
17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
        self.text.bind("<<goto-file-line>>", self.goto_file_line)

    # Customize EditorWindow

    def ispythonsource(self, filename):
        # No colorization needed
        return 0

    def short_title(self):
        return "Output"

    def maybesave(self):
        # Override base class method -- don't ask any questions
        if self.get_saved():
            return "yes"
        else:
            return "no"

    # Act as output file

37
    def write(self, s, tags=(), mark="insert"):
38
        if isinstance(s, (bytes, bytes)):
39
            s = s.decode(IOBinding.encoding, "replace")
40
        self.text.insert(mark, s, tags)
David Scherer's avatar
David Scherer committed
41 42 43
        self.text.see(mark)
        self.text.update()

44 45 46
    def writelines(self, lines):
        for line in lines:
            self.write(line)
David Scherer's avatar
David Scherer committed
47 48 49 50 51 52 53 54 55 56 57

    def flush(self):
        pass

    # Our own right-button menu

    rmenu_specs = [
        ("Go to file/line", "<<goto-file-line>>"),
    ]

    file_line_pats = [
58
        # order of patterns matters
David Scherer's avatar
David Scherer committed
59 60
        r'file "([^"]*)", line (\d+)',
        r'([^\s]+)\((\d+)\)',
61 62 63
        r'^(\s*\S.*?):\s*(\d+):',  # Win filename, maybe starting with spaces
        r'([^\s]+):\s*(\d+):',     # filename or path, ltrim
        r'^\s*(\S.*?):\s*(\d+):',  # Win abs path with embedded spaces, ltrim
David Scherer's avatar
David Scherer committed
64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
    ]

    file_line_progs = None

    def goto_file_line(self, event=None):
        if self.file_line_progs is None:
            l = []
            for pat in self.file_line_pats:
                l.append(re.compile(pat, re.IGNORECASE))
            self.file_line_progs = l
        # x, y = self.event.x, self.event.y
        # self.text.mark_set("insert", "@%d,%d" % (x, y))
        line = self.text.get("insert linestart", "insert lineend")
        result = self._file_line_helper(line)
        if not result:
            # Try the previous line.  This is handy e.g. in tracebacks,
            # where you tend to right-click on the displayed source line
            line = self.text.get("insert -1line linestart",
                                 "insert -1line lineend")
            result = self._file_line_helper(line)
            if not result:
                tkMessageBox.showerror(
                    "No special line",
                    "The line you point at doesn't look like "
                    "a valid file name followed by a line number.",
                    master=self.text)
                return
        filename, lineno = result
92
        edit = self.flist.open(filename)
David Scherer's avatar
David Scherer committed
93 94 95 96
        edit.gotoline(lineno)

    def _file_line_helper(self, line):
        for prog in self.file_line_progs:
97 98 99 100 101 102 103 104 105
            match = prog.search(line)
            if match:
                filename, lineno = match.group(1, 2)
                try:
                    f = open(filename, "r")
                    f.close()
                    break
                except IOError:
                    continue
David Scherer's avatar
David Scherer committed
106 107 108 109 110 111 112
        else:
            return None
        try:
            return filename, int(lineno)
        except TypeError:
            return None

113
# These classes are currently not used but might come in handy
David Scherer's avatar
David Scherer committed
114 115 116 117 118 119

class OnDemandOutputWindow:

    tagdefs = {
        # XXX Should use IdlePrefs.ColorPrefs
        "stdout":  {"foreground": "blue"},
120 121 122
        "stderr":  {"foreground": "#007700"},
    }

David Scherer's avatar
David Scherer committed
123 124 125
    def __init__(self, flist):
        self.flist = flist
        self.owin = None
126 127 128

    def write(self, s, tags, mark):
        if not self.owin:
David Scherer's avatar
David Scherer committed
129 130 131 132
            self.setup()
        self.owin.write(s, tags, mark)

    def setup(self):
133
        self.owin = owin = OutputWindow(self.flist)
David Scherer's avatar
David Scherer committed
134 135 136
        text = owin.text
        for tag, cnf in self.tagdefs.items():
            if cnf:
137
                text.tag_configure(tag, **cnf)
David Scherer's avatar
David Scherer committed
138
        text.tag_raise('sel')
139
        self.write = self.owin.write