Django预取和选择相关

Question

我很难理解Django ORM中的prefetch_related和select_related。我有以下型号：

class City(models.Model):
    name = models.CharField(max_length=35)
    state = models.ForeignKey('states.State', on_delete=models.CASCADE)

class Street(models.Model):
    name = models.CharField(max_length=35)
    building = models.ForeignKey('buildings.Building', on_delete=models.CASCADE)
    city = models.ForeignKey('cities.City', on_delete=models.CASCADE)

现在我的views.py：

cities = City.objects.all()
streets = Street.objects.all()

for city in cities: 
    has_bank = streets.filter(building_id=1, city=city)
    if has_bank:
        city.has_bank = 1

    has_cinema = streets.filter(building_id=2, city=city)
    if has_cinema:
        city.has_cinema = 1

    has_church = streets.filter(building_id=3, city=city)
    if has_church:
        city.has_church = 1

但是现在，每次for循环迭代时，它都会访问数据库3次。我试图提高时间复杂度——现在是3N 2，N是城市的数量，但是我不能理解select_related和prefetch_related。

你能给我一个例子，我该如何改进它，使它不会在for循环中命中数据库3次吗？

Answer 1

选择相关。让我解释一下。我添加了虚拟数据来解释。

class City(models.Model):
    name = models.CharField(max_length=35)
    state = models.ForeignKey('states.State', on_delete=models.CASCADE)

class Street(models.Model):
    name = models.CharField(max_length=35)
    building = models.ForeignKey('buildings.Building', on_delete=models.CASCADE)
    city = models.ForeignKey('cities.City', on_delete=models.CASCADE)

您的城市表应该是。

id    name         state
1   Cityname1      state1  
2   Cityname2        2

你的街头桌子应该是。

id  name   city ..
1   st 1    1
2   stno.2  1
3   st no3  2

如果你的orm查询将是这个。

street = Street.objects.select_related('city')

此查询将两个表合并为一个表。这意味着所有的城市id外键将连接到每个街道id以创建如下新表。它将返回3条记录，因为我们使用的是选择相关的城市，主表在这种情况下是街道，所以它将返回3条记录。主表在所有情况下将首先返回dajngo。

 id   name   city ..  city.id  city.name  city.state
  1   st 1    1         1      Cityname1   state1
  2   stno.2  1         1      Cityname1   state1
  3   st no3  2         2      Cityname2    2

Answer 2

在您的具体情况下，我认为最好使用注释而不是预取：

from django.db.models import Count, Q

cities = City.objects
.annotate(bank_count=Count("street", filter=Q(street__building_id=1)))
.annotate(cinema_count=Count("street", filter=Q(street__building_id=2)))
.annotate(church_count=Count("street", filter=Q(street__building_id=3)))

现在您可以直接使用bank\u count、cinema\u count和church\u count属性：

for city in cities: 
   print(city.bank_count)
   print(city.cinema_count)
   print(city.church_count)

如果要使用prefetch\u related，则需要使用prefetch对象。这允许您过滤预取的对象：

City.objects.prefect_related(
    Prefetch("street_set", queryset=Street.objects.filter(building_id=1), to_attr='has_bank'),
    Prefetch("street_set", queryset=Street.objects.filter(building_id=2), to_attr='has_cinema'),
    Prefetch("street_set", queryset=Street.objects.filter(building_id=3), to_attr='has_church')
)

注意to_attr参数这有助于使用不同的过滤器将同一模型的对象预取到不同的属性。所以你现在可以做：

for city in cities: 
   print(city.has_bank)
   print(city.has_cinema)
   print(city.has_church)