root/projects/WinDictator/trunk/test/mocktypist.py

#!/usr/bin/python
# Copyright (c) 2001-2004 Twisted Matrix Laboratories.
# See LICENSE for details.
#
# Modified by Ed Suominen 2006 for use with WinDictator unit tests.

from twisted.internet import stdio, reactor
from twisted.protocols import basic


class CommandProcessorProtocol(basic.LineReceiver):
    """
    A protocol for a mock I{typist} command processor
    """
    delimiter = '\n'
    
    def connectionMade(self):
        self.responseDelay = 0.05
        self.pressed = {}
        self.keycodes = {
            'a':38,
            'b':56,
            'c':54,
            '1':10,
            '2':11,
            '3':12,
            'slow':1,
            'Shift_L':50,
            'Return':36
            }

    def complain(self, *args):
        if len(args) > 1:
            msg = args[0] % args[1:]
        else:
            msg = args[0]
        self.sendLine("ERROR: %s" % msg)
        raise Exception(msg)
    
    def lineReceived(self, line):
        """
        Process the command and keysym by returning the keycode and complaining
        if more than two keys are pressed (not including shift) at a time.
        """
        cmd, keySym = line.split()
        if cmd not in ('d', 'u', 'c'):
            self.complain("Improper command '%s'", cmd)
        else:
            keycode = self.keycodes[keySym]
            reactor.callLater(
                self.responseDelay, self.transport.write, "%d\n" % keycode)
        if cmd == 'd':
            self.pressed[keySym] = (keySym != "Shift_L")
            if sum(self.pressed.values()) > 1:
                self.complain(
                    "Trying to press keys %s together at once!",
                    ', '.join(self.pressed.keys()))
                self.pressed.clear()
        elif cmd == 'u':
            if keySym not in self.pressed:
                self.complain(
                    "Tried releasing a key that hadn't been pressed!")
            else:
                del self.pressed[keySym]

    def connectionLost(self, reason):
        """
        I'm not needed anymore...sniff...
        """
        reactor.stop()


if __name__ == '__main__':
    stdio.StandardIO(CommandProcessorProtocol())
    reactor.run()