在已过滤的查询集上使用 Django 自定义管理器功能

Question

考虑以下情况：

class MyModelManager(models.Manager):
    def my_filter(self):
     return [some code here].filter(field_A__gt=3)

class MyModel(models.Model):
    # some fields
    objects = MyModelManager()


# The way I'd like to use it:
qs = MyModel.objects.filter(field_B__lt=1)
# some code
qs = qs.my_filter()

请注意，我想在已过滤的查询集上使用自定义 my_filter() 函数。我应该在上面的[some code here] 中放置什么代码？整体结构是否正确？

Answer 1

class MyModelManager(models.Manager):
    def my_filter(self, qs=None):
        if qs is None:
            qs = [however you got it before]
        return qs.filter(field_A__gt=3)

然后将您的查询集传递给它。

Answer 2

我想没有办法做到这一点。 MyModel.objects.filter(field_B__lt=1) 将返回 QuerySet 对象，而不是 MyModelManager。 您可以像 Ignacio Vazquez-Abrams 提到的那样将它传递给 my_filter 方法，或者首先将 my_filter 应用到您的经理，这将返回 QuerySet 对象，您可以进一步过滤

Answer 3

您的方法是错误的，因为您只能在从数据库中检索数据时使用 Model.Manager，因此使用两个管理器或尝试在查询集上使用一个管理器会导致错误。

data = SomeModel.objects.my_manager.all()

或

data = SomeModel.objects.all()
data = data.my_manager.filter(...)

这是错误的，因为您不能使用两个管理器...

一种可能的方法是定义一个函数，该函数将查询集作为参数并返回过滤后的查询集...

def extra_filter(queryset):
    queryset = queryset.filter(...)

但我不确定这是否有帮助。

最好的方法是定义另一个模型管理器，并在需要时将其用于对象...

class SomeManager(models.Manager):
    def get_query_set(self):
        return super(SomeManager, self).get_query_set().filter(<filteer criteria>)

class MyModel(models.Model):
    # some fields
    objects = models.Manager()
    mymanager = SomeManager()

data = MyModel.mymanager.all() 

只用一个...

更新：在 django 中，查询集是惰性的。这意味着，您可以应用任意数量的过滤器，除非您尝试从过滤器中获取特定记录或尝试对其进行切片，否则不会从数据库中检索数据...Documentation is here

所以，没有区别

qs = MyModel.objects.filter(field_B__lt=1)
qs = qs.filter(field_A__gt=3)

和

qs = MyModel.objects.filter(field_A__gt=3)
qs = qs.filter(field_B__lt=1)

所以定义一个经理来应用一个特定的过滤是使用经理的主要原因...

你定义你的经理：

class SomeManager(models.Manager):
    def get_query_set(self):
        return super(SomeManager, self).get_query_set().filter(field_A__gt=3)

然后你用其他过滤条件调用它：

Somemodel.mymodelmanager.filter(field_B__lt=1, ....)

django 将添加所有过滤器并在您希望查询时评估查询。

所以如果是你经常使用的过滤器，manager i 是最好的选择……

举个例子，我的大多数数据库模型中都有 valid_ 字段和一个过滤无效条目的 val 管理器。因此，当我希望过滤所有有效数据（90% 的时间）时，我使用

Somemodel.val.filter(...)

在我需要所有数据的情况下，我只使用基本的 django 管理器：

Somemodel.objects.filter(...)