postgresql - 具有多态实体的子类关联表的 SQLAlchemy 设置

Question

我是两个新手 python和 sqlalchemy但是有一个相当复杂的建模情况，我在设置时遇到了麻烦。它涉及一个关联表，其中关联表具有实体之一的多态关联。
我非常接近让这个工作。当数据库中已经存在数据时，我可以按预期读取它和模型并与之交互。问题出在写作上，我会在展示代码后解决这个问题:
首先，有一个共享基类将表名和 id 定义为 postgres uuid

@as_declarative()
class Base(object):
    @declared_attr
    def __tablename__(cls):
        return cls.__name__.lower()

    id = Column(pg.UUID(as_uuid=True), primary_key=True, default=uuid4)

多态列中允许有一个已定义类型的枚举。虽然我最终会支持 Aaa 和 Bbb，但为了清楚起见，这个例子到目前为止只定义了 Bbb。

class EntityTypes(Enum):
    AAA = Aaa.__name__.lower()
    BBB = Bbb.__name__.lower()

这是表示多态关联表的模型。它有 entity_id , entity_type , 和 ccc_id用于连接的列。 “实体”可以是 Aaa 或 Bbb，但 Ccc 始终是 Ccc(非多态)。

class EntityCcc(Base):
    """Polymorphic mapping between an EntityType and Ccc."""
    
    # entity_id is defined in subclasses, with foreign keys
    entity_type = Column(
        Enum(EntityTypes, values_callable=lambda x: [e.value for e in x]), nullable=False
    )

    ccc_id = Column(
        pg.UUID(as_uuid=True),
        ForeignKey(f"{Ccc.__tablename__}.id"),
        nullable=False
    )

    __mapper_args__ = {
        "polymorphic_on": entity_type
    }

这是多态模型的一个子类，它知道 Bbb s，因此它相应地设置外键，并建立关系。类似的 AaaCcc类将被定义

class BbbCcc(EntityCcc):
    """
        Mapping between a Bbb and Ccc.
        Subclasses polymorphic join model to get specific
        bbb accessor and set appropriate foreign key.
    """

    __tablename__ = EntityCcc.__tablename__

    entity_id = Column(
        pg.UUID(as_uuid=True), ForeignKey(f"{Bbb.__tablename__}.id"), nullable=False
    )

    bbb = relationship("Bbb", back_populates='bbb_ccc')
    ccc = relationship("Ccc", back_populates='bbb_ccc')

    __mapper_args__ = {
        "polymorphic_identity": EntityTypes(Bbb.__name__.lower())
    }

这是一个 Bbb ，它具有到子类连接模型的关系设置，以及指向其 Ccc 的辅助关系设置通过连接表(使用 secondary )

class Bbb(Base):
    """Represents a Bbb entity."""

    name = Column(TEXT)

    bbb_ccc = relationship("BbbCcc", back_populates="bbb", uselist=False)

    ccc = relationship(
        "Ccc",
        secondary="entity_ccc",
        back_populates="bbb",
        uselist=False
    )

这是一个 Ccc，它具有到子类连接模型的关系设置，以及指向其 Bbb 的辅助关系设置。通过连接表(使用 secondary )

class Ccc(Base):
    """Represents a Ccc entity."""

    name = Column(TEXT)

    bbb_ccc = relationship("BbbCcc", back_populates="ccc", uselist=False)

    bbb = relationship(
        "Bbb",
        secondary="entity_ccc",
        back_populates="ccc",
        uselist=False
    )

所以有什么问题？
使用已经在数据库中播种的适当条目，我可以按预期与它们交互:

