自定义QuerySet和Manager而不破坏DRY？

一尘不染

自定义QuerySet和Manager而不破坏DRY？

django

我正在尝试找到一种方法来实现自定义QuerySet和自定义Manager而不破坏DRY。这是我到目前为止所拥有的：

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方法与的方法不同Manager。我尝试过创建一个自定义QuerySet类，并在中实现它MyInquiryManager，但最终我复制了所有方法定义。

我也发现此代码片段有效，但是我需要将额外的参数传递给，for_user因此它无法使用，因为它严重依赖于重新定义get_query_set。

有没有一种方法可以在不重新定义子类QuerySet和Manager子类中的所有方法的情况下进行此操作？

阅读 401

2020-03-26

共1个答案

一尘不染

我实现此目标的方法是将实际值添加get_active_for_account为自定义方法QuerySet。然后，要使其脱离管理器，你只需捕获__getattr__并相应地将其返回

为了使该模式可重用，我将这些Manager位提取到单独的模型管理器中：

custom_queryset / 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)

一旦掌握了这些，就可以在模型上将QuerySeta定义为自定义内部类，并将管理器设置为自定义管理器：

your_app / models.py

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）一起使用，则需要
从

class CustomQuerySetManager(models.Manager):

至

from django.contrib.auth.models import UserManager

class CustomQuerySetManager(UserManager):
    ***

2020-03-26