Source code for rfc7464.impl

# Copyright (c) AUTHORS
# See LICENSE for details.

"""
rfc7464 -- streaming JSON
"""

import json

DELIMIT_TYPES = (type(u''), list, dict)


def emit(thing):
    """emit an RFC 7464 record

    :param thing: something that can be JSON-serialized
    :returns: RFC 7464 record (something that can be written to a stream)
    """
    return b'\x1e' + json.dumps(thing).encode('utf-8') + b'\n'


class Parser(object):

    buffer = b''

    def receive(self, data):
        begin = 0
        if self.buffer != b'':
            self.buffer, data = b'', self.buffer + data
        while begin < len(data):
            begin = data.find(b'\x1e', begin)
            if begin == -1:
                return
            max_end = data.find(b'\x1e', begin + 1)
            if max_end == -1:
                max_end = len(data)
            end = data.find(b'\n', begin, max_end)
            if end == -1:
                if max_end == len(data):
                    self.buffer = data[begin:]
                    break
                end = max_end
            try:
                val = json.loads(data[begin + 1:end].decode('utf-8'))
                if data[end:end+1] == b'\n' or isinstance(val, DELIMIT_TYPES):
                    yield val
            except ValueError:
                pass
            begin = end