如何在 Django 中检查用户是否是模型对象的成员?

问题描述

我在model.py中写了这个模型

class Group(models.Model):
   MEMBERS_NUMBER = [
      (2,'Two people'),(5,'3 to 5'),(10,'5 to 10'),]
   group_name = models.CharField(max_length=50)
   slug = models.SlugField(max_length=50,unique=True)
   members = models.IntegerField(choices=MEMBERS_NUMBER)
   people = models.ManyToManyField('Profile',through='Membership',blank = True)
   summary = models.TextField(
      'Summary and indeas',max_length=500,help_text='Say what you want to do with your study or jub partner.'
   )
   date_created = models.DateTimeField(default=timezone.Now)

class Profile(models.Model):
  user = models.OnetoOneField(User,on_delete=models.CASCADE)
  birth_date = models.DateField(blank=True,null=True)
  profile_picture = models.ImageField(default = 'default.jpeg',upload_to = 'profile_pics',null=True,blank = True)
  date_joined = models.DateTimeField(default=timezone.Now,blank = True)

class Membership(models.Model):
  group = models.ForeignKey(Group,on_delete=models.CASCADE)
  members = models.ForeignKey(Profile,on_delete=models.CASCADE)

然后我为 Group 模型的每个对象创建了基于类的通用视图,但我想限制仅访问该 Group 对象中的配置文件,这是我的基于类的视图:

class GroupDetail(UserPassesTestMixin,generic.DetailView):
  model = models.Group
  context_object_name = 'group'
  template_name = 'projects/group_detail.html'

  def test_func(self):
    return self.request.user.profile.group_set.filter(group_name='My-Django-Group')

  def get_queryset(self):
    return models.Group.objects.all()

test_func(self) 检查用户是否是“My-Django-Group”的成员,然后向用户显示组详细信息,但我想获取我们在其中的对象名称,我对此进行了测试:

def test_func(self):  
return self.request.user.profile.group_set.filter(group_name=self.group.group_name)

但是没有用

解决方法

您可以使用以下方法过滤查询集:

from django.contrib.auth.mixins import LoginRequiredMixin

class GroupDetail(LoginRequiredMixin,generic.DetailView):
  model = models.Group
  context_object_name = 'group'
  template_name = 'projects/group_detail.html'

  def get_queryset(self):
    return super().get_queryset().filter(
        people__user=self.request.user
    )

如果用户不是群组成员,他们将收到 HTTP 404 响应。


注意:您可以将视图限制为基于类的视图,仅允许使用 LoginRequiredMixin mixin [Django-doc]