如何从 ForeignKey 的 Manager 访问自定义 QuerySet 方法

Question

我正在使用 Django Managers 来创建一个更高的 API 来与我的数据库进行交互，并使我的代码更清晰、更易读。但是如果我有Foreignkey 关系，我就不能使用 ForeignKey 模型的管理器。查询更复杂，如下所示，但我只是简化了示例，这样可以更容易阅读和理解问题：

models.py：

class Community(models.Model):
    objects = CommunityManager()
    ...


class Inscription(models.Model):
    objects = InscriptionManager()

    ...
    community = models.ForeignKey("Community", related_name="inscriptions")
    created_at = models.DateTimeField()

ma​​nagers.py：

from datetime import date
from django.db import models


class InscriptionQuerySet(models.query.QuerySet):
    def by_day(self, day=date.today()):
        return self.filter(created_at__day=day)

    ... # more queries


class InscriptionManager(models.Manager):
    def get_query_set(self):
        return InscriptionQuerySet(self.model, using=self._db)

    def today(self):
        return self.get_query_set().by_day()

    ... # more queries


class CommunityQuerySet(models.query.QuerySet):
    def by_type(self, type):
        return self.filter(type=type)

    ... # more queries


class CommunityManager(models.Manager):
    def get_query_set(self):
        return OrganistaionQuerySet(self.model, using=self._db)

    def by_type(self, type):
        return self.get_query_set().by_type(type)

    ... # more queries

用法：

Inscription.objects.by_day() # return correctly all the inscriptions made today
Community.objects.by_type('type1') # return correctly all Communities that match

问题：但问题就在这里

community_b = Community.objects.get(id=12)
community_b.inscriptions.by_day() 
>>> AttributeError: 'ForeignRelatedObjectsDescriptor' object has no attribute 'by_day'

我该如何解决这个问题。如何自定义管理器以考虑模型关系。

Answer 1

我认为您的方法不需要Manager 和QuerySet。你可以取消QuerySet。这就是我通常实现自定义管理器的方式，我尝试在我的一个项目中访问反向关系的自定义管理器方法，它工作正常：

class InscriptionQuerySet(models.QuerySet):
    def by_day(self, day=date.today()):
        return self.filter(created_at__day=day)

    def today(self):
        return self.by_day()

class CommunityQuerySet(models.QuerySet):
    def by_type(self, type):
        return self.filter(type=type)

然后在您的模型中，像这样更改您的objects：

class Community(models.Model):
    objects = CommunityQuerySet.as_manager()
    ...


class Inscription(models.Model):
    objects = InscriptionQuerySet.as_manager()

我认为您可以使用此设置访问自定义查询集方法。