You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Django ORM查询优化:俱乐部列表接口单查询实现咨询

问题描述

需要为俱乐部列表接口编写Django ORM查询,筛选出两类俱乐部:

  • 当前用户是俱乐部所有者
  • 当前用户是赛事裁判,且对应裁判记录的end_date为空或晚于当前时间

现有模型定义如下:

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(owner=self.request.user) | Q(id__in=(referee_clubs_list)))
    return super().get_queryset()
优化方案

可以通过Django ORM的关联查询(双下划线语法)直接在Club的查询中整合裁判条件,实现单SQL查询:

from django.utils import timezone

def get_queryset(self):
    if self.request.method == 'GET':
        now = timezone.now()  # 用Django时区工具替代原生datetime,适配项目时区配置
        return Club.objects.filter(
            Q(owner=self.request.user) |
            Q(
                tournament__tournament_referee__user=self.request.user,
                tournament__tournament_referee__end_date__isnull=True
            ) |
            Q(
                tournament__tournament_referee__user=self.request.user,
                tournament__tournament_referee__end_date__gte=now
            )
        ).distinct()  # 去重,避免同一俱乐部因多条符合条件的裁判记录重复返回
    return super().get_queryset()

核心说明:

  1. 关联路径:通过Club→Tournament(反向关联)→TournamentReferee(借助related_name)的链式关联,直接在Club查询中嵌套裁判条件
  2. 条件整合:把裁判的两个有效条件(end_date为空/未过期)拆分为独立Q对象,与所有者条件用OR连接
  3. 去重处理:添加.distinct()防止同一俱乐部因关联多条合规裁判记录被重复输出
  4. 时区适配:使用timezone.now()替代原生datetime.now(),符合Django项目的时区规范

也可以将裁判的两个条件合并,让代码更简洁:

def get_queryset(self):
    if self.request.method == 'GET':
        now = timezone.now()
        referee_q = Q(
            tournament__tournament_referee__user=self.request.user,
            Q(tournament__tournament_referee__end_date__isnull=True) | Q(tournament__tournament_referee__end_date__gte=now)
        )
        return Club.objects.filter(Q(owner=self.request.user) | referee_q).distinct()
    return super().get_queryset()

内容的提问来源于stack exchange,提问作者Anna Berezko

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.06 05:25:09