typoenforcer.py

#!/usr/bin/python
# -*- coding:utf-8; -*-

import sys, os
import Tkinter as tk, tkFileDialog, tkFont
import pykey

LONG_TITLE = 'TypoEnforcer'
SHORT_TITLE = 'TE: {0}'

class Application(tk.Frame):
    def __init__(self, master=None):
        tk.Frame.__init__(self, master)
        self.grid(sticky='nesw')
        self.createWidgets()
        self.outputHandlers = [self.debugOutputHandler,
                               self.windowOutputHandler]
        self.master.title(LONG_TITLE)
        self.target = None
        self.filename = None

    def set_text(self, *lines):
        self.theText.delete(0, tk.END)
        for line in lines:
            self.theText.insert(tk.END, line)
        self.gotoLine(0)

    def newButton(self, row, column, text, command):
        button = tk.Button(self, text=text, command=command)
        button.grid(row=row, column=column)

    def createWidgets(self):
        top = self.winfo_toplevel()
        top.rowconfigure(0, weight=1)
        top.columnconfigure(0, weight=1)
        self.rowconfigure(1, weight=1)
        self.columnconfigure(0, weight=1)

        self.newButton(0, 0, 'Quit', self.quit)
        self.newButton(0, 1, 'Load file', self.fileHandler)
        self.newButton(0, 2, 'Set target', self.targetHandler)
        self.newButton(0, 3, 'Bigger font', self.biggerfontHandler)
        self.newButton(0, 4, 'Smaller font', self.smallerfontHandler)
        self.master.bind('=', self.biggerfontHandler)
        self.master.bind('-', self.smallerfontHandler)

        self.textFont = tkFont.Font(size=15)

        self.theText = tk.Listbox(self, height=50, width=72,
                                  selectmode=tk.EXTENDED, font=self.textFont)
        self.theText.bind('<KeyPress-Return>', self.keyHandler)
        self.theText.focus_set()
        self.theText.grid(sticky='nesw', columnspan=5)

    def debugOutputHandler(self, *lines):
        for line in lines:
            print 'output ->', line

    def windowOutputHandler(self, *lines):
        if self.target is None:
            return
        for line in lines:
            print 'sending line:', line
            pykey.send_string(self.target, line + '\n')
            pykey.display.sync()

    def gotoLine(self, index):
        self.theText.see(index)
        self.theText.activate(index)
        self.theText.selection_set(index)

    def keyHandler(self, event):
        print 'got:', event
        selected = [int(index) for index in self.theText.curselection()]
        print 'selected:', selected
        if not selected:
            return
        first, last = min(selected), max(selected)
        lines = self.theText.get(first, last)

        # fix for get(0, 0) returning justs a single string
        if isinstance(lines, basestring):
            lines = [lines]
        for handler in self.outputHandlers:
            handler(*lines)
        self.theText.selection_clear(first, last)
        self.gotoLine(last + 1)

    def fileHandler(self, filename=None):
        if filename is None:
            options = {}
            if self.filename:
                options['initialdir'] = os.path.dirname(self.filename)
            filename = tkFileDialog.askopenfilename(**options)
            if not filename:
                print 'file selection cancelled'
                return

        self.filename = filename

        try:
            input = open(filename).read()
        except IOError as e:
            print 'file input failed', e
            self.bell()
            return

        self.master.title(SHORT_TITLE.format(filename))
        self.set_text(*input.split('\n'))
        print 'loaded file', filename

    def targetHandler(self, windowid=None):
        self.target = pykey.get_window(windowid)
        print 'target set to', self.target

    def changeFont(self, increase):
        font_size = self.textFont.cget("size")
        font_size = font_size + 1 if increase else font_size - 1
        self.textFont.config(size=font_size)

    def biggerfontHandler(self, event=None):
        self.changeFont(True)

    def smallerfontHandler(self, event=None):
        self.changeFont(False)

app = Application()

filename = sys.argv[1] if len(sys.argv)==2 else None
app.fileHandler(filename=filename)
app.targetHandler(windowid=None)

app.mainloop()