Передача request.user в набор запросов в ModelChoiceFilter - PullRequest
Новая
       24

Передача request.user в набор запросов в ModelChoiceFilter

0 голосов
/ 07 января 2019

Я хочу передать request.user в набор запросов в ModelchoiceFilter, потому что выбор записи должен быть ограничен в соответствии с группой пользователей для входа. Я пытаюсь, но получение kwargs в 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 голосов
/ 07 января 2019

Попробуйте это 

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)
...