Django模型,自定义函数
我正在用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)
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>]