RawExtensions.py

from Mouse import Mouse
from TipiVariable import TipiVariable
from TiSocket import TiSocket
from TiUdpSocket import TiUdpSocket
from TlsSocket import TlsSocket
from UserLog import UserLog

import sys


class RawExtensions(object):
    def __init__(self, tipi_io):
        # Extend this list to register new special request handlers
        #
        self.__reg = {
            0x20: Mouse(tipi_io),
            0x21: TipiVariable(tipi_io),
            0x22: TiSocket(tipi_io),
            0x23: TiUdpSocket(tipi_io),
            0x24: TlsSocket(tipi_io),
            0x25: UserLog(tipi_io),
        }

    def handle(self, bytes):
        if not bytes[0] in self.__reg:
            return False
        handler = self.__reg[bytes[0]]

        return handler.handle(bytes)