Django 在不 destruct DRY 的情况下自定义 QuerySet 和 Manager

Django has changed!在使用2009年编写的这个答案中的代码之前，一定要查看其他答案和Django文档，看看是否有更合适的解决方案.

我实现这一点的方式是将实际的get_active_for_account添加为自定义QuerySet的方法.然后，要让它在管理器之外工作，您可以简单地捕获__getattr__并相应地将其返回

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

cus至m_queryset/models.py

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

class Cus至mQuerySetManager(models.Manager):
    """A re-usable Manager 至 access a cus至m QuerySet"""
    def __getattr__(self, attr, *args):
        try:
            return getattr(self.__class__, attr, *args)
        except AttributeError:
            # don't delegate internal methods 至 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)

一旦获得了这些，在您的模型上，您所需要做的就是将QuerySet定义为自定义内部类，并将管理器设置为您的自定义管理器:

your_app/models.py个

from cus至m_queryset.models import Cus至mQuerySetManager
from django.db.models.query import QuerySet

class Inquiry(models.Model):
    objects = Cus至mQuerySetManager()

    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)

更新如果您将其与自定义用户(AbstractUser)一起使用，则需要更改
从…

class Cus至mQuerySetManager(models.Manager):

至

from django.contrib.auth.models import UserManager

class Cus至mQuerySetManager(UserManager):
    ***