Django模型,自定义函数

33 投票
3 回答
48867 浏览
提问于 2025-04-15 14:02

我正在用Django创建一个简单的应用程序。发现我经常在做同样的操作。例如,我经常需要获取所有isPublic = True的文章对象。所以我在想,能不能在模型里定义一个get_published函数呢?

如果模型看起来像这样(简化版)

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

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

但这似乎不起作用。

你能建议一下怎么实现这个函数吗?

3 个回答

20

你可以使用staticmethod这个装饰器。

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

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

这个文档中提到,如果你想给你的对象添加一些特别的功能,比如针对每一行数据的操作,你需要在你的模型里定义一些自定义的方法。不过,如果你想要的是针对整个表的功能,比如获取所有符合特定条件的文章对象,那你就需要在模型管理器里定义自定义的方法(就像上面提到的aciniglio在他们的回答中指出的那样)。

86

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

来自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>]

撰写回答