Question

我用django创建简单的应用程序。而且，我意识到我经常做某种操作。例如，我经常需要获取所有具有isPublick = True的Article对象。所以我在想是可以在模型中定义get_published函数吗？

如果模型看起来像这样（简化）

class Article(models.Model):
    title = models.CharField(...)
    isPublished = models.BooleandField()

    def get_active(self):
       return Article.objects.filter(isPublicshed = 1)

但它不会这样工作

你能建议一种实现这个功能的方法吗？

Answer 1

您可能想要的是custom manager

来自django docs：

        # An example of a custom manager called "objects".

class PersonManager(models.Manager):
    def get_fun_people(self):
        return self.filter(fun=True)

class Person(models.Model):
    first_name = models.CharField(max_length=30)
    last_name = models.CharField(max_length=30)
    fun = models.BooleanField()
    objects = PersonManager()

    def __unicode__(self):
        return u"%s %s" % (self.first_name, self.last_name)

然后允许您执行以下操作：

>>> p1 = Person(first_name='Bugs', last_name='Bunny', fun=True)
>>> p1.save()
>>> p2 = Person(first_name='Droopy', last_name='Dog', fun=False)
>>> p2.save()
>>> Person.objects.get_fun_people()
[<Person: Bugs Bunny>]

Answer 2

正如docs here中所述，如果您需要向对象添加自定义行级功能，则需要在模型上定义自定义方法。但是，如果您使用的是自定义表格范围功能（例如获取符合特定条件的所有文章对象），则必须在model Managers上定义自定义方法（就像 aciniglio 以上指出他们的答案。）

Answer 3

您可以使用staticmethod装饰器。

class Article(models.Model):
    title = models.CharField(...)
    isPublished = models.BooleandField()

    @classmethod
    def get_active():
        return Article.objects.filter(isPublished = 1)

Django模型，自定义功能

3 个答案: