Django Model : 使用自定义查询集定义自定义管理器

2023-04-08 00:00:00 定义自定义管理器

在Django中，我们可以通过定义自定义查询集（Custom QuerySet）来定制我们的模型管理器（Model Manager），以便于我们更方便地访问数据库。

下面是一个简单的示例代码，演示如何定义自定义查询集来定制管理器：

from django.db import models

class CustomQuerySet(models.QuerySet):
    def published(self):
        return self.filter(is_published=True)

    def created_by(self, author):
        return self.filter(created_by=author)


class ArticleManager(models.Manager):
    def get_queryset(self):
        return CustomQuerySet(self.model, using=self._db)


class Article(models.Model):
    title = models.CharField(max_length=100)
    content = models.TextField()
    is_published = models.BooleanField(default=False)
    created_by = models.CharField(max_length=50)

    objects = ArticleManager()

在上面的例子中，我们定义了一个自定义查询集 CustomQuerySet，并在其中定义了两个自定义查询方法：published 和 created_by。其中 published 方法返回所有 is_published 属性值为 True 的文章；created_by 方法根据作者筛选文章。

接下来，我们再定义了一个 ArticleManager 管理器类，将 CustomQuerySet 设置为其默认查询集。

最后，在 Article 模型中我们将 objects 属性设置为 ArticleManager 的实例对象。这样，我们就可以通过调用对象的 objects 属性获取 Article 对象实例，并使用其中的 published 和 created_by 方法筛选数据。

以下是一些示例代码：

# 返回所有已发布的文章
published_articles = Article.objects.published()

# 返回所有由 “皮蛋编程” 创建的文章
my_articles = Article.objects.created_by("皮蛋编程")

总之，定制自己的查询集和管理器是 Django 模型中非常强大和有用的特性，可以为我们的应用程序提供更好的性能和更方便的数据访问方式。

相关文章