如何基于Python自定义类实例通用创建Py2neo的Node节点

核心实现思路 #

要实现通用转换，只需要解决两个核心问题：

• 提取实例所属类的类名作为Node的标签
• 提取实例的所有公开属性作为Node的键值对

基础通用转换函数 #

from py2neo.data import Node

def obj_to_node(obj, exclude_attrs=None):
    # 处理要排除的属性，默认不过滤
    exclude = exclude_attrs if exclude_attrs is not None else []
    # 获取节点标签：取类名作为标签
    node_label = type(obj).__name__
    # 提取实例属性，过滤私有属性和指定要排除的属性
    node_props = {
        k: v for k, v in obj.__dict__.items()
        if k not in exclude and not k.startswith('_')
    }
    return Node(node_label, **node_props)

使用示例 #

对应你给出的Person类，调用方式如下：

# 业务类定义
class Person:
    def __init__(self, name, surname):
        self.name = name
        self.surname = surname

# 实例化业务对象
p = Person(name="John", surname="Doe")
# 转换为Py2neo Node
person_node = obj_to_node(p)

转换后的person_node完全等价于你需要的Node("Person", name="John", surname="Doe")，且该函数可适配任意自定义类，不需要单独修改逻辑。

进阶适配方案 #

如果需要支持多标签、自定义标签的场景，可以给业务类增加约定类属性，修改转换函数即可实现更灵活的适配：

def obj_to_node(obj, exclude_attrs=None):
    exclude = exclude_attrs if exclude_attrs is not None else []
    # 优先取类自定义的节点标签，没有则默认用类名作为单标签
    if hasattr(type(obj), '__node_labels__'):
        labels = type(obj).__node_labels__
    else:
        labels = [type(obj).__name__]
    node_props = {
        k: v for k, v in obj.__dict__.items()
        if k not in exclude and not k.startswith('_')
    }
    return Node(*labels, **node_props)

# 业务类自定义多标签示例
class Person:
    __node_labels__ = ["Person", "Human"]
    def __init__(self, name, surname):
        self.name = name
        self.surname = surname

转换后得到的节点会同时带有Person和Human两个标签，更贴合Neo4j的建模习惯。

注意事项 #

• 如果实例属性包含Neo4j不支持的类型（比如自定义类实例、函数等），需要提前把这类属性加入exclude_attrs排除，或者提前做类型转换
• 如果你用的是dataclass定义的业务类，也可以直接用dataclasses.asdict(obj)替代obj.__dict__来提取属性，适配性更好

内容的提问来源于stack exchange，提问作者Michal Cikatricis