Django模型方法中使用prefetch_related及关联对象存在性判断问题
Django模型关联对象存在性判断问题解答
问题1:在模型方法中结合self使用prefetch_related实现单次查询
prefetch_related是QuerySet层面的方法,无法直接在已有的模型实例(self)上调用。要实现单次查询获取所有关联对象,有两种可行方案:
方案1:获取Person实例时提前预取关联
在业务代码中获取Person对象时,主动用prefetch_related预取关联的phone和laptop,后续调用has_device()时会直接使用缓存数据,不会触发额外查询:
# 业务代码中获取Person实例时 person = Person.objects.prefetch_related('phone_set', 'laptop_set').get(id=xxx) # 调用方法时无额外数据库查询 person.has_device()
方案2:在模型方法中重新查询当前实例并预取
如果无法提前预取关联,可以在has_device()里重新查询当前Person对象并预取关联,这样只会触发一次数据库查询:
class Person(models.Model): # <fields> def has_device(self): # 重新查询当前实例并预取所有关联设备 person_with_devices = Person.objects.prefetch_related('phone_set', 'laptop_set').get(id=self.id) return person_with_devices.phone_set.exists() or person_with_devices.laptop_set.exists()
问题2:判断关联模型是否存在的标准方法
Django中判断关联是否存在的标准高效方式是使用exists(),它仅检查数据库中是否存在匹配记录,不会返回完整的模型实例,性能比first()更优(first()会取出第一条数据并实例化对象)。
修改后的has_device()方法可以直接用exists()实现:
class Person(models.Model): # <fields> def has_device(self): # 直接判断关联是否存在,无额外实例化开销 return self.phone_set.exists() or self.laptop_set.exists()
如果已经通过prefetch_related预取过关联数据,exists()会直接使用缓存的结果,不会再触发数据库查询,兼顾性能和代码简洁性。
内容的提问来源于stack exchange,提问作者Shmuelt
相关产品推荐
相关产品推荐