(Pdb) found_bbb_ccc = db.session.query(BbbCcc).get(uuid)
(Pdb) found_bbb_ccc
<app.models.mappings.bbb_ccc.BbbCcc object at 0x7f488ce6ebe0>
(Pdb) found_bbb_ccc.bbb
<app.models.entities.bbb.Bbb object at 0x7f488dd73f10>
(Pdb) found_bbb_ccc.ccc
<app.models.entities.ccc.Ccc object at 0x7f488ce6ec40>
(Pdb) found_bbb_ccc.bbb.ccc
<app.models.entities.ccc.Ccc object at 0x7f488ce6ec40>
(Pdb) found_bbb_ccc.ccc.bbb
<app.models.entities.bbb.Bbb object at 0x7f488dd73f10>

这表明 bbb可以引用并找到它的 ccc通过连接模型，反之亦然。通过协会阅读很好。但是通过写入创建新的关联是有问题的:

new_bbb = Bbb(name='Bbb instance')
new_ccc = Ccc(name='Ccc instance')

new_bbb.ccc = new_ccc
db.session.commit()

*** sqlalchemy.exc.IntegrityError: (psycopg2.errors.NotNullViolation) null value in column "entity_type" violates not-null constraint
DETAIL:  Failing row contains (4b1f7ac7-16b0-4972-9577-bda1b5efe2aa, 2021-08-05 17:50:05.233465, 2021-08-05 17:50:05.233482, 63463492-0a9d-492f-b42a-72ec276f2768, null, a75d06af-33bd-4345-abbd-c6098e9a797d).

[SQL: INSERT INTO entity_ccc (created, updated, id, ccc_id, entity_id) VALUES (%(created)s, %(updated)s, %(id)s, %(ccc_id)s, %(entity_id)s)]
[parameters: {'created': datetime.datetime(2021, 8, 5, 17, 50, 5, 233465), 'updated': datetime.datetime(2021, 8, 5, 17, 50, 5, 233482), 'id': UUID('4b1f7ac7-16b0-4972-9577-bda1b5efe2aa'), 'ccc_id': UUID('a75d06af-33bd-4345-abbd-c6098e9a797d'), 'entity_id': UUID('63463492-0a9d-492f-b42a-72ec276f2768')}]
(Background on this error at: http://sqlalche.me/e/14/gkpj)

看到的错误是这个数据的写入没有设置多态 entity_type ，应该是 bbb这里。我相信问题在于 secondary关系定义需要一个表名，而不是一个对象(传递 BbbCcc 可能会选择 entity_type 对吗？)但也许它是别的东西。
如何调整此代码以允许设置所描述的多态关联？谢谢!

Answer 1

尝试

from sqlalchemy import inspect

    [...]
    eng_mapper = inspect(Engineer)
    query.filter(
        eng_mapper.polymorphic_on.in_(
            m.polymorphic_identity
            for m in eng_mapper.polymorphic_iterator()
        ),
    )

我更喜欢稍微不那么冗长的咒语，但这有效并且不需要了解多态层次结构的特定配置。
细节
当在 ORM 映射类上调用 inspect() 时，它返回该类的 Mapper。这与模型相同。 映射器 类属性。
Mapper 包含内省(introspection)多态层次结构所需的所有信息。特别是:
.polymorphic_on 是模型中位于层次结构顶部的字段(列)，它包含记录的多态标识值(例如，工程师的 Employee.type 字段)。
.polymorphic_identity 是映射模型的每个实例将在 .polymorphic_on 字段中具有的值(例如，对于工程师来说，将是“工程师”)。
.polymorphic_iterator() 迭代包含模型的模型映射器集合。 映射器 和。 映射器 Model 的所有子类的递归(例如，对于 Engineer 来说，它是一个只包含 Engineer 的迭代器。 映射器 )。
为了使其更具可读性，可以轻松地将上述过滤器表达式转换为函数:
从 sqlalchemy 导入检查

def filter_instances_of(cls):
    mapper = inspect(cls)
    return mapper.polymorphic_on.in_(
        m.polymorphic_identity for m in mapper.polymorphic_iterator()
    )

并像这样使用它:

query = query.filter(
    filter_instances_of(Engineer),
    [... other filter criteria ...]
)