python - 使用查询集将过滤器应用于嵌套的反向外键关系
问题描述
语境
我正在尝试根据反向外键属性的值过滤对象列表。我能够在视图级别解决它,但是使用 ORM 功能解决的其他尝试会导致额外的查询。
我想要的结果是所有对象的查询集,但相关的 fkey 对象在每个对象中被过滤。
样品模型
class Student(models.Model):
name = models.CharField(max_length=128)
class Subject(models.Model):
title = models.CharField(max_length=128)
class Grade(models.Model):
student = models.ForeignKey("Student", related_name="grades", on_delete=models.CASCADE)
subject = models.ForeignKey("Subject", related_name="grades", on_delete=models.CASCADE)
value = models.IntegerField()
鉴于夹具
+------+------------------------+
| name | subject | grade_value |
+------+----------+-------------+
| beth | math | 100 |
| beth | history | 100 |
| beth | science | 100 |
| mark | math | 90 |
| mark | history | 90 |
| mark | science | 90 |
| mike | math | 90 |
| mike | history | 80 |
| mike | science | 80 |
+------+----------+-------------+
期望的结果
我想呈现学生列表,但只包括数学和历史成绩。
例如,也许我想要一个学生列表,但只包括他们成绩的一个子集:
GET students/?subjects=math,history
哪些被过滤可能会在请求中提供,或硬编码。如果可能的话,我们可以将其排除在此问题的范围之外,并假设过滤参数固定为math
和history
。
{
"students": [
{
"name": "beth",
"grades": [
{"subject": "math", "grade": 100 },
{"subject": "history", "grade": 100 },
// Exclude one or more grades - eg.
// science grade not included
]
},
...
]
}
尝试的解决方案
简单过滤器
只是过滤。我想这会过滤所有在列表中有科目的成绩的学生,仅此而已。
queryset = Students.objects.all()\
.prefetch_related("grades")\
.filter(grades__subject__in=["math", "history"])
)
# all grades for each student eg.
...
"grades": [
{"subject": "math", "grade": 100 },
{"subject": "history", "grade": 100 },
{"subject": "science", "grade": 100 },
]
...
子查询
我不太了解子查询的工作原理,但是使用我尝试过的一些示例:
subjects = Subject.objects.filter(
name__in=["math", "history"]
)
queryset = Students.objects.all()\
.prefetch_related("grades")\
.filter(grades__subject__name__in=Subquery(subjects.values("name")))
还有另一个变体:
grades = Grades.objects.filter(
student_id=OuterRef("id"), subject__name__in=["math", "history"]
)
queryset = Students.objects.all()\
.prefetch_related("grades")\
.filter(grades__pk__in=Subquery(grades.values("pk)))
两人都以所有成绩返回学生。
解决方法
该解决方案使用 python 过滤等级。它有效,但我宁愿让它与查询集一起使用
# in view:
serializer = StundentSerializer(queryset, many=True)
response_data = serializer.data
for student in response_data:
student.grades = [g for g in students.grades if g["subject"] in ["math", "history"]]
...
# return response_data
解决方案
您可以使用该Prefetch
对象:https ://docs.djangoproject.com/en/3.0/ref/models/querysets/#django.db.models.Prefetch
例如:
qs = Students.objects.all().prefetch_related(
Prefetch('grades', queryset=Grade.objects.filter(subject__title__in=["math", "history"])
)
qs[0].grades.all()
现在只有数学和历史成绩。
或者,您可以将to_attr='math_history_grades'
参数提供给 Prefetch,然后您将通过以下方式访问成绩:qs[0].math_history_grades.all()
推荐阅读
- c - 使用 struct 作为“数据库”的任意数量单词(字母顺序)的排序算法
- ruby - 如何防止类在 Ruby 中被覆盖?
- linux - pubsubbeat 与 elasticsearch 的集成
- scala - sbt assembly with Junit test fail
- python - 为什么“B”代码没有打印所需的范围?
- cookies - 为什么 cookie 没有在来自子域的请求中传递?
- bash - 如何使用带有awk的match()函数中的变量定义范围'r(n,)'
- angular - 在 Angular .subscribe() 函数中更改关联变量后,HTML 组件未触发
- .net - 在 C# 中,将文件(xml)提交到服务器后,得到具有一些边界标头的响应 XML
- angularjs - AngularJS 以智能方式处理 $http 错误/成功