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()
核心说明:
- 关联路径:通过
Club→Tournament(反向关联)→TournamentReferee(借助related_name)的链式关联,直接在Club查询中嵌套裁判条件 - 条件整合:把裁判的两个有效条件(end_date为空/未过期)拆分为独立Q对象,与所有者条件用
OR连接 - 去重处理:添加
.distinct()防止同一俱乐部因关联多条合规裁判记录被重复输出 - 时区适配:使用
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
相关产品推荐
相关产品推荐

