Django自定义唯一的一起约束[英] Django custom unique together constraint

本文是小编为大家收集整理的关于Django自定义唯一的一起约束的处理方法,想解了Django自定义唯一的一起约束的问题怎么解决?Django自定义唯一的一起约束问题的解决办法?那么可以参考本文帮助大家快速定位并解决问题。

问题描述

我有一个用户共享类似于以下的模型:

class Share( models.Model ):
    sharer = models.ForeignKey(User, verbose_name=_("Sharer"), related_name='sharer')
    receiver = models.ForeignKey(User, verbose_name=_("Receiver"), related_name='receiver')

    class Meta:
        unique_together = ( ("sharer", "receiver"), ("receiver", "sharer") )

我想为共享者(S)和接收器(R)保存一个对象(订单无关紧要R-S或S-R).但是上面的unique_together将无法实现这一点;假设R-S在数据库中,然后如果我保存S-R,我将不会为此获得验证.为此,我为共享模型编写了自定义唯一验证.

    def validate_unique(
        self, *args, **kwargs):
            super(Share, self).validate_unique(*args, **kwargs)
            if self.__class__.objects.filter( Q(sharer=self.receiver, receiver=self.sharer) ).exists():
                raise ValidationError(
                    {
                        NON_FIELD_ERRORS:
                        ('Share with same sharer and receiver already exists.',)
                    }
                )

    def save(self, *args, **kwargs):
        # custom unique validate
        self.validate_unique()
        super(Share, self).save(*args, **kwargs)

此方法在正常使用中正常工作.

问题: 我有一个匹配的算法,该算法获得了共享的请求,并保存共享对象(s-r或r-s),然后将它们发送给他们响应(共享对象) 几乎同时 .当我使用查询(无数据库级别)检查重复时,这需要时间,所以最后我有2个对象S-R和R-S.

我想要一些解决方案,对于共享者s和a接收者r,我只能保存单个共享对象,s-r或r-s else否则会获得一些验证错误(例如databse的IntegrityError).

django = 1.4,数据库= postgresql

推荐答案

您可能可以使用PostgreSQL的indexes on expressions解决此问题,但这是另一种方法:

class Share( models.Model ):
    sharer = models.ForeignKey(User)
    receiver = models.ForeignKey(User), related_name='receiver')
    key = models.CharField(max_length=64, unique=True)

    def save(self, *args, **kwargs):
        self.key = "{}.{}".format(*sorted([self.sharer_id, self.receiver_id]))
        super(Share, self).save(*args, **kwargs)

,如果您使用QuerySet.update方法更改值,那显然是行不通的.您还可以查看 django-denorm ,它可以用触发器解决此问题.

本文地址:https://www.itbaoku.cn/post/1763816.html