'User'对象没有属性'staffUser'

Question

我使用Django一对一字段自定义了用户模型

我的型号.py

class StaffUser(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE)
department = models.ForeignKey(Dept, on_delete=models.RESTRICT)
def __str__(self):
return self.user.username

当上传表格时，我想得到用户的部门

"我的视图.py">

def FileUploadForm(request, pk):
if request.method == 'POST':
form = UploadFileForm(request.POST, request.FILES)
if form.is_valid():
form.save(commit=False)
u = User.objects.get(username=request.user)
form.instance.username = u
folder = Folder.objects.get(id=pk)
department = u.staffUser.department
form.save()
messages.success(request, f'File Successfully uploaded to {folder} in {department}!')
return redirect('home')
else:
form = UploadFileForm()
context = {'pk':pk, 'form':form}
return render(request, "pages/fileup_form.html", context)

但它给出了错误Attribute/file/new/1处出现错误"User"对象没有属性"staffUser">

伙计们，我需要帮助

Answer 1

在StaffUser型号的用户字段中添加related_name

class StaffUser(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE, related_name=staff_user)
department = models.ForeignKey(Dept, on_delete=models.RESTRICT)
def __str__(self):
return self.user.username

现在，您可以通过用户对象上的staff_user访问StaffUser

Answer 2

您需要用小写指定反向，因此staffuser而不是staffUser。

或者，您可以指定用于调用反向访问器的相关名称，如下所示，以代替some_name。

user = models.OneToOneField(User, on_delete=models.CASCADE, related_name=some_name)