GraphElementEncoder.py

class GraphElementEncoder(json.JSONEncoder):
    def default(self,obj):
        if isinstance(obj, graph.Node):
            return {
                "element_id": obj.element_id,
                "labels": list(obj.labels),
                "properties": {k:v for k,v in obj.items()}
            }
        elif isinstance(obj, graph.Relationship):
            return {
                "element_id": obj.element_id,
                "type": obj.type,
                "properties": {k:v for k,v in obj.items()},
                "start_element_id": obj.start_node.element_id,
                "end_element_id": obj.end_node.element_id
            }
        else:
            return super().default(obj)