我可以创建一个 Django 查询来返回每个项目的列表吗？

Question

我有 3 个这样的模型：

class Person(models.Model):
    name = models.CharField(max_length=100)

class Place(models.Model):
    name = models.CharField(max_length=100)

# ManyToMany Through Table
class PersonPlace(models.Model):
    person = models.ForeignKey(Person, on_delete=models.CASCADE)
    place = models.ForeignKey(Place, on_delete=models.CASCADE)

PersonPlace 以多对多关系链接 Person 和 Place。

我想要一个数据库查询，它将为我提供每个人 id 的地点列表（每个人访问过的地点列表）。

是否可以通过 ORM 进行聚合，而无需 Python 将其放在一起？

预期回报类似于：{1: [4,5,6], 2: [1,2,5]}

这里的键是用户 ID，值是每个用户访问过的地点 ID。注意：结果不需要是字典，但我认为它会是类似字典的东西。

Answer 1

如果你使用的是postgres，你可以使用ArrayAgg..[Django-doc]:

from django.contrib.postgres.aggregates import ArrayAgg


for person in Person.objects.annotate(places=ArrayAgg("personplace__place")): 
    print(person.__dict__)

这给出了以下输出：

# ommitted unnecessary keys
{'id': 1, 'name': '1', 'places': ['2', '3']}
{'id': 2, 'name': '2', 'places': ['1', '2', '3']}

在后台，这一切都是通过以下查询在您的数据库上完成的：

SELECT 
    "person"."id", 
    "person"."name", 
    ARRAY_AGG("personplace"."place_id" ) AS "places" 
FROM 
    "person" 
LEFT OUTER JOIN 
    "personplace" 
ON 
    ("person"."id" = "personplace"."person_id") 
GROUP BY 
    "person"."id"

Answer 2

这将返回一个人的地点列表

person_object = Person.objects.get(name="admin")
place_objects_ids_query_set = PersonPlace.objects.filter(person=person_object).values_list('place', flat=True)
place_objects_ids_list = list(place_objects_ids_query_set)
print(place_objects_ids_list)

如果您需要很多人的地点列表

persons_objects = Person.objects.all()
place_objects_ids_query_set = PersonPlace.objects.filter(person__in=persons_objects).values_list('place', flat=True)
place_objects_ids_list = list(place_objects_ids_query_set)
print(place_objects_ids_list)