Quizz Fever networking

network.py

def messageToBytes(type, body=None):
    data = {'type': type}
    if body is not None:
        data['body'] = body
    try:
        return json.dumps(data).encode('ascii') + b"\n"
    except ValueError:
        raise ValueError("message could not be JSON-encoded")
    except UnicodeEncodeError:
        raise UnicodeEncodeError("JSON could not be converted to bytes")


def bytesToMessage(msg):
    try:
        data = json.loads(msg.decode('ascii'))
        return data['type'], data.get('body', None)
    except UnicodeDecodeError:
        raise UnicodeEncodeError("Incoming message is not ASCII-encoded")
    except ValueError:
        raise ValueError("Incoming message is not valid JSON")
    except KeyError:
        raise ValueError("Incoming message is malformed")