SqlAlchemy外键向下传播3层

时间:2019-05-27 02:28:32

标签: python sqlalchemy

我正在尝试更新一个简单的3层关系表集。

他们是

  • 父母
  • 孩子
  • GrandChild

该模型的SQLAlchemy代码如下

class Parent(Base):
    __tablename__ = 'parent'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    children = relationship("Child", back_populates="parents")


class Child(Base):
    __tablename__ = 'child'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    parent_id = Column(Integer, ForeignKey('parent.id'))
    parents = relationship("Parent", back_populates="children")
    grandchildren = relationship("GrandChild",
                                back_populates="grandparent",
                                 )

class GrandChild(Base):
    __tablename__ = 'grandchild'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    parent_id = Column(Integer, ForeignKey('parent.id'))
    child_id = Column(Integer, ForeignKey('child.id'))
    grandparent = relationship("Child", back_populates="grandchildren")

插入代码如下所示。...

p3 = Parent(name="P3")
c5 = Child(name="C5")
c6 = Child(name="C6")
gc1 = GrandChild(name="gc1")
gc2 = GrandChild(name="gc2")
gc3 = GrandChild(name="gc3")
gc4 = GrandChild(name="gc4")

p3.children = [c5, c6]
c5.grandchildren = [gc1]
c6.grandchildren = [gc2,  gc3, gc4]

session.add_all([p2, p3])

session.commit()

该记录已添加-并且“父/子”已正确链接-但“孙子”缺少“父”外键。

我一直在寻找添加此功能的正确机制-有人能指出我正确的方向吗?

1 个答案:

答案 0 :(得分:1)

您不会在孙子与父母之间建立关系。
在数据模型中,孙子与父母之间的关系并不隐含;一个孩子的父母不会自动成为所有孩子的孙子的父母。

您必须明确定义该关系,即将其添加到GrandChild

class GrandChild(Base):
    [...]
    parent = relationship("Parent")

,然后在实例上创建关系:

gc1.parent = p3
gc2.parent = p3
gc3.parent = p3
gc4.parent = p3

这将相应地添加记录:

sqlalchemy.engine.base.Engine INSERT INTO grandchild (name, parent_id, child_id) VALUES (?, ?, ?)
sqlalchemy.engine.base.Engine ('gc1', 1, 1)
[...]

但是,由于数据模型中的父子关系并不意味着任何孙子-父母关系,因此您可以创建一个没有子孙且有孙子的父级。

sink = Parent(name="SINK")
gc1.parent = sink

print("Name: {}, Parent: {}, Parent.children: {}, Child.parent: {}"
      .format(gc1.name, gc1.parent.name, gc1.parent.children, gc1.grandparent.parents.name))
# Name: gc1, Parent: SINK, Parent.children: [], Child.parent: P3

基于我对三层关系的理解,我想不出一个用例。这样就能找到应用程序。
如果您希望通过孩子在父母与孙子女之间建立隐式和一致的关系,请删除父母与孙子女之间的直接关系:

class Parent(Base):
    __tablename__ = 'parent'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    children = relationship("Child", back_populates="parent")

    def __repr__(self):
        return "{}(name={})".format(self.__class__.__name__, self.name)

class Child(Base):
    __tablename__ = 'child'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    parent_id = Column(Integer, ForeignKey('parent.id'))
    parent = relationship("Parent", back_populates="children")
    children = relationship("GrandChild", back_populates="parent")

    # same __repr__()

class GrandChild(Base):
    __tablename__ = 'grandchild'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    child_id = Column(Integer, ForeignKey('child.id'))
    parent = relationship("Child", back_populates="children")

    # same __repr__()

p3 = Parent(name="P3")
c5 = Child(name="C5")
gc1 = GrandChild(name="gc1")    
p3.children = [c5]
c5.children = [gc1]

您可以通过以下方式访问孙子的祖父母:

print(gc1.parent.parent)
# Parent(name=P3)

相反,由于层次结构中的两个一对多关系,因此另一种方式较为繁琐:

for child in p3.children:
    for gc in child.children:
        print(p3, child, gc)
# Parent(name=P3) Child(name=C5) GrandChild(name=gc1)