Question

我有以下设置：

from django.db import models

class BaseInfoQuerySet(models.query.QuerySet):

    def public(self):
        return self.filter(public=True)

    def not_grouped(self):
        return self.filter(bu_group=True)


class BUManager(models.Manager):

    def get_queryset(self):
        return BaseInfoQuerySet(self.model, using=self._db).extra(
            select={'null_group': 'bu_group_id IS NULL'},
            order_by=['null_group'])


class BU(models.Model):
    # some field definitions

    # option 1 (preferred)
    objects = BaseInfoQuerySet.as_manager()
    # option 2
    objects = BUManager()

我正在使用Django 1.8和Python 2.7。

我想要实现的是能够使用# option 1，但BaseInfoQuerySet中的所有方法都应使用get_queryset()中的修改后的BUManager。 BaseInfoQuerySet用作其他模型的多个查询集的基类，因此我不想删除它并仅使用models.Manager。我还希望能够链接QuerySet过滤器（例如BU.objects.public().not_grouped()）。

我认为解决方案的方法是以某种方式修改方法as_manager()以使用覆盖Manager方法返回修改后的get_queryset。

Answer 1

我用Python3和Django1.10对此进行了测试，因为这是我现在可用的环境。让我知道是否有任何我应该改变以使其在您的环境中工作。

您可以在运行时在管理器实例上设置新的get_queryset方法，如下所示：

import types
from django.db import models

class BaseInfoQuerySet(models.query.QuerySet):

    def public(self):
        return self.filter(public=True)

    def not_grouped(self):
        return self.filter(bu_group=True)

    @classmethod
    def as_manager(cls):

        def get_queryset(self):
            return BaseInfoQuerySet(self.model, using=self._db).extra(
                select={'null_group': 'bu_group_id IS NULL'},
                order_by=['null_group'])

        manager = super(BaseInfoQuerySet, cls).as_manager()
        manager.get_queryset = types.MethodType(get_queryset, manager)
        return manager

但这对我来说看起来不那么复杂。我终于选择了＃2选项。

如果您选择＃2选项，请记住将QuerySet的方法克隆到Manager上或使用常见的mixin。

Answer 2

尝试这种方法：

class BUManager(models.Manager.from_queryset(BaseInfoQuerySet)):
    def get_queryset(self):
        return super().get_queryset().extra(
            select={'null_group': 'bu_group_id IS NULL'},
            order_by=['null_group'])


class BU(models.Model):
    objects = BUManager()

如果由QuerySet.as_manager（）创建，则修改Django Manager get_queryset（）方法

2 个答案: