我需要在 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()
您可以通过
查询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()