时间戳

首页 > 解决方案 > Django：在组上过滤 ListView / 在列表上迭代查询集

python - Django：在组上过滤 ListView / 在列表上迭代查询集

问题描述

鉴于一个用户可以属于多个组，我想问一下如何在以下场景中过滤组。

在 Django 中，我有一个 ListView 显示产品列表。

Inspector 没有超级用户身份或员工身份，应该被视为属于两个组 ['Company_1', 'Company_2'] 的普通用户

所需功能：

Worker_1 在应用程序中注册了自己制造的产品，可以看到 Company_1 制造的所有产品的列表。

Inspector 可以查看 Company_1 和 Company_2 的所有产品

然而，在下面的代码尝试中，Inspector 看不到任何产品。

由于 Inspector 属于 ['Company_1', 'Company_2'] 我需要一种方法来检查列表中的每个组成员资格与注册产品的组。请注意，以下被简化为两组，但实际上是 x 个组和用户。

在模型.py

from django.contrib.auth.models import User

class Product(models.Model):
    worker = models.ForeignKey(User, blank=True, on_delete=models.CASCADE)
    company = models.CharField(max_length=100, null=True, blank=True)

    def save(self, *args, **kwargs):
        self.company = list(self.worker.groups.all().values_list('name', flat=True))
        super(Product, self).save(*args, **kwargs)

在views.py中

class ProductListView(LoginRequiredMixin, ListView):
    model = Product
    template_name = 'product_list.html'

    def get_queryset(self):
        queryset = super(ProductListView, self).get_queryset()
        return queryset.filter(company__contains=list(self.request.user.groups.all().values_list('name', flat=True)))

标签： pythondjangodjango-querysetusergroups

解决方案

由于您对检查器的定义很明确，因此我们可以在 get_queryset 函数中添加一些条件。这是一个伪代码：

from django.contrib.auth.models import User, Group

class ProductListView(LoginRequiredMixin, ListView):
    model = Product
    template_name = 'product_list.html'

    def get_queryset(self):
        Company1 = Group.objects.get(name='Company1')
        Company2 = Group.objects.get(name='Company2')
        inspectors = User.objects.filter(groups=Company1 and groups=Company2).distinct() 
        if self.request.user in inspectors:
            return queryset.filter(company=Company1 or company=Company2).distinct().values_list('name', flat=True)))
        return queryset.filter(company__contains=list(self.request.user.groups.all().values_list('name', flat=True)))

我希望这会有所帮助，如果它不起作用，请告诉我:)

推荐阅读