Django新增User时，如何检查关联的Device是否已被占用？

这问题我之前做Django项目的时候刚好碰到过，给你几个实用的解决方案，从底层约束到业务层检查都覆盖到，还能解决并发场景的坑：

1. 模型层添加唯一性约束（最稳妥，从根源避免问题） #

既然要求一个Device只能被一个User关联，最直接的方式是给User的device字段加上unique=True。这样数据库层面会强制唯一性，就算代码层面漏了检查，数据库也会直接报错阻止重复关联，从根源杜绝问题：

class User(models.Model):
    id = models.CharField(max_length=10, primary_key=True)
    # 新增 unique=True 约束
    device = models.ForeignKey(Device, blank=True, null=True, on_delete=models.SET_NULL, unique=True)
    def __str__(self):
        return self.id

如果后续有需求要允许一个Device关联多个User，再去掉这个约束即可，当前场景下这个方案是最省心的。

2. 重写模型的clean方法（覆盖所有创建/修改场景） #

如果不想修改字段约束，或者需要更灵活的检查逻辑，可以在User模型里重写clean方法，在保存前自动验证Device是否已被占用：

from django.core.exceptions import ValidationError

class User(models.Model):
    id = models.CharField(max_length=10, primary_key=True)
    device = models.ForeignKey(Device, blank=True, null=True, on_delete=models.SET_NULL)
    def __str__(self):
        return self.id

    def clean(self):
        # 先调用父类的clean方法处理基础验证
        super().clean()
        # 仅当当前用户要关联的设备不为空时才检查
        if self.device:
            # 排除当前用户（避免修改用户自身时误判），检查是否有其他用户关联了该设备
            is_occupied = User.objects.exclude(id=self.id).filter(device=self.device).exists()
            if is_occupied:
                raise ValidationError(f"设备 {self.device.id} 已经被其他用户占用了！")

这个方法的优势是：不管你是通过Django Admin、表单还是直接用代码创建用户，只要触发了full_clean()（Django Admin和ModelForm会自动调用），就会执行检查。如果是自己写代码创建用户，记得手动调用user.full_clean()再保存，不然不会触发这个验证。

3. 在ModelForm中做表单验证（适合前端提交场景） #

如果是通过前端表单让用户提交数据，可以在ModelForm的字段验证方法里做检查，这样能把错误信息直接返回给前端，提升用户体验：

from django import forms
from .models import User

class UserForm(forms.ModelForm):
    class Meta:
        model = User
        fields = ['id', 'device']

    def clean_device(self):
        # 获取表单提交的device数据
        device = self.cleaned_data.get('device')
        if device:
            # 排除当前正在编辑的用户（如果是修改用户的场景）
            current_user_id = self.instance.id
            is_occupied = User.objects.exclude(id=current_user_id).filter(device=device).exists()
            if is_occupied:
                raise forms.ValidationError(f"设备 {device.id} 已被占用，请选择其他设备")
        return device

用户在前端提交表单时，如果选了已被占用的设备，会立刻看到错误提示，不用等到后端处理完再跳转。

4. 视图层手动检查（适合自定义业务逻辑的视图） #

如果是在自定义视图里处理用户创建逻辑，也可以手动查询检查设备状态：

from django.shortcuts import render, redirect
from .models import User, Device
from django.contrib import messages

def create_user(request):
    if request.method == 'POST':
        user_id = request.POST.get('user_id')
        device_id = request.POST.get('device_id')
        try:
            device = Device.objects.get(id=device_id)
        except Device.DoesNotExist:
            messages.error(request, "指定的设备不存在")
            return redirect('create_user')
        
        # 检查设备是否已被其他用户占用
        if User.objects.filter(device=device).exists():
            messages.error(request, f"设备 {device.id} 已经被其他用户占用")
            return redirect('create_user')
        
        # 设备未被占用，创建用户
        User.objects.create(id=user_id, device=device)
        messages.success(request, "用户创建成功")
        return redirect('user_list')
    
    # GET请求返回创建页面，传递所有可选设备
    devices = Device.objects.all()
    return render(request, 'create_user.html', {'devices': devices})

额外提醒：处理并发冲突问题 #

上面的代码在高并发场景下可能会出现“竞态条件”：比如两个请求同时检查同一个设备，都发现未被占用，然后同时创建用户，导致一个设备被两个用户关联。

解决这个问题的最优方案就是前面提到的给device字段加unique=True，数据库会帮你阻止重复插入。如果不能加这个约束，也可以用数据库事务加行锁的方式解决：

from django.db import transaction

def create_user(request):
    if request.method == 'POST':
        user_id = request.POST.get('user_id')
        device_id = request.POST.get('device_id')
        
        with transaction.atomic():
            # 给目标设备加行锁，防止其他事务同时修改该设备的关联状态
            device = Device.objects.select_for_update().get(id=device_id)
            # 再次检查设备是否已被占用
            if User.objects.filter(device=device).exists():
                messages.error(request, f"设备 {device.id} 已经被其他用户占用")
                return redirect('create_user')
            # 创建用户
            User.objects.create(id=user_id, device=device)
        
        messages.success(request, "用户创建成功")
        return redirect('user_list')
    
    # ... 其他代码

通过transaction.atomic()开启事务，并用select_for_update()给设备加行锁，其他请求必须等当前事务完成才能访问该设备，彻底避免并发冲突。

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