Django - 具有Single True的布尔字段

时间:2014-01-04 17:11:57

标签: python django django-models database

我想要一个布尔字段,只允许其中一个基于外键( User True 。我现在肯定如何说这个或者真正搜索它,但我能够得到一个简单的实现工作,但我不喜欢它。感觉笨拙和错误。无论如何,获得我的意思的最好方法是你看到它,你走了:

class MyModel(models.Model):
    owner = models.ForeignKey(User)
    _is_main = models.BooleanField(default=False)

    def __get_is_main(self):
        return self._is_main

    def __set_is_main(self, is_main):
        if (is_main):
            active_keys = API_Key.objects.filter(_is_main=True, owner=self.owner)
            if (len(active_keys) == 1 and self in active_keys):
                return
            else:
                for key in active_keys:
                    if (key.is_main):
                        key.is_main = False
                        key.save()
        self._is_main = is_main
        self.save()

    is_main = property(__get_is_main, __set_is_main)

我只希望每个所有者 ONCE的is_main为 True ,不超过一次。这些是用于API密钥。因此,所有者可以拥有许多API密钥,但只有一个设置为主密钥。这是我弄清楚如何实现这一点的唯一方法。谁有更好的方法来做到这一点?如果重要的话,我正在使用MySQL。

2 个答案:

答案 0 :(得分:1)

也许您不需要该字段但是有一个指向主键的单例。这些方面的东西:

class Key(models.Model):
    owner = models.ForeignKey(User)

    @property
    def is_main(self):
        return hasattr(self, 'is_main')

    def set_main(self):
        # update MainKey instance here


class MainKey(models.Model):
    key = models.ForeignKey(Key, related_name='is_main')

    def save(self, *args, **kwargs):
        self.id = 1
        super(MainKey, self).save(*args, **kwargs)

    def delete(self, *args, **kwargs):
        pass

答案 1 :(得分:0)

我最近制作了一个自定义字段来实现这一目标:

https://pypi.python.org/pypi/django-exclusivebooleanfield

class MyModel(models.Model):
    owner = models.ForeignKey(User)
    _is_main = ExclusiveBooleanField(default=False, on=('owner',))

它通过包装模型的保存方法来工作。