Django 定制QuerySet和Manager而不干涸？

我试图找到一种方法来实现自定义

QuerySet

和自定义

Manager

，而不必中断。这就是我到目前为止所做的：

class MyInquiryManager(models.Manager):
    def for_user(self, user):
        return self.get_query_set().filter(
                    Q(assigned_to_user=user) |
                    Q(assigned_to_group__in=user.groups.all())
                )

class Inquiry(models.Model):   
    ts = models.DateTimeField(auto_now_add=True)
    status = models.ForeignKey(InquiryStatus)
    assigned_to_user = models.ForeignKey(User, blank=True, null=True)
    assigned_to_group = models.ForeignKey(Group, blank=True, null=True)
    objects = MyInquiryManager()

这很好，直到我做了这样的事情：

inquiries = Inquiry.objects.filter(status=some_status)
my_inquiry_count = inquiries.for_user(request.user).count()

这会立即破坏一切，因为

QuerySet

与

管理器的方法不同。我尝试创建一个自定义的
QuerySet
类，并在
MyInquiryManager
中实现它，但最终复制了所有的方法定义

我还找到了有效的方法，但我需要将额外的参数传递给
for\u user
，因此它会崩溃，因为它严重依赖于重新定义
get\u query\u set

有没有一种方法可以做到这一点，而不必在
QuerySet
和
Manager
子类中重新定义我的所有方法？
以下内容对我很有用

def get_active_for_account(self,account,*args,**kwargs):
    """Returns a queryset that is 
    Not deleted
    For the specified account
    """
    return self.filter(account = account,deleted=False,*args,**kwargs)

这是在默认管理器上；所以我过去常做这样的事情：

Model.objects.get_active_for_account(account).filter()

但是没有理由认为它不适用于二级经理。
Django已经改变了在使用2009年编写的此答案中的代码之前，请务必查看其余答案和Django文档，看看是否有更合适的解决方案


我实现这一点的方法是为账户

添加实际的

get\u active\u作为自定义
QuerySet
的方法。然后，为了让它在管理器中正常工作，您可以简单地捕获
\uuu getattr\uuu
，并相应地返回它

为了使此模式可重用，我将
管理器
位提取到一个单独的模型管理器：

class InquiryQuerySet(models.QuerySet):
    def for_user(self, user):
        return self.filter(
            Q(assigned_to_user=user) |
            Q(assigned_to_group__in=user.groups.all())
        )

class Inquiry(models.Model):
    objects = InqueryQuerySet.as_manager()

自定义查询集/models.py

from django.db import models
from django.db.models.query import QuerySet

class CustomQuerySetManager(models.Manager):
    """A re-usable Manager to access a custom QuerySet"""
    def __getattr__(self, attr, *args):
        try:
            return getattr(self.__class__, attr, *args)
        except AttributeError:
            # don't delegate internal methods to the queryset
            if attr.startswith('__') and attr.endswith('__'):
                raise
            return getattr(self.get_query_set(), attr, *args)

    def get_query_set(self):
        return self.model.QuerySet(self.model, using=self._db)

from custom_queryset.models import CustomQuerySetManager
from django.db.models.query import QuerySet

class Inquiry(models.Model):
    objects = CustomQuerySetManager()

    class QuerySet(QuerySet):
        def active_for_account(self, account, *args, **kwargs):
            return self.filter(account=account, deleted=False, *args, **kwargs)

一旦获得了该类，在模型上只需将
QuerySet
定义为自定义内部类，并将管理器设置为自定义管理器：

你的应用程序/模型.py

from django.db import models
from django.db.models.query import QuerySet

class CustomQuerySetManager(models.Manager):
    """A re-usable Manager to access a custom QuerySet"""
    def __getattr__(self, attr, *args):
        try:
            return getattr(self.__class__, attr, *args)
        except AttributeError:
            # don't delegate internal methods to the queryset
            if attr.startswith('__') and attr.endswith('__'):
                raise
            return getattr(self.get_query_set(), attr, *args)

    def get_query_set(self):
        return self.model.QuerySet(self.model, using=self._db)

from custom_queryset.models import CustomQuerySetManager
from django.db.models.query import QuerySet

class Inquiry(models.Model):
    objects = CustomQuerySetManager()

    class QuerySet(QuerySet):
        def active_for_account(self, account, *args, **kwargs):
            return self.filter(account=account, deleted=False, *args, **kwargs)

使用此模式，以下任何一项都将起作用：

>>> Inquiry.objects.active_for_account(user)
>>> Inquiry.objects.all().active_for_account(user)
>>> Inquiry.objects.filter(first_name='John').active_for_account(user)

UPD如果与自定义用户一起使用（
AbstractUser
），则需要更改

从

到

T.Stone方法的稍微改进版本：

def objects_extra(mixin_class):
    class MixinManager(models.Manager, mixin_class):
        class MixinQuerySet(QuerySet, mixin_class):
            pass

        def get_query_set(self):
            return self.MixinQuerySet(self.model, using=self._db)

    return MixinManager()

