如何定义关系，使其在 SQLAlchemy 中只能有两个条目？

Question

我有两张桌子 DOOR 和 ROOM （如下），每扇门都应该与两个房间相关联，每个房间可以与一个或多个门相关联。 如何使用 SQLAlchemy 定义此约束？

door_room_association = Table('association:door-room',
                              DeclarativeBase.metadata,
                              Column('door_id', ForeignKey('door.id'), primary_key=true),
                              Column('room_id', ForeignKey('room.id'), primary_key=true))

class Door(DeclarativeBase):
    id: Column(String, primary_key=true)
    locked: Column(Boolean)
    rooms = relationship('Room', secondary=door_room_association, back_populates="doors")

class Room(DeclarativeBase):
    id: Column(String, primary_key=true)
    name: Column(String)
    doors = relationship('Door', secondary=door_room_associaiton, back_populates="rooms")

Answer 1

在 Python 级别，您可以侦听修改与房间相关的门数的事件。

@sa.event.listens_for(Room.doors, 'append')
def door_append_handler(target, value, initiator):
    if len(target.doors) == 2:
        raise ValueError('Too many doors')

由于这是 Python 级别，因此它不会阻止应用程序之外的进程添加门。 如果您需要该级别的保护，您可以考虑添加触发器（可能使用自定义 DDL构造）。 触发器的 DDL 可能因数据库而异。