Формы Django: как отображать в выпадающем списке только объекты, связанные с пользователем - PullRequest
Новая
       31

Формы Django: как отображать в выпадающем списке только объекты, связанные с пользователем

0 голосов
/ 26 ноября 2018

Я ветеринарная больница, с классом Pet и классом Records.У каждого домашнего животного может быть много записей, т.е. каждый раз, когда он посещает больницу, он получает новую запись.

На данный момент в моей форме отображаются все животные, когда-либо связанные с моим приложением (пожалуйста, просмотрите https://i.stack.imgur.com/8j7V8.png).

Я хочу, чтобы отображались только зарегистрированные домашние животные пользователя (зачем Бобу приносить чужую кошку к ветеринару?)

Просмотр для добавления записи: 

@login_required(login_url="/accounts/login/")
def record_create(request):
    #this line retrieves Pets only belonging to the user logged in
    pets = Pet.objects.filter(author=request.user)
    if request.method == 'POST':
        form = forms.CreateRecord(request.POST, request.FILES)
        print(form)
        if form.is_valid():
            instance = form.save(commit=False)
            instance.author = request.user
            instance.save()
            return redirect('records')
    else:
        form = forms.CreateRecord()
    return render(request, 'records/record_create.html', {'form': form,})

forms.py 

class CreateRecord(forms.ModelForm):
    class Meta:
        model = models.Record
        fields = ['feedID', 'amountLeftOver', 'amountDispensed', 'additionalInfo', 'selectPet']

models.py 

class Pet(models.Model):
    petName = models.CharField(max_length=100, default='My Pet')
    petImage = models.ImageField(default='default.png', blank=True)
    author = models.ForeignKey(User, on_delete=models.CASCADE, default=None)

    def __str__(self):
        return self.petName

class Record(models.Model):
    feedID = models.AutoField(primary_key=True)
    dateTime = models.DateTimeField(auto_now_add=True)
    amountLeftOver = models.IntegerField(default='0')
    amountDispensed = models.IntegerField(default='0')
    additionalInfo = models.TextField(default=" ")
    selectPet = models.ForeignKey(Pet, on_delete=models.CASCADE)
    author = models.ForeignKey(User, on_delete=models.CASCADE, default=None)

Как получить из раскрывающегося списка selectPet показ только зарегистрированного Бобадомашние животные?

Спасибо за ваше время!

1 Ответ

0 голосов
/ 26 ноября 2018

Вы можете переопределить метод __init__() вашей формы для передачи дополнительных аргументов, в этом случае вы можете передать пользовательский экземпляр и установить набор запросов для раскрывающегося виджета.

(Не уверено том, как вы связали модель Pet с моделью User, в следующем примере предполагается, что у вас есть внешний ключ) 

class CreateRecord(forms.ModelForm):

    class Meta:
        model = models.record
        fields = ['feedID', 'amountLeftOver', 'amountDispensed', 'additionalInfo', 'selectPet']

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        user = self.kwargs.get('user')
        if user:
            self.fields['selectPet'].queryset = user.pet_set.all()
...