类装饰器的使用非常简单：

class SomeModel(models.Model):
    ...
    @objects_extra
    class objects:
        def filter_by_something_complex(self, whatever parameters):
            return self.extra(...)
        ...

更新：支持非标准的Manager和QuerySet基类，例如。g@额外对象（django.contrib.gis.db.models.GeoManager、django.contrib.gis.db.models.query.GeoQuerySet）：

可以使用mixin在manager和queryset上提供方法

这也避免了使用
\uuu getattr\uuu（）
方法

from django.db.models.query import QuerySet

class PostMixin(object):
    def by_author(self, user):
        return self.filter(user=user)

    def published(self):
        return self.filter(published__lte=datetime.now())

class PostQuerySet(QuerySet, PostMixin):
    pass

class PostManager(models.Manager, PostMixin):
    def get_query_set(self):
        return PostQuerySet(self.model, using=self._db)

Django 1.7发布了一种创建组合查询集和模型管理器的新的简单方法：

class InquiryQuerySet(models.QuerySet):
    def for_user(self, user):
        return self.filter(
            Q(assigned_to_user=user) |
            Q(assigned_to_group__in=user.groups.all())
        )

class Inquiry(models.Model):
    objects = InqueryQuerySet.as_manager()

有关更多详细信息，请参阅。
您现在可以使用管理器上的方法更改其基本查询集

这允许您只定义一次Queryset方法和manager方法

从文件中

对于高级用法，您可能需要自定义管理器和自定义查询集。您可以通过调用Manager.from_queryset（）来实现这一点，该函数返回基本管理器的子类以及自定义queryset方法的副本：

在一些用例中，我们需要使用QuerySet的
get\u manager
方法来代替

基于发布在一个已接受的解决方案注释中的解决方案，mixin就足够了

class customquerystmanagermixin:
"""
允许使用自定义queryset的管理器直接访问queryset方法。
"""
def _ugetattr _;（self，name）：
#不要将内部方法委托给queryset
#注意：没有这个，经理.\u copy\u to\u model将调用
#引起qs的*queryset*上的
#像被腌制一样评价自己（`len（self）`）
如果name.startswith（“”“”）：
提高属性错误
返回getattr（self.get\u queryset（），name）

比如说,

类BookQuerySet（models.QuerySet）：
已发布def（自我）：
返回self.filter（published=True）
def小说（自我）：
return self.filter（类型=“虚构”）
def非虚构（自我）：
return self.filter（类型=“非虚构”）
类BookManager（CustomQuerySetManager，models.Manager）：
def get_queryset（自我）：
return BookQuerySet（self.model，using=self.\u db）.published（）
教材（models.Model）：
title=models.CharField（最大长度=200）
流派=models.CharField（选项=[（‘虚构’、‘虚构’）、（‘非虚构’、‘非虚构’）]）
published=models.BooleanField（默认值=False）
author=models.ForeignKey（author，on_delete=models.CASCADE，related_name=“books”）
objects=BookManager（）
类作者（models.Model）：
name=models.CharField（最大长度=200）

通过上面的内容，我们可以访问下面这样的相关对象（书籍），而无需在管理器中为每个queryset方法定义新方法

fiction\u books=作者.书籍.小说（）
尝试执行
筛选
，然后使用
为帐户
获取活动。它在您的示例中起作用，但在您已经使用了
过滤器
，然后正在使用
查询集
，这是我的示例。您能决定如何使用Stone吗？您必须自己编辑它，declinedWARNING之后无法进行编辑：我尝试了此方法，发现它严重减慢了.defer和.only调用。我已将此响应更新为社区wiki。那些进行性能优化的人可以根据需要调整代码。可能类似于PassThroughManager from Warning：当使用.defer或.only方法时，T.Stone选择的答案会导致严重的性能损失（从毫秒响应时间到多秒响应）。例如，在Django 1.3中，一个查询，例如：MyModel.objects.only（'some_字段'）.get（id=1）=>以3.7ms的速度返回，但是，如上所述添加CustomManager，我得到：MyModel.objects.only（'some_字段'）。get（id=1）=>以3.7ms的速度返回~
class InqueryQueryset(models.Queryset):
    def custom_method(self):
        """ available on all default querysets"""

class BaseMyInquiryManager(models.Manager):
    def for_user(self, user):
        return self.get_query_set().filter(
                    Q(assigned_to_user=user) |
                    Q(assigned_to_group__in=user.groups.all())
                )

MyInquiryManager = BaseInquiryManager.from_queryset(InquiryQueryset)

class Inquiry(models.Model):   
    ts = models.DateTimeField(auto_now_add=True)
    status = models.ForeignKey(InquiryStatus)
    assigned_to_user = models.ForeignKey(User, blank=True, null=True)
    assigned_to_group = models.ForeignKey(Group, blank=True, null=True)
    objects = MyInquiryManager()