过滤 Django JSONField 中的列表项

Question

这是我正在研究的 Django 模型的简化版本。

class Course(models.Model):
    infos = JSONField()

信息 JSONField 如下所示：

infos = {
    category: "Maths",
    students: [
        {
            name: "Alice",
            result: 8
        },
        {
            name: "Bob",
            result: 12
        }
    ]
}

# (students can have 0 -> n items)

我正在尝试获取在任何课程中至少获得 10 分的所有学生的列表（结果 >= 10）。 但是，我无法找到相应地过滤 QuerySet 的方法。

我正在尝试做这样的事情：

(Course.objects.filter(students__result__gte=10)
               .values_list('students', flat=True))

但由于学生是一个列表，我无法直接访问每个项目的结果属性。此外，我认为它不会从结果中排除“Alice”对象。

我想得到这样的结果：

items = [
    {
        name: "Bob",
        result: 12
    }
]

如果有办法将过滤的学生和课程类别联系起来，则可以加分：

items = [
    {
        category: "Maths",
        students: [
            {
                name: "Bob",
                result: 12
            }
        ]
    }
 ]

我怎样才能达到预期的效果？

Answer 1

通过使用原始 SQL，我设法获得了结果。

select array_to_json(array_agg(students_array))
FROM
    course_table,
    json_array_elements(CAST(course_table.infos->>'students' as json)) students_array
WHERE
    CAST(students_array->>'result' as integer) >= 10
;

结果

sample_db=# SELECT version();
PostgreSQL 10.12 (Ubuntu 10.12-0ubuntu0.18.04.1) on x86_64-pc-linux-gnu, compiled by gcc (Ubuntu 7.4.0-1ubuntu1~18.04.1) 7.4.0, 64-bit

sample_db=# select infos from course_table;
{"category": "Maths", "students": [{"name": "Alice", "result": 8}, {"name": "Bob", "result": 12}]}
{"category": "Science", "students": [{"name": "Jerin", "result": 8}, {"name": "George", "result": 12}]}
{"category": "Physics", "students": [{"name": "Vivek", "result": 17}, {"name": "Osama", "result": 6}]}

sample_db=# select array_to_json(array_agg(students_array))
sample_db-# FROM
sample_db-# course_table,
sample_db-#  json_array_elements(CAST(course_table.infos->>'students' as json)) students_array
sample_db-# WHERE
sample_db-#  CAST(students_array->>'result' as integer) >= 10
sample_db-# ;
[{"name": "Bob", "result": 12},{"name": "George", "result": 12},{"name": "Vivek", "result": 17}]

---

这个原始 SQL 可以被 Django 执行为，

raw_sql = """
select 1 as id, array_to_json(array_agg(students_array)) as result
FROM
    course_table,
    json_array_elements(CAST(course_table.infos->>'students' as json)) students_array
WHERE
    CAST(students_array->>'result' as integer) >= 10
;
"""
qs = Course.objects.raw(raw_sql)
for i in qs:
    print(i.result)

Answer 2

如果您使用的是 PostgreSQL 数据库，则可以切换到 HStoreField（[docs here][1]）。 HStoreField 将允许您准确使用您提到的查找 (students__result__gte)。否则，最具 Django 风格的方法是创建一个学生模型，如 Arakkal 在您的问题下的 cmets 中建议的那样。

编辑：这是最简单的过滤方式，因为所有地图/字典都相同（如果您真的想要与 JSON 相同的结构）：

class Course(models.Model):
    pass

class Info(models.Model):
    course = models.ForeignKey(Course, related_name='infos')
    category = models.CharField(max_length=128)
    # students = models.HStoreField()  # available in postgresql

class Student(models.Model):
    """If not using PostgreSQL, also more Django-esque."""
   info = models.ForeignKey(Info, related_name='students')

   name = models.CharField(max_length=128)
   result = models.IntegerField()

我正在尝试获取在任何课程中至少获得 10 分的所有学生的列表（结果 >= 10）

鉴于您的目标和我对逻辑关系的直觉，这将是我的首选结构：

class Course(models.Model):
    students = models.ManyToManyField(Student, through=Result)
    category = models.CharField(max_length=128)

class Student(models.Model):
    name = models.CharField(max_length=128)

class Result(models.Model):
    course = models.ForeignKey(Course, on_delete=models.deletion.CASCADE)
    student = models.ForeignKey(Student, on_delete=models.deletion.CASCADE)
    grade = models.IntegerField()

这允许：Result.objects.filter(grade__gte=10).values_list('student__name', flat=True)

Answer 3

您可以尝试为您的课程班级定义custom manager。这将使您能够以 Django-ORM 样式进行您想要的查询，而无需使用原始 SQL。

PS：- 如果适合您的需要，我将使用更多代码和详细信息编辑此答案。