Question

我正在创建一个注册系统。我正在使用django的默认用户模型来存储用户详细信息，并将其对象传递到名为“雇员”的模型中，该模型存储了他们的部门，经理和其他信息。这就是我所拥有的。

models.py：

from django.contrib.auth.models import User

class Employee(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='user')
....

forms.py：

from users.models import Employee
from django.contrib.auth.models import User
from django.forms import ModelForm
from django.contrib.auth.forms import UserCreationForm


class AddNewUserForm(UserCreationForm):

    class Meta:
        model = User
        fields = ('username', 'first_name', 'last_name', 'email', 'password1', 'password2')


class AddNewEmployeeForm(ModelForm):

    class Meta:
        model = Employee
        fields = {'department', 'manager', 'total_leaves'}

add_new_user.html：

<form method="POST">
    {% csrf_token %}
    {{ form_user.as_p }}
    {{ form_emp.as_p }}

       <div class="text-center">
          <button type="submit" class="btn btn-success">Submit</button>
       </div>
  </form>

这就是我在views.py中处理数据的方式：

class AddNewUser(TemplateView):
    template_name = 'LMSAdmin/add_new_user.html'

    def get(self, request, *args, **kwargs):
        form_user = AddNewUserForm()
        form_emp = AddNewEmployeeForm()
        return render(request, self.template_name, {'form_user': form_user, 'form_emp': form_emp})

    def post(self, request):
        form_user = AddNewUserForm(request.POST)
        form_emp = AddNewEmployeeForm(request.POST)

        if form_user.is_valid():
            emp_info = form_emp.save(commit=False)
            emp_info.leaves_taken = 0
            emp_info.user = form_user.save()
            emp_info.save()
            messages.success(request, "Success!")
       else:
           messages.error(request, form_user.errors)
           messages.error(request, form_emp.errors)

       return redirect("add_new_user")

问题在于，运行post方法时，它将尝试从头开始将条目添加到Employee表中；假设其中没有任何条目。但是我在该表中有条目，并且出现以下错误：

重复的键值违反了唯一约束“ users_employee_pkey” 详细信息：密钥（id）=（6）已经存在。

它给了我这个错误，因为我的表中已经有6个用户。之后，当我第七次尝试使用此表单创建用户时，它可以正常工作。有没有一种方法可以查看表中已经有多少个条目，并自动将其递增1以避免此错误？任何帮助将不胜感激。

Answer 1

您可以手动设置自己的主键字段

class Employee(models.Model):
    employee_id = models.IntegerField(primary_key=True)

，然后您可以在保存过程中计算并设置主键。只需输入最高employee_id，然后将其递增1。例如：

from django.db.models import Max

# double underscores in employee_id__max
max_employee_id = Employee.objects.all().aggregate(Max('employee_id'))['employee_id__max']

ModelForm根据最后一个条目的主ID在表中添加一个条目的主ID

1 个答案: