Source code for wishbone.protocol

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
#  __init__.py
#
#  Copyright 2018 Jelle Smet <development@smetj.net>
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 3 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#  MA 02110-1301, USA.
#
#

from wishbone.error import ProtocolError
import types
from scalpl import Cut


class Decode(object):

    handle_buffer_size = False
    current_size = 0

    def flush(self):

        raise Exception("This decoder has no flush() method implemented.")

    def getBufferSize(self):

        return 0

    def handler(self, data):

        if not self.handle_buffer_size and self.getBufferSize() >= self.buffer_size:
            raise ProtocolError("Buffer full")
        try:
            if data is None:
                return self.flush()
            elif isinstance(data, bytes):
                return self.handleBytes(data)
            elif isinstance(data, str):
                return self.handleString(data)
            elif isinstance(data, int):
                return self.handleInt(data)
            elif isinstance(data, float):
                return self.handleFloat(data)
            elif isinstance(data, dict):
                return self.handleDict(data)
            elif isinstance(data, Cut):
                return self.handleDict(
                    dict(data)
                )
            elif isinstance(data, list):
                return self.handleList(data)
            elif isinstance(data, types.GeneratorType):
                return self.handleGenerator(data)
            elif hasattr(data, "readlines") and callable(data.readlines):
                return self.handleReadLinesMethod(data)
            else:
                raise ProtocolError("%s is not supported by this Decoder." % (type(data)))
        except Exception as err:
            raise ProtocolError(err)

    def handleBytes(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def handleDict(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def handleFloat(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def handleGenerator(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def handleInt(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def handleList(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def handleReadLinesMethod(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def handleString(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def handleUnicode(self, data):
        raise ProtocolError("%s is not supported by this Decoder." % (type(data)))

    def resetBuffer(self):

        raise Exception("Decoder has no resetBuffer() method.")


class Encode(object):

    def handler(self, data):
        try:
            if isinstance(data, bytes):
                return self.handleBytes(data)
            elif isinstance(data, str):
                return self.handleString(data)
            elif isinstance(data, int):
                return self.handleInt(data)
            elif isinstance(data, float):
                return self.handleFloat(data)
            elif isinstance(data, dict):
                return self.handleDict(data)
            elif isinstance(data, list):
                return self.handleList(data)
            elif isinstance(data, Cut):
                return self.handleCut(data)
            else:
                raise ProtocolError("%s is not supported by this Encoder." % (type(data)))
        except Exception as err:
            raise ProtocolError(err)

    def handleBytes(self, data):
        raise ProtocolError("%s is not supported by this Encoder." % (type(data)))

    def handleString(self, data):
        raise ProtocolError("%s is not supported by this Encoder." % (type(data)))

    def handleInt(self, data):
        raise ProtocolError("%s is not supported by this Encoder." % (type(data)))

    def handleFloat(self, data):
        raise ProtocolError("%s is not supported by this Encoder." % (type(data)))

    def handleUnicode(self, data):
        raise ProtocolError("%s is not supported by this Encoder." % (type(data)))

    def handleDict(self, data):
        raise ProtocolError("%s is not supported by this Encoder." % (type(data)))

    def handleList(self, data):
        raise ProtocolError("%s is not supported by this Encoder." % (type(data)))