Как запросить все общие отношения в Джангу

Question

Я делаю корм для видов деятельности, но я не могу выяснить, как избежать N + 1 запросов.

В основном, команда или место имеют подписчики / мероприятия, каждый пользователь является последователем, который имеет корм для действия, вытянутые из команд / мест, которые они следуют.

фрагменты из моих моделей:

class Follower(models.Model):
    user = models.ForeignKey(User)
    content_type = models.ForeignKey(ContentType)
    object_id = models.PositiveIntegerField()
    content_object = GenericForeignKey('content_type', 'object_id')

class Activity(models.Model):
    ...
    content_type = models.ForeignKey(ContentType)
    object_id = models.PositiveIntegerField()
    content_object = GenericForeignKey('content_type', 'object_id')

class Team(models.Model):
    ...
    followers = GenericRelation('Follower')
    activities = GenericRelation('Activity')

class Place(models.Model):
    ...
    followers = GenericRelation('Follower')
    activities = GenericRelation('Activity')

.

Я использую DRF и иметь базовый сериал для активности, я думаю, что сложная часть находится внутри вида:

 class ActivityViewSet(viewsets.ModelViewSet):

     def list(self, request):
         #N+1 query
         activities = [f.content_object.activities.all() for f in request.user.follower_set.all()]

         ...

.

Так что это работает, но есть ли лучший способ сделать этот запрос?

Answer 1

def list(self, request):
    items = {}
    for f in request.user.follower_set.all():
        items.setdefault(f.content_type_id, []).append(f.object_id)
    activities = Activity.objects.get_empty_queryset() # get_empty_query_set in <1.5
    for k, v in items.iteritems():
        activities |= Activity.objects.filter(content_type=k, object_id__in=v)

.

Это сводится к 2 запросам.Он использует тот факт, что если генеракодицетагкод и генеракодицетагCode связаны с тем же объектом (командой или местом), генеракодицетагкод и генеракодицетагкод одинаковы.Маленькая обработка Python и TADA у вас есть все мероприятия, связанные с текущим пользователем в 2 запросах.