SQLAlchemy 与 SHARED 复合键的多对多关系

Question

我使用复合主键创建了一个 SQLAlchemy 多对多关系。因此，三个表之间的外键也是复合的。其中一个主键列是共享的。它必须具有相同的值。 我使用的代码似乎有效，但是我现在确实收到警告说：relationship 'A.bs' will copy column B.group to column A_B.group，这与关系冲突：'A.bs '（将 Ab 复制到 A_B.group）。考虑将 viewonly=True 应用于只读关系，或提供一个primaryjoin 条件，使用foreign() 注释标记可写列。

class A(Model):
  group = Column(Int, primary_key=True)
  id = Column(Int, primary_key=True)
  bs = relationship('B', secondary=Table('A_B',
    Column('group', Int, primary_key=True),
    Column('a', Int, primary_key=True),
    Column('b', Int, primary_key=True),
    ForeignKeyConstraint(['group', 'a'], ['A.group', 'A.id']),
    ForeignKeyConstraint(['group', 'b'], ['B.group', 'B.id'])
  ))

class B(Model):
  group = Column(Int, primary_key=True)
  id = Column(Int, primary_key=True)

我无法将关系设置为仅查看。如何修改我的代码以使其在没有警告的情况下工作？

Answer 1

除了providing the primaryjoin 带有错误消息中的注释外，您还必须在通过辅助表建立多对多关系时提供secondaryjoin：

a_b = Table('A_B',
  Column('group', Integer, primary_key=True),
  Column('a', Integer, primary_key=True),
  Column('b', Integer, primary_key=True),
  ForeignKeyConstraint(['group', 'a'], ['a.group', 'a.id']),
  ForeignKeyConstraint(['group', 'b'], ['b.group', 'b.id'])
)

class A(Model):
  group = Column(Integer, primary_key=True)
  id = Column(Integer, primary_key=True)
  bs = relationship(
    'B', secondary=a_b,
    primaryjoin='and_(A.group == foreign(A_B.c.group), A.id == foreign(A_B.c.a))',
    secondaryjoin='and_(B.group == A_B.c.group, B.id == foreign(A_B.c.b))')

class B(Model):
  group = Column(Integer, primary_key=True)
  id = Column(Integer, primary_key=True)

A_B.c.group 在辅助连接中没有被注释为foreign()，所以 SQLAlchemy 不会复制它两次。这样做的缺点是可能会做一些意想不到的事情，例如：

In [4]: a1 = A(group=1, id=1)

In [5]: a2 = A(group=2, id=1)

In [6]: session.add_all([a1, a2])

In [7]: b_oops = B(group=1, id=2)

In [8]: session.add(b_oops)

In [9]: a1.bs = [B(group=1, id=1), B(group=2, id=2)]

In [10]: session.commit()

In [11]: b_oops in a1.bs
Out[11]: True

所以B(group=2, id=2) 被插入到数据库中，但b_oops 却与a1 关联。