问题描述
我在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]。