如何使用.filter()/.get()查看一个值是否在Django中验证的外部模型中



我从一个模型(EmployeeWorkAreaLog(中创建了一个表单,用户可以在其中输入他们的Employee#,一旦他们按enter/Leave,它就会在数据库中搜索一个单独的模型(Salesman(,以确保这个人是有效的员工,如果不是,则会从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():

为什么会出现这种错误?是否有更有效/更好的方法来检查员工的团队?

型号.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

我过滤它的部分,因此只有Salesman的员工#在小部件属性下,如下所示

表单.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

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

get((返回单个Salesman对象,filter((返回查询集。

要检查给定employee_number的销售人员和get((的团队"WF"是否存在:

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

最新更新