三列上的唯一约束如何定义？

Question

下面的EventInvitation模型是一个事件的简单邀请，从一个用户发送给另一个用户。我想确保邀请函在三个栏目中是唯一的：致用户、发自用户和活动。在

class EventInvitation(db.Model):
    __tablename__ = 'event_invitations'

    id = db.Column(db.Integer, primary_key = True)

    event_id = db.Column(db.Integer, db.ForeignKey('events.id'))
    event = db.relationship('Event',  foreign_keys=[event_id])
    created = db.Column(db.DateTime(), default=datetime.now)
    updated = db.Column(db.DateTime(), default=datetime.now,onupdate=datetime.now)

    from_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    from_user = db.relationship('User',  foreign_keys=[from_id])

    to_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    to_user = db.relationship('User',  foreign_keys=[to_id])

    cstrt = db.UniqueConstraint('event_id', 'from_id','to_id', name='uix_1')

我试过使用这个cstrt列，但它不起作用。该约束应该适用于SQLite，也适用于生产中的MySQL。如何定义这个唯一的约束？在

Answer 1

您需要将约束添加到表中，而不是模型中。要使用声明性方法执行此操作，请执行以下操作：

class EventInvitation(db.Model):
    # ...
    __table_args__ = (
        db.UniqueConstraint(event_id, from_id, to_id),
    )

如果已经在数据库中创建了表，则需要删除该表并再次运行db.create_all()，或者使用Alembic通过迁移更改现有的表。在