Как использовать .filter () /. Get (), чтобы увидеть, находится ли значение во внешней модели для проверки в Django? - PullRequest
Новая
       28

Как использовать .filter () /. Get (), чтобы увидеть, находится ли значение во внешней модели для проверки в Django?

0 голосов
/ 05 ноября 2019

Я сделал форму из модели (EmployeeWorkAreaLog), в которой пользователь может ввести свой Employee #, и после нажатия Enter / Leave он ищет отдельную модель (продавец) в БД, чтобы убедиться, что этот человек является действительным сотрудником,в противном случае на html-стороне появится сообщение о том, что оно недействительно.

Теперь я пытаюсь добавить другое ограничение, например, если сотрудник не входит в команду WF,также выдаст ошибку и скажет связаться с менеджером. Я пытался сделать это из форм, но я продолжаю получать эту ошибку:

int() argument must be a string, a bytes-like object or a number, not 'Salesman'

и согласно трассировке, это происходит от if Salesman.objects.get(id=employee_number).count():

Почемуэта ошибка происходит? Есть ли более эффективный / лучший способ проверить команду сотрудника?

models.py 

class EmployeeWorkAreaLog(TimeStampedModel, SoftDeleteModel, models.Model):
    employee_number = models.ForeignKey(Salesman, on_delete=models.SET_NULL, null=True, blank=False)
    ...

    def __str__(self):
        return self.employee_number

Часть, где я фильтрую ее, поэтому только # сотрудникаот продавца находится под виджетом attrs, как показано ниже

forms.py 

class WarehouseForm(AppsModelForm):
    class Meta:
        model = EmployeeWorkAreaLog
        widgets = {
            'employee_number': ForeignKeyRawIdWidget(EmployeeWorkAreaLog._meta.get_field('employee_number').remote_field, site, attrs={'id':'employee_number_field'}),
        }
        fields = ('employee_number', 'work_area', 'station_number')

    def clean_employee_number(self):
        employee_number = self.cleaned_data.get('employee_number')

        if Salesman.objects.filter(id=employee_number).team is 'WF':
            raise forms.ValidationError(_("Employee not valid, please contact manager"))
        return employee_number

views.py 

class EnterExitArea(CreateView):
    model = EmployeeWorkAreaLog
    template_name = "operations/enter_exit_area.html"
    form_class = WarehouseForm

    def form_valid(self, form):
        emp_num = form.cleaned_data['employee_number']
        area = form.cleaned_data['work_area']
        station = form.cleaned_data['station_number']

        if 'enter_area' in self.request.POST:
            form.save()
            return HttpResponseRedirect(self.request.path_info)

        elif 'leave_area' in self.request.POST:
            form.save()
            return HttpResponseRedirect(self.request.path_info)

1 Ответ

0 голосов
/ 05 ноября 2019

get () возвращает один объект Salesman, а filter () возвращает набор запросов.

Чтобы проверить, существует ли продавец с указанным номером employee_ и командой 'WF' с get (): 

def clean_employee_number(self):
    employee_number = self.cleaned_data.get('employee_number')

    try:
        salesman = Salesman.objects.get(id=employee_number, team='WF')
    except Salesman.DoesNotExist:
        raise forms.ValidationError(_("Employee not valid, please contact manager"))
    return employee_number

с фильтром (): 

  def clean_employee_number(self):
    employee_number = self.cleaned_data.get('employee_number')

    salesman = Salesman.objects.filter(id=employee_number, team='WF').first()
    if not salesman:
        raise forms.ValidationError(_("Employee not valid, please contact manager"))
    return employee_number
...