如何在SQLAlchemy中指定关联字段为创建必填项？

要在SQLAlchemy中实现创建Study记录时必须关联至少一个Participant，分两种场景给出简洁实现：

1. 数据库层面强制约束（仅支持PostgreSQL等部分数据库） #

部分数据库（如PostgreSQL）支持在CHECK约束中使用子查询，我们可以给Study表添加CHECK约束，确保关联的Participant数量至少为1：

class Study(db.Model):
    id = db.Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    type = db.Column(Enum(StudyTypeChoices), nullable=False)
    owner_id = db.Column(UUID(as_uuid=True), db.ForeignKey('owner.id'), nullable=False)
    participants = db.relationship('Participant', lazy=True, cascade='save-update, merge, delete')

    __table_args__ = (
        db.CheckConstraint(
            "(SELECT COUNT(*) FROM participant WHERE participant.study_id = study.id) >= 1",
            name="study_has_at_least_one_participant"
        ),
    )

注意：MySQL不支持CHECK约束中的子查询，这种方法在MySQL环境下无效。

2. SQLAlchemy事件监听（跨数据库兼容，应用层接近数据库层） #

利用SQLAlchemy的before_insert事件监听，在写入数据库前检查participants是否非空，不符合则抛出异常：

from sqlalchemy import event

class Study(db.Model):
    id = db.Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    type = db.Column(Enum(StudyTypeChoices), nullable=False)
    owner_id = db.Column(UUID(as_uuid=True), db.ForeignKey('owner.id'), nullable=False)
    participants = db.relationship('Participant', lazy=True, cascade='save-update, merge, delete')

@event.listens_for(Study, 'before_insert')
def check_study_has_participants(mapper, connection, target):
    if not target.participants:
        raise ValueError("创建Study时必须至少关联一个Participant")

这种方法跨数据库兼容，且逻辑直接嵌入ORM层，不需要额外包装器。

补充说明 #

如果要同时确保Participant必须关联Study，可以修改Participant的study_id字段为非空：

class Participant(UserBase):
    id = db.Column(UUID(as_uuid=True), db.ForeignKey("user_base.id"), primary_key=True)
    study_id = db.Column(UUID(as_uuid=True), db.ForeignKey('study.id'), nullable=False)

但这只能保证Participant不能脱离Study存在，无法反向约束Study创建时必须有Participant，需要配合上述两种方法之一使用。

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