Django Query 在基于通用类的 UpdateView 中重复了 2 次

发布于04月22日

我有一个项目模型

class Project(models.Model)
    name = models.CharField(max_length=255)
    members = models.ManyToManyField(User, related_name="members")

我使用一个基于类的更新视图来限制那些属于项目成员的用户.

class ProjectUpdateView(LoginRequiredMixin, UserPassesTestMixin, UpdateView):
    model = Project
    fields = ["name"]

    def test_func(self):
        members = self.get_object().members.all()
        if members.contains(self.request.user):
            return True
        return False

在通过Djang Debug工具栏查看SQL查询之后，我看到该查询重复了两次.

推荐答案

只需过滤queryset，只检索request.user是Project成员的项目，因此:

class ProjectUpdateView(LoginRequiredMixin, UpdateView):
    model = Project
    fields = ['name']

    def get_querset(self):
        return Project.objects.filter(members=self.request.user)

Note:通常使用settings.AUTH_USER_MODEL ^[Django-doc]来表示用户模型比直接使用User model ^[Django-doc]更好.有关更多信息，请参阅referencing the User model section of the documentation.