如何在SQLAlchemy中为单表多对多关系的关联表添加额外数据

核心思路 #

单表自关联的多对多关系如果需要存储额外字段，不能用普通的关联表，必须用关联对象模式——也就是把关联关系单独做成一个模型类，通过这个类来维护主表的双向关联和额外字段。

1. 完整模型定义 #

from sqlalchemy import Column, Integer, String, Float, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship, sessionmaker
from sqlalchemy import create_engine

Base = declarative_base()

# 关联对象类：维护父子节点的关联关系和额外字段
class NodeAssociation(Base):
    __tablename__ = 'node_association'
    # 复合主键：同一对父子节点只能有一条关联记录
    parent_id = Column(Integer, ForeignKey('node.id'), primary_key=True)
    child_id = Column(Integer, ForeignKey('node.id'), primary_key=True)
    factor = Column(Float, nullable=False)  # 自定义额外字段

    # 双向关联到Node类，必须指定外键避免混淆
    parent = relationship("Node", back_populates="child_links", foreign_keys=[parent_id])
    child = relationship("Node", back_populates="parent_links", foreign_keys=[child_id])

# 主节点类
class Node(Base):
    __tablename__ = 'node'
    id = Column(Integer, primary_key=True)
    name = Column(String(50), nullable=False)

    # 关联到关联对象：作为父节点时的所有关联记录
    child_links = relationship("NodeAssociation", back_populates="parent", foreign_keys=[NodeAssociation.parent_id])
    # 关联到关联对象：作为子节点时的所有关联记录
    parent_links = relationship("NodeAssociation", back_populates="child", foreign_keys=[NodeAssociation.child_id])

    # 快捷属性：直接获取(子节点, factor)列表
    @property
    def children(self):
        return [(link.child, link.factor) for link in self.child_links]

    # 快捷属性：直接获取(父节点, factor)列表
    @property
    def parents(self):
        return [(link.parent, link.factor) for link in self.parent_links]

    # 快捷方法：添加子节点并设置factor
    def add_child(self, child_node, factor):
        link = NodeAssociation(parent=self, child=child_node, factor=factor)
        self.child_links.append(link)

2. 初始化数据库 #

# 创建SQLite数据库（换成你需要的数据库连接字符串即可）
engine = create_engine('sqlite:///nodes.db')
Base.metadata.create_all(engine)

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

3. 写入数据示例 #

# 创建节点实例
node_a = Node(name="节点A")
node_b = Node(name="节点B")
node_c = Node(name="节点C")

# 方式1：用快捷方法添加关联
node_a.add_child(node_b, 0.7)
node_a.add_child(node_c, 0.4)

# 方式2：直接创建关联对象添加
node_b.add_child(node_c, 0.2)

# 提交到数据库
session.add_all([node_a, node_b, node_c])
session.commit()

4. 读取数据示例 #

# 查询节点A的所有子节点及对应的factor
node_a = session.query(Node).filter_by(name="节点A").first()
print("节点A的子节点：")
for child, factor in node_a.children:
    print(f"- {child.name} | factor: {factor}")

# 查询节点C的所有父节点及对应的factor
node_c = session.query(Node).filter_by(name="节点C").first()
print("\n节点C的父节点：")
for parent, factor in node_c.parents:
    print(f"- {parent.name} | factor: {factor}")

# 直接查询所有关联记录
all_links = session.query(NodeAssociation).all()
print("\n所有关联关系：")
for link in all_links:
    print(f"{link.parent.name} → {link.child.name} : {link.factor}")

关键注意点 #

• 关联对象类必须用复合主键（parent_id + child_id），避免同一对父子节点生成多条重复关联。
• 定义关系时必须通过foreign_keys明确指定关联的外键，否则SQLAlchemy会无法区分自关联的两个方向。
• 快捷属性children/parents是为了方便使用，你也可以直接操作child_links/parent_links来访问关联对象本身（比如修改factor值）。

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