如何在SQLModel/SQLAlchemy中表示可辨识联合关系？

我们有一个包含三张表的数据库结构：

• Parent表：包含id（主键）和child_type（ENUM类型，取值为a/b）字段，二者组合为唯一约束
• ChildTypeA/ChildTypeB表：各自包含supertype_key（对应Parent的child_type）、parent_id字段，通过复合外键(parent_id, supertype_key)关联Parent的(id, child_type)，同时有各自专属字段

使用SQLModel编写模型时，尝试在Parent类中定义child: Union["ChildTypeA", "ChildTypeB"] = Relationship(back_populates="parent")会触发错误：

ValueError: Cannot have a Union of None and None as a SQLAlchemy field

SQLModel当前对Union类型的Relationship支持存在限制，我们可以通过直接使用SQLAlchemy原生的relationship函数并手动指定关联条件来解决问题，同时保留SQLModel的类型提示：

from enum import Enum
from typing import Union, Optional

from sqlalchemy import relationship
from sqlalchemy.schema import ForeignKeyConstraint, UniqueConstraint
from sqlmodel import Field, SQLModel


class ChildType(str, Enum):
    A = "a"
    B = "b"


class Parent(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    child_type: ChildType
    
    # 为ChildTypeA定义关联，指定精确匹配条件
    child_a: Optional["ChildTypeA"] = relationship(
        "ChildTypeA",
        primaryjoin="and_(Parent.id == ChildTypeA.parent_id, Parent.child_type == ChildTypeA.supertype_key)",
        uselist=False,
        back_populates="parent",
        foreign_keys="[ChildTypeA.parent_id, ChildTypeA.supertype_key]"
    )
    # 为ChildTypeB定义关联，指定精确匹配条件
    child_b: Optional["ChildTypeB"] = relationship(
        "ChildTypeB",
        primaryjoin="and_(Parent.id == ChildTypeB.parent_id, Parent.child_type == ChildTypeB.supertype_key)",
        uselist=False,
        back_populates="parent",
        foreign_keys="[ChildTypeB.parent_id, ChildTypeB.supertype_key]"
    )

    @property
    def actual_child(self) -> Union["ChildTypeA", "ChildTypeB", None]:
        """统一返回对应类型的子记录"""
        return self.child_a if self.child_type == ChildType.A else self.child_b

    __table_args__ = (
        UniqueConstraint("id", "child_type"),
    )


class ChildTypeA(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    supertype_key: ChildType = Field(sa_column_kwargs={
        "server_default": ChildType.A.value,
    })
    parent_id: int = Field()
    parent: Parent = relationship(back_populates="child_a")

    some_special_A_type_field: int

    __table_args__ = (
        ForeignKeyConstraint(
            ["parent_id", "supertype_key"],
            ["parent.id", "parent.child_type"],
        ),
        UniqueConstraint("parent_id", "supertype_key"),
    )


class ChildTypeB(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    supertype_key: ChildType = Field(sa_column_kwargs={
        "server_default": ChildType.B.value,
    })
    parent_id: int = Field()
    parent: Parent = relationship(back_populates="child_b")

    some_special_B_type_field: str

    __table_args__ = (
        ForeignKeyConstraint(
            ["parent_id", "supertype_key"],
            ["parent.id", "parent.child_type"],
        ),
        UniqueConstraint("parent_id", "supertype_key"),
    )

关键修改点： #

• 替换SQLModel的Relationship为SQLAlchemy原生relationship函数，手动指定primaryjoin条件，确保只匹配对应child_type的子表记录
• 为两个子表分别定义独立关联，通过actual_child属性统一返回对应类型的子记录
• 修正server_default取值：使用ChildType.A.value而非name，匹配ENUM对应的字符串值（数据库中存储的是"a"而非"A"）
• 为子表添加UniqueConstraint("parent_id", "supertype_key")，确保每个Parent只能关联一条对应类型的子记录（替代原代码中parent_id的单字段唯一约束，逻辑更准确）

运行原有的SQL生成代码，现在可以正常生成表结构，同时Parent类的actual_child属性可以正确返回对应类型的子记录，不会触发之前的错误。

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