Python 3 Django Rest Framework - 如何向这个 M-1-M 模型结构添加自定义管理器？

Question

我有这些模型：

• 组织

• 学生

• 课程

• 注册

学生属于组织

学生可以注册一门或多门课程

因此，注册记录基本上由给定的课程和给定的学生组成

from django.db import models
from model_utils.models import TimeStampedModel

class Organisation(TimeStampedModel):
  objects = models.Manager()
  name = models.CharField(max_length=50)

  def __str__(self):
    return self.name

class Student(TimeStampedModel):
  objects = models.Manager()
  first_name = models.CharField(max_length=50)
  last_name = models.CharField(max_length=50)
  email = models.EmailField(unique=True)
  organisation = models.ForeignKey(to=Organisation, on_delete=models.SET_NULL, default=None, null=True)

  def __str__(self):
    return self.email

class Course(TimeStampedModel):
  objects = models.Manager()
  language = models.CharField(max_length=30)
  level = models.CharField(max_length=2)

  def __str__(self):
    return self.language + ' ' + self.level

  class Meta:
    unique_together = ("language", "level")

class EnrollmentManager(models.Manager):
  def org_students_enrolled(self, organisation):
    return self.filter(student__organisation__name=organisation).all()

class Enrollment(TimeStampedModel):
  objects = EnrollmentManager()
  course = models.ForeignKey(to=Course, on_delete=models.CASCADE, default=None, null=False, related_name='enrollments')
  student = models.ForeignKey(to=Student, on_delete=models.CASCADE, default=None, null=False, related_name='enrollments')
  enrolled = models.DateTimeField()
  last_booking = models.DateTimeField()
  credits_total = models.SmallIntegerField(default=10)
  credits_balance = models.DecimalField(max_digits=5, decimal_places=2)

请注意自定义 EnrollmentManager，它允许我查找从给定组织注册的所有学生。

如何添加自定义管理器以检索学生注册的给定组织的所有课程？

我的尝试

我想创建一个 CourseManager 并以某种方式从关系的那一侧查询/过滤：

class CourseManager(models.Manager):
  def org_courses_enrolled(self, organisation):
    return self.filter(enrollment__student__organisation__name=organisation).all()

这可行，但它给了我相同的 100 条注册记录 :(

我想要得到的是： 基于给定的组织 查找所有已注册的学生 然后 (DISTINCT?) 获取该组织的已注册课程列表

这是风景：

class OrganisationCoursesView(mixins.ListModelMixin, mixins.RetrieveModelMixin, viewsets.GenericViewSet):
  serializer_class = CourseSerializer
  queryset = Course.objects.get_courses(1)

和网址：

# The below should allow: /api/v1/organisations/1/courses/
router.register('api/v1/organisations/(?P<organisation_pk>\d+)/courses', OrganisationCoursesView, 'organisation courses')

更新 1

根据 h1dd3n 的回答，我接下来尝试了这个：

class CourseManager(models.Manager):

  def get_queryset(self):
     return super(CourseManager, self).get_queryset()

  def get_courses(self, organisation):
    return self.get_queryset().filter(student__organisation_id=organisation)

但这会引发错误（正如我所料）：

FieldError：无法将关键字“学生”解析为字段。选择是： 课程、创建、id、语言、级别、修改、进度

更新 2 - 越来越近了！

好的，在 @AKX 的 cmets 的帮助下：

class CourseManager(models.Manager):

  def get_queryset(self):
     return super(CourseManager, self).get_queryset()

  def get_courses(self, organisation):
    return self.get_queryset().filter(courses__student__organisation_id=organisation)

现在确实会返回课程，但它会为每个注册学生返回一份副本。所以现在我需要对它们进行分组，这样每条记录只会出现一次......

Answer 1

首先您需要将self.filter 更改为self.get_queryset().filter() 或在manager 中创建一个单独的方法。

  def get_queryset(self):
     return super(CourseManager, self).get_queryset()

在管理器中创建一个函数

  def get_courses(self,organisation): 
     return self.get_queryset.filter(student__oraganisation=organisation)

这应该返回students，您不需要调用.all() - 过滤后的qs 会返回它找到的所有对象。

编辑

试试这个：

class CourseManager(models.Manager):

  def get_queryset(self):
     return super(CourseManager, self).get_queryset()

  def get_courses(self, organisation):
    return self.get_queryset().filter( \
      enrollments__student__organisation_id=organisation).distinct()

更新 2

你可以试试from django.db.models import Qhttps://docs.djangoproject.com/en/3.0/topics/db/queries/

或使用 annotate 基于此答案 Get distinct values of Queryset by field 过滤掉每个学生，使其出现一次。