预取一对多关系django ORM

时间:2017-02-27 22:09:19

标签: python django database orm prefetch

我正在开发一个API函数,它将返回用户朋友的列表以及该用户照片的所有参考。 (虽然我只需要按ord排序的引用)

下面是带有外键的模型。

class Personimage(models.Model):
    photo_uuid = models.CharField(db_column='photoUUID', max_length=100, blank=True,
                                 primary_key=True, default='')
    user_id = models.ForeignKey('Member', db_column='userID', on_delete=models.CASCADE, blank=True,
                               null=True)
    ord = models.IntegerField(db_column='ord', blank=True, null=True)
    reference = models.CharField(db_column='reference', max_length=200, blank=True, null=True)

class Member(AbstractBaseUser):
    user_id = models.CharField(db_column='userID')

class Friends(models.Model):
    id = models.IntegerField(db_column='id', primary_key=True, blank=False, null=False)
    puser = models.ForeignKey('Member', on_delete=models.CASCADE, db_column='puser')
    suser = models.ForeignKey('Member', on_delete=models.CASCADE, db_column='suser',
                              related_name='secondary_friend')

到目前为止,我能够做到这一点:

friends = Friend.objects.filter(puser=cur_user).prefetch_related('suser')

    for r in friends:
        photos = Personimage.objects.filter(user_id=r.suser)
        united.append({'user_id': r.suser.user_id,
                       'name': r.suser.name,
                       'photos': [photo.reference for photo in photos.filter(user_id=r.suser)]})

但是因为你必须为朋友列表中的每个用户调用Personimage表,所以你的图像速度非常慢。我知道有一种方法可以使用prefetch_related或它的某种组合和其他东西来做到这一点,但我找不到如何在网上或我的实验中做到这一点。我得到的最接近的是为朋友中的每个成员返回一张照片。

1 个答案:

答案 0 :(得分:0)

将Friends模型替换为ManyToMany字段' friends'在会员班。 此外,我认为您不需要user_id字段 - 您有默认的' ID'字段,你不需要别名(Django会为你制作)。

class Member(AbstractBaseUser):
  friends = models.ManyToManyField('self')

然后以您想要的形式获取数据:

friends = Member.objects.get(pk=cur_user).friend.all().prefetch_related('personimage_set')
united = []
for friend in friends:
  united.append({'user_id': friend.id, 
                 'name': friend.name, 
                 'photos': list(friend.personimage_set.values_list('reference', flat=True))})