protocol.py

import json

BUFFER_SIZE = 4096

def send_json(sock, data):
    try:
        message = json.dumps(data) + "\n"
        sock.sendall(message.encode())
    except Exception as e:
        print("Send error:", e)


def recv_json(sock, buffer):
    try:
        chunk = sock.recv(BUFFER_SIZE).decode()
        if not chunk:
            return [], buffer

        buffer += chunk
        messages = []

        while "\n" in buffer:
            message, buffer = buffer.split("\n", 1)
            try:
                messages.append(json.loads(message))
            except json.JSONDecodeError:
                print("Invalid JSON skipped")

        return messages, buffer

    except Exception as e:
        print("Receive error:", e)
        return [], buffer