0

我正在使用 Django 过滤器,并且我希望其中一个字段 ( supervisor) 成为ChoiceFilter选择来自模型的对象的位置。最有效的方法是什么?我尝试关注这篇文章,但无论我改变了什么(目前cannot unpack non-iterable int object),都会不断出错。

# models.py
class people(models.Model):
    namelast = models.CharField(max_length=100, verbose_name='Last Name')
    namefirst = models.CharField(max_length=100, verbose_name='First Name')
    supervisor = models.ForeignKey('self', blank=True, null=True, on_delete=models.SET_NULL, verbose_name='Supervisor')
    
    def __str__(self):
        return "%s %s" % (self.namefirst, self.namelast)

# filters.py
class office_filter(django_filters.FilterSet):
    supervisor = django_filters.ChoiceFilter(choices=[], lookup_expr='icontains', label='Supervisor')
    # other fields

    class Meta:
        model = people
        fields = ['namelast', 'namefirst', 'supervisor']

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        try:
            self.filters['supervisor'].extra['choices'] = [x for x in
                people.objects.all().values_list('supervisor', flat=True).distinct()]
        except (KeyError, AttributeError):
            pass

目标是使该supervisor字段成为一个很好的菜单,其中包含已添加为people模型中的主管的所有人员。

4

2 回答 2

1

我不是 100% 肯定,但你能试试吗?

[people.objects.get(pk=x[0]) for x in people.objects.all().values_list('supervisor', flat=True).distinct()]

或者

[people.objects.get(id=x[0]) for x in people.objects.all().values_list('supervisor', flat=True).distinct()]

在您提到的链接中,我相信 DatedResource.objects.all().values_list('date', flat=True).distinct()) 返回一个字符串数组

在您的代码 people.objects.all().values_list('supervisor', flat=True).distinct() 中将返回一个 int 字符串 - 记录的 ID,因为它是外键

于 2020-07-02T18:24:50.417 回答
1

Ananya 的回答帮助我正确了解了该语句返回的内容,但在考虑了错误以及通常如何构造选择列表之后,我意识到我需要它来返回一个元组(而不仅仅是一个值)。这是最终工作的相关代码:

class office_filter(django_filters.FilterSet):
    supervisor = django_filters.ChoiceFilter(choices=[], label='Supervisor')
    #...
        def __init__(self, *args, **kwargs):
            super().__init__(*args, **kwargs)
            try:
                self.filters['supervisor'].extra['choices'] = [(people.objects.get(pk=x).id, people.objects.get(pk=x)) for x in people.objects.all().values_list('supervisor', flat=True).distinct() if x is not None]
            except (KeyError, AttributeError):
                pass

重要的部分是:(people.objects.get(pk=x).id, people.objects.get(pk=x))而不仅仅是people.objects.get(pk=x).

还必须lookup_expr在过滤器字段中删除。

于 2020-07-02T20:10:07.480 回答