Django模型,自定义功能

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

你可能想要的是一个自定义经理

来自django文档:

        # 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

如此处的文档中所述,如果需要向对象添加自定义行级功能,则需要在模型上定义自定义方法.但是,如果您所使用的是自定义表范围的功能(例如获取符合某些条件的所有Article对象),则必须在模型管理器上定义自定义方法(就像上面的aciniglio指出的那样).

Answer 3

您可以使用staticmethod装饰器.

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

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