将request.user传递给ModelChoiceFilter中的queryset

时间:2019-01-07 05:47:10

标签: django filter modelchoicefield

我想将request.user传递给ModelchoiceFilter中的查询集,因为应该根据登录用户组来限制选择条目。我尝试过,但是在filters.py中弄乱了效果。

models.py

class Score(models.Model):
    group   = models.ForeignKey(Group, null=True, blank=True, on_delete=models.DO_NOTHING)
    member  = models.ForeignKey(Member, on_delete=models.CASCADE,)
    date    = models.DateField()
    lane    = models.PositiveSmallIntegerField(null=True)
    score   = models.PositiveSmallIntegerField(null=True)

filters.py

class ScoreFilter(django_filters.FilterSet):
    class Meta:
        model = Score
        fields = {'member': ['exact'],
                'date': ['gte','lte'],
                }
    def __init__(self, *args, **kwargs):
        group = kwargs.pop('group', None)
        super(ScoreFilter, self).__init__(*args, **kwargs)
        self.filters['date__gte'].label="Start Date"
        self.filters['date__lte'].label="End Date"
        self.filters['member'].queryset = Member.objects.filter(group=group)

views.py

@login_required(login_url='/login/')
def MemberResult(request, *args, **kwargs):
    for group in request.user.groups.filter(Q(name='mon') | Q(name='wed')):
        group = group
    s_filter = ScoreFilter(request.GET, group=group, queryset=Score.objects.filter(group=group).order_by('-date')

我从该站点尝试了许多与类似问题相关的解决方案,但仍无法解决。

1 个答案:

答案 0 :(得分:1)

尝试一下

s_filter = ScoreFilter(request.GET, user=request.user, queryset=Score.objects.filter(group=group).order_by('-date')

def __init__(self, *args, **kwargs):
    self.user = kwargs.pop('user', None)
    for group in self.user.groups.filter(Q(name='mon') | Q(name='wed')):
        group = group
    super(ScoreFilter, self).__init__(*args, **kwargs)
    self.filters['date__gte'].label="Start Date"
    self.filters['date__lte'].label="End Date"
    self.filters['member'].queryset = Member.objects.filter(group=group)