root/projects/WinDictator/trunk/test/mockxnee.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.

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


class KeystrokeReporterProtocol(basic.LineReceiver):
    """
    A protocol for a mock I{xnee} keystroke reporter
    """
    delimiter = '\n'
    
    def connectionMade(self):
        pass
    
    def lineReceived(self, line):
        """
        Process a 'back-channel' command to send a line of mock keystroke
        reporting.
        """
        cmd, keycode = line.split()
        fields = [0]
        if cmd == 'd':
            fields.append(2)
        elif cmd == 'u':
            fields.append(3)
        else:
            raise Exception("Command must be 'd' or 'u'!")
        fields.extend([0,0,0])
        fields.append(keycode)
        fields.append(0)
        fields.append(self.fakeTime())
        reportLine = ','.join([str(x) for x in fields])
        self.transport.write(reportLine + "\n")

    def fakeTime(self):
        """
        Returns a faked time for the report that is sort of analogous to what
        xnee generates.
        """
        timeValue = divmod(int(500*time.time()),2**32)[1]
        return "%d" % timeValue

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


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