Question

我有多张桌子。所有表都有多个列，这些列无法在逻辑上存储在一个合并表中。

class Foo(Model):
    foo_id = Column(Integer(unsigned=True), primary_key=True, nullable=False)
    foo_data = Column(...)

class Bar(Model):
    bar_id = Column(Integer(unsigned=True), primary_key=True, nullable=False)
    bar_info = Column(...)

class Baz(Model):
    baz_id = Column(Integer(unsigned=True), primary_key=True, nullable=False)
    baz_content = Column(...)

现在我想在历史记录表中保存更改：

class Diff(Model):
    diff_id = Column(...)
    foreign_id = Column(Integer(unsigned=True), index=True, nullable=False)
    foreign_table = Column(Char(16, collation='ascii_bin'), index=True, nullable=False)

class SingleDiff(Model):
     ...
     diff = relationship(Diff, backref=backref('changes', uselist=True, ...))

现在我的问题是：在插入新的Diff.foreign_id，Foo或Bar时，如何在同一次提交中填充Baz？

这样做有效，但如果插入Diff会出现问题，则将更改回滚到foo为时已晚：

foo = Foo(foo_data='TODO: changes table names')
try:
    session.add(foo)
    session.commit()  # first commit
except ...:
    ....
else:
    try:
         diff = Diff(changes=[...])
         diff.foreign_id = foo.foo_id
         diff.foerein_table = 'foo'
         session.add(diff)
         session.commit()  # second commit
     except ...:
         raise Exception('Cannot rollback commit #1 anymore :-(')

对于普通关系，id会自动插入：

class FooDiff(Model):
    diff_id = Column(...)
    foo_id = Column(Integer(unsigned=True), ForeignKey(...), ...)
    foo = relationship(Foo)

但我没有Diff.foo，因为Diff.foreign_id可以指向许多不同的表格。

Answer 1

我提出的解决方案与@van已关联的解决方案几乎相同：ORM Examples: Generic Associations。

def _foreign_id_rel(Cls):
    return relationship(
        Cls,
        uselist=False,
        lazy=False,  # This is just my use case, not needed
        primaryjoin=lambda: and_(
            Diff.foreign_table == Cls.__tablename__,
            foreign(Diff.foreign_id) == Cls.id,
        ),
        backref=backref(
            'diffs',
            uselist=True,
            lazy=True,  # This is just my use case, not needed
            primaryjoin=lambda: and_(
                Diff.foreign_table == Cls.__tablename__,
                foreign(Diff.foreign_id) == Cls.id,
            ),
        ),
    )


class Diff(Model):
    ...

    foo = _foreign_id_rel(Foo)
    bar = _foreign_id_rel(Bar)
    baz = _foreign_id_rel(Baz)

    @property
    def foreign(self):
        if self.foreign_table:
            return getattr(self, self.foreign_table)

    @foreign.setter
    def foreign(self, value):
        if value is None:
            self.foreign_table = None
            self.foreign_id = None
        else:
            tbl_name = value.__tablename__
            self.foreign_table = tbl_name
            setattr(self, tbl_name, value)

与SQLalchemy中的不同表的关系

1 个答案: