如何用UniqueConstraint限制同一用户组合仅创建一个对象

要实现user1和user2用户组合（无论顺序）的唯一性，你可以借助Django的Least和Greatest数据库函数，配合UniqueConstraint的expressions参数来定义约束。这种方式能让两个用户的无序组合生成一致的约束键，从而避免重复创建对象。

实现步骤 #

1. 导入所需的类和函数：

from django.db import models
from django.db.models import UniqueConstraint
from django.db.models.functions import Least, Greatest

2. 在模型的Meta类中定义约束：

class MyModel(models.Model):
    user1 = models.ForeignKey(User, on_delete=models.CASCADE, related_name='user1_relations')
    user2 = models.ForeignKey(User, on_delete=models.CASCADE, related_name='user2_relations')
    # 其他业务字段...

    class Meta:
        constraints = [
            UniqueConstraint(
                expressions=[
                    Least('user1', 'user2'),  # 取两个用户ID的较小值
                    Greatest('user1', 'user2'),  # 取两个用户ID的较大值
                ],
                name='unique_user_pair'  # 约束名称，需唯一
            )
        ]

原理说明 #

Least和Greatest函数会分别提取两个用户ID的最小、最大值，不管你传入的是(userX, userY)还是(userY, userX)，最终生成的约束组合都是(min_id, max_id)。UniqueConstraint会确保这个组合在数据库中唯一，从根源上阻止重复的无序用户组合对象被创建。

注意事项 #

• 确保使用的数据库支持Least和Greatest函数：PostgreSQL、MySQL、SQLite 3.33.0及以上版本均支持该函数。
• 执行数据库迁移后，约束会在数据库层面生效，无论是通过Django ORM操作还是直接修改数据库数据，都会受到该约束的限制。

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