Django - 针对特定情况的 prefetch_related 不起作用答案

【问题标题】：Django - prefetch_related for specific situation doesn't workDjango - 针对特定情况的 prefetch_related 不起作用
【发布时间】：2018-05-19 08:52:38
【问题描述】：

我有三个模型：Trip、City 和 GoogleApiCityCache。

如果我想知道城市的名称，我在City 对象上使用@property，它会尝试获取当前语言的GoogleApiCityCache。

我想减少模板中的查询次数：

{% for trip in trips %}
    {{ trip.city.formatted_address }}
{% endfor %}

现在，它执行新的数据库查询以在每次迭代中找到GoogleApiCityCache。我想预取这些缓存。

所以属性formatted_address 必须找到一个GoogleApiCityCache 并从此对象中获取formatted_address。

尝试使用prefetch_related，但没有帮助。

def profile(request, slug):
    owner = User.objects.filter(userprofile__slug=slug).prefetch_related('trips__city__city_caches').first()
    trips = Trip.objects.filter(user=owner).select_related('city').prefetch_related('city__city_caches')
    return render(request, 'profiles/profile/profile.html', {'owner': owner,'trips':trips})

我的模型：

class Trip(models.Model):
    user = models.ForeignKey('auth.User', related_name='trips')
    city = models.ForeignKey('locations.City', related_name='trips')
    date_from = models.DateField()
    date_to = models.DateField()
    detail = models.TextField(null=True, blank=True)
    # participants = models.ManyToManyField('auth.User', related_name='participating_on_trips', blank=True)
    objects = TripManager()

class City(models.Model):
    ...
    def get_cache_by_lang(self, lang=None, force_refresh=False):
        if not lang:
            lang = get_language() or 'en'
        cache, requested = GoogleApiCityCache.objects.get_or_request(city=self, lang=lang)
        return cache

    @property
    def formatted_address(self):
        return self.get_cache_by_lang().formatted_address

class GoogleApiCityCacheManager(models.Manager):
    def _request_create(self, city, lang):
        """ Create cache (wasn't found) """
        ....
        return cache

    def get_or_request(self, city, lang):
        created = False
        try:
            cache = GoogleApiCityCache.objects.get(city=city, language_code=lang)
        except GoogleApiCityCache.DoesNotExist:
            cache = self._request_create(city, lang)
            created = True
        return cache,created

class GoogleApiCityCache(models.Model):
    language_code = models.CharField(max_length=5)
    api_json_response = JSONField(null=True, blank=True)
    city = models.ForeignKey('locations.City', on_delete=models.CASCADE, related_name='city_caches')
    objects = GoogleApiCityCacheManager()

    class Meta:
        unique_together = ('city', 'language_code')


    @property
    def formatted_address(self):
        try:
            return self.api_json_response['result']['formatted_address']
        except:
            return None

如您所见，GoogleApiCityCache 代表 city 和 language_code 可以为空。在这种情况下，请求 Google Api，创建一个新的 GoogleApiCityCache 对象并返回。

【问题讨论】：

标签： python sql django orm django-orm

【解决方案1】：

当您prefetch_related('trips__city__city_caches') 时，您正在预取city_caches。这意味着下面的循环不会引起任何额外的查询。

for cache in city.city_caches.all()
    print(cache)

但是，在您的代码中 formatted_address 调用 get_cache_by_lang 再调用 get_or_request 再调用 GoogleApiCityCache.objects.get()，这将导致额外的查询。

为了利用prefetch_related，您需要重写get_cache_by_lang，使其循环通过self.city_caches.all()以找到正确的对象。

【讨论】：