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()

managers.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()

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

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

1 个答案: