pos.py

from .mitracomm import Streamer as BaseStreamer


class Streamer(BaseStreamer):
    def get_size(self, raw):  # Override
        a, b = raw
        b = b * 256
        return b + a

    def set(self, raw):  # Override
        size = len(raw)
        b = size % 256
        a = int(size / 256)
        header = bytes([b]) + bytes([a])
        return header + raw