django orm 优化查询

问题描述 投票:0回答:1

我需要在 django orm 中查询俱乐部列表端点。与用户是俱乐部所有者或用户是锦标赛裁判的俱乐部(结束日期为空,或者结束日期比现在更新)。

我有模型:

class User(AbstractBaseUser):
    id = models.UUIDField(default=uuid.uuid4, primary_key=True, max_length=128)
    ...

class Club(models.Model):
    id = models.UUIDField(default=uuid.uuid4, primary_key=True, max_length=128)
    owner = models.ForeignKey(User, on_delete=models.SET_NULL, blank=True, null=True)

class Tournament(models.Model):
    id = models.UUIDField(default=uuid.uuid4, primary_key=True, max_length=128)
    club = models.ForeignKey(Club, on_delete=models.CASCADE)

class TournamentReferee(models.Model):
    id = models.UUIDField(default=uuid.uuid4, primary_key=True, max_length=128)
    tournament = models.ForeignKey(Tournament, on_delete=models.CASCADE, related_name='tournament_referee')
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    end_date = models.DateTimeField("access expire date", blank=True, null=True)

我像这样创建了查询,但也许有更好的方法来使用单个查询来完成它?

    def get_queryset(self):
        if self.request.method == 'GET':
            now = datetime.datetime.now()
            referee_clubs_list = TournamentReferee.objects.filter(Q(user=self.request.user), (Q(end_date__isnull=True) | Q(end_date__gte=now))).values_list('tournament__club_id')
            return Club.objects.filter(Q(user=self.request.user) | Q(id__in=(referee_clubs_list)))
        return super().get_queryset()
django django-orm
1个回答
0
投票

您可以通过

查询
from django.db.models import Q
from django.utils.timezone import now


def get_queryset(self):
    if self.request.method == 'GET':
        return Club.objects.filter(
            Q(user=self.request.user)
            | (
                Q(tournament__tournament_referee__user=request.user)
                & (
                    Q(tournament__tournament_referee__end_date=None)
                    | Q(tournament__tournament_referee__end_date__gte=now())
                )
            )
        ).distinct()
    return super().get_queryset()
© www.soinside.com 2019 - 2024. All rights reserved.