DRF get_queryset 检查是否在 ManyToManyField 中
DRF get_queryset check if in ManyToManyField
我有这样一个模型:
class Project(models.Model):
name = models.CharField("Name", max_length=8, unique=True)
status = models.CharField(
"Status",
max_length=1,
choices=[("O", "Open"), ("C", "Closed")],
default="O",
)
description = models.CharField("Description", max_length=3000, default="")
owner = models.ForeignKey(
User, on_delete=models.SET_NULL, null=True, related_name="project_owner"
)
participants = models.ManyToManyField(User, related_name="project_participants", blank=True)
created_at = models.DateTimeField(auto_now_add=True)
现在在这个模型的 ViewSet 中我有一个 get_queryset
方法:
def get_queryset(self):
if self.request.user.is_superuser:
return Project.objects.all()
else:
return Project.objects.filter(owner=self.request.user.id)
所以当我得到一个项目或项目s时,我只搜索一个用户(隐藏在请求中,在JWT中)拥有的项目。但是现在我想搜索用户参与的项目(所以用户出现在 participants
字段列表中)。
我如何在 get_queryset
中做到这一点?
使用 Q
应该是可能的:
Project.objects.filter(
Q(owner=self.request.user.id) |
Q(participants=self.request.user.id)
)
https://docs.djangoproject.com/en/3.2/topics/db/queries/#complex-lookups-with-q-objects
我有这样一个模型:
class Project(models.Model):
name = models.CharField("Name", max_length=8, unique=True)
status = models.CharField(
"Status",
max_length=1,
choices=[("O", "Open"), ("C", "Closed")],
default="O",
)
description = models.CharField("Description", max_length=3000, default="")
owner = models.ForeignKey(
User, on_delete=models.SET_NULL, null=True, related_name="project_owner"
)
participants = models.ManyToManyField(User, related_name="project_participants", blank=True)
created_at = models.DateTimeField(auto_now_add=True)
现在在这个模型的 ViewSet 中我有一个 get_queryset
方法:
def get_queryset(self):
if self.request.user.is_superuser:
return Project.objects.all()
else:
return Project.objects.filter(owner=self.request.user.id)
所以当我得到一个项目或项目s时,我只搜索一个用户(隐藏在请求中,在JWT中)拥有的项目。但是现在我想搜索用户参与的项目(所以用户出现在 participants
字段列表中)。
我如何在 get_queryset
中做到这一点?
使用 Q
应该是可能的:
Project.objects.filter(
Q(owner=self.request.user.id) |
Q(participants=self.request.user.id)
)
https://docs.djangoproject.com/en/3.2/topics/db/queries/#complex-lookups-with-q-objects