在Django中获取反向外键的反向外键,同时仍利用prefetch_related

时间:2018-11-01 15:55:40

标签: django orm django-rest-framework foreign-keys

我有以下模型(大大简化了这些模型):

class Job(models.Model):
    name = models.CharField(max_length=255)

    def get_date_details(self):
        pass
        #return a list of all jobdate --> details


class JobDate(models.Model):
    job = models.ForeignKey(Job, on_delete=models.CASCADE, related_name='dates')
    date = models.DateField()
    # ... and a bunch of other fields


class JobDateDetail(models.Model):
    job_date = models.ForeignKey(JobDate, on_delete=models.CASCADE, related_name='details')
    detail = models.CharField(max_length=255)
    # ... and a bunch of other fields

并且定期需要处理数据库中所有作业的JobDateDetails(数千个)。我知道我可以使用以下查询来预取所有数据:

q = Job.objects.all().prefetch_related('dates', 'dates__details')

但是,我不确定如何最好地利用Job.get_date_details()中的预取数据。一种选择是做类似的事情:

class Job(models.Model):
    def get_date_details(self):
        details = []
        for job_date in self.dates.all():
            details += [detail for detail in job_date.details.all()]
        return details

但是,我怀疑可能存在一种直接查询所有JobDateDetail对象的方法。需要注意的一件事-如前所述,我意识到我可以朝另一个方向一次性获取所有细节,但是在这种情况下,我使用的是Django Rest Framework,并且需要将这些细节组装为Job序列化程序中的字段,要求我朝这个方向走。有想法吗?

1 个答案:

答案 0 :(得分:1)

我输入了一些随机数据来演示查询。这是两个选项:

  • 第一个在表级别, 1db命中

    In [23]: Job.objects.all().values_list('id', 'dates__details__detail')                                                                                                                                             
    Out[23]: <QuerySet [(1, '132'), (1, '4324gg'), (1, 'hrthrthrth'), (1, 'erhehrgnfgnmfgghmn'), (1, 'herhehrnfn'), (1, 'erg eg cvb dfg vb'), (1, 'greg egr erg erg erg'), (1, 'ewrg erg db cvb'), (2, None), (3, None)]>
    
  • 第二个是行级 2db命中(一个用于获取对象,另一个用于获取其日期详细信息):

    In [36]: Job.objects.all().first().dates.all().values_list('details__detail')                                                                                                                                      
    Out[36]: <QuerySet [('132',), ('4324gg',), ('hrthrthrth',), ('erhehrgnfgnmfgghmn',), ('herhehrnfn',), ('erg eg cvb dfg vb',), ('greg egr erg erg erg',), ('ewrg erg db cvb',)]>
    

如果从模型方法self.dates.all().values_list('details__detail')中调用,则第二个等效于get_date_details,并且在 1db命中时可以带来以下结果:

In [30]: from itertools import chain
In [31]: list(chain.from_iterable( 
    ...:     Job.objects.all().first().dates.all().values_list('details__detail') 
    ...: ))        
Out[31]: 
['132',
 '4324gg',
 'hrthrthrth',
 'erhehrgnfgnmfgghmn',
 'herhehrnfn',
 'erg eg cvb dfg vb',
 'greg egr erg erg erg',
 'ewrg erg db cvb']

因此,您的模型方法可以变为:

class Job(models.Model):
    def get_date_details(self):
        return (
            list(chain.from_iterable(
                self.dates.all().values_list('details__detail')
            ))                                                                                                           
        )