Django添加员工时遇IntegrityError：外键约束失败问题求助

当通过自定义表单向数据库添加员工数据时，访问/add_emp路径出现IntegrityError，提示外键约束失败。删除db.sqlite3和__pycache__文件后问题仍存在。

问题本质 #

外键约束失败的核心原因是：你提交的部门ID（dept）或角色ID（role）在Department或Role数据表中不存在。当前表单让用户手动输入数字ID，极易出现输入错误，或数据库中根本没有对应的部门/角色数据。

解决方案 #

1. 确认Department和Role表中存在有效数据 #

首先要确保数据库里已有可用的部门和角色记录：

• 运行命令进入Django交互Shell：

python manage.py shell

• 执行以下命令查看现有数据：

from your_app_name.models import Department, Role
# 替换your_app_name为你的实际应用名
print(Department.objects.all())
print(Role.objects.all())

如果输出为空，先通过Shell添加测试数据：

# 添加部门
dept_dev = Department(name="技术部", location="北京")
dept_dev.save()
# 添加角色
role_engineer = Role(name="开发工程师")
role_engineer.save()

2. 改进表单：用下拉选择框替代手动输入ID #

将表单中部门和角色的数字输入框改为下拉选择，直接从数据库读取已有选项，彻底避免手动输入错误：
修改后的表单代码：

<!DOCTYPE html>
<html lang="zh-CN">
<head>
    <meta charset="UTF-8">
    <title>添加员工</title>
    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-1BmE4kWBq78iYhFldvKuhfTAU6auU8tT94WrHftjDbrCEXSU1oBoqyl2QvZ6jIW3" crossorigin="anonymous">
</head>
<body>
<div class="container">
    <h1>添加员工</h1>
    <hr>

<form action="/add_emp" method="post">
    {% csrf_token %}
    <label for="first_name">名字</label>
    <br>
    <input type="text" id="first_name" name="first_name" required><br>

    <label for="last_name">姓氏</label>
    <br>
    <input type="text" id="last_name" name="last_name"><br>

    <label for="salary">薪资</label>
    <br>
    <input type="number" id="salary" name="salary" required> <br>

    <label for="dept">部门</label>
    <br>
    <select id="dept" name="dept" required>
        {% for dept in departments %}
            <option value="{{ dept.id }}">{{ dept.name }}</option>
        {% endfor %}
    </select><br>

    <label for="role">角色</label>
    <br>
    <select id="role" name="role" required>
        {% for role in roles %}
            <option value="{{ role.id }}">{{ role.name }}</option>
        {% endfor %}
    </select><br>

    <label for="bonus">奖金</label>
    <br>
    <input type="number" id="bonus" name="bonus"><br>

    <label for="phone">手机号</label><br>
    <input type="number" id="phone" name="phone"><br>

    <hr>
    <button type="submit" class="btn btn-primary">提交</button>
</form>

</div>

<script src="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/js/bootstrap.bundle.min.js" integrity="sha384-ka7Sk0Gln4gmtz2MlQnikT1wXgYsOg+OMhuP+IlRH9sENBO0LRn5q+8nbTov4+1p" crossorigin="anonymous"></script>
</body>
</html>

同时修改add_emp视图的GET请求逻辑，传递部门和角色数据到模板：

def add_emp(request):
    if request.method == 'POST':
        # 原有POST逻辑不变
        first_name = request.POST['first_name']
        last_name = request.POST['last_name']
        salary = int(request.POST['salary'])
        bonus = int(request.POST.get('bonus', 0))
        phone = int(request.POST.get('phone', 0))
        dept = int(request.POST['dept'])
        role = int(request.POST['role'])
        new_emp = Employee(
            first_name=first_name, 
            last_name=last_name, 
            salary=salary, 
            bonus=bonus, 
            phone=phone, 
            dept_id=dept,
            role_id=role,
            hire_date=datetime.now()
        )
        new_emp.save()
        return HttpResponse("员工添加成功")
    
    elif request.method=='GET':
        # 新增：传递部门和角色数据到模板
        context = {
            'departments': Department.objects.all(),
            'roles': Role.objects.all()
        }
        return render(request, 'add_emp.html', context)
    else:
        return HttpResponse("发生异常！员工未添加成功")

3. 捕获异常，给出友好提示 #

在视图中捕获IntegrityError，当用户输入无效ID时给出明确提示：

from django.db import IntegrityError  # 导入异常类

def add_emp(request):
    if request.method == 'POST':
        try:
            first_name = request.POST['first_name']
            last_name = request.POST['last_name']
            salary = int(request.POST['salary'])
            bonus = int(request.POST.get('bonus', 0))
            phone = int(request.POST.get('phone', 0))
            dept = int(request.POST['dept'])
            role = int(request.POST['role'])
            new_emp = Employee(
                first_name=first_name, 
                last_name=last_name, 
                salary=salary, 
                bonus=bonus, 
                phone=phone, 
                dept_id=dept,
                role_id=role,
                hire_date=datetime.now()
            )
            new_emp.save()
            return HttpResponse("员工添加成功")
        except IntegrityError:
            return HttpResponse("错误：部门或角色ID不存在，请检查输入")
        except ValueError:
            return HttpResponse("错误：请输入有效的数字格式")
    
    elif request.method=='GET':
        context = {
            'departments': Department.objects.all(),
            'roles': Role.objects.all()
        }
        return render(request, 'add_emp.html', context)
    else:
        return HttpResponse("发生异常！员工未添加成功")

4. 确保数据库迁移正确执行 #

删除db.sqlite3和__pycache__后，必须重新执行迁移以重建数据库结构：

python manage.py makemigrations
python manage.py migrate

内容的提问来源于stack exchange，提问作者MUDIT DAHRA