Django中OneToOne关联的User与Customer自动绑定问题求助

现有models.py代码如下：

from django.contrib.auth.models import User

class Customer(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, null=True, 
    blank=True)

新用户注册并登录时，必须在Admin后台手动给Customer模型关联用户，否则登录会抛出RelatedObjectDoesNotExist异常，提示“User has no customer”。需求是实现用户与Customer模型的自动关联，同时需要解决views.py中cartData函数捕获异常后如何创建关联Customer实例的问题。

相关views.py代码片段：

def cartData(request): 
    if request.user.is_authenticated:
        try:
            customer = request.user.customer
        except ObjectDoesNotExist:
            customer = ?
            
        order, created =Order.objects.get_or_create(customer=customer, 
        complete=False)
        items = order.orderitem_set.all()
        cartItems = order.get_cart_items
    else:
        cookieData = cookieCart(request)
        cartItems = cookieData['cartItems']
        order = cookieData['order']
        items = cookieData['items']
    return {'cartItems': cartItems, 'order': order, 'items': items}

def register(request): #Register user
    form = CreateUserForm()

    if request.method == 'POST':
        form = CreateUserForm(request.POST)

        if form.is_valid():
            form.save()

            user = form.cleaned_data.get('username')
            messages.success(request, f'Account created for {user} ')
            return redirect('index')

def login_user(request): # Try to login
    if request.method == 'POST':
        username = request.POST.get('username')
        password = request.POST.get('password')

        user = authenticate(request, username=username, 
        password=password)
        
        if user is not None:
            login(request, user)
            return redirect('index')
        else:
            messages.info(request, "Wrong password or username.")
            return redirect('login')

def processOrder(request):
    transaction_id = datetime.datetime.now().timestamp()
    data = json.loads(request.body)

    if request.user.is_authenticated:
        customer = request.user.customer
        # it is possible to change the method like this: customer = Customer.objects.filter(pk=id).update(email=email, last_name=last_name, phone=phone,)
        order, created = Order.objects.get_or_create(customer=customer, complete=False)

    else:
        customer, order = guestOrder(request, data)

• 注册逻辑仅保存User实例，未自动创建关联的Customer实例，导致User与Customer的一对一关系未建立。
• 未处理用户登录后可能不存在Customer的场景，直接调用request.user.customer触发异常。

方法1：注册时直接创建Customer #

修改register视图，在用户创建完成后立刻生成对应的Customer实例：

def register(request): #Register user
    form = CreateUserForm()

    if request.method == 'POST':
        form = CreateUserForm(request.POST)

        if form.is_valid():
            user = form.save()  # 获取保存后的user对象
            # 自动创建关联的Customer
            Customer.objects.create(user=user)
            
            username = form.cleaned_data.get('username')
            messages.success(request, f'Account created for {username} ')
            return redirect('index')

方法2：登录时检查并创建Customer #

兼容已注册但未绑定Customer的老用户，在login_user视图中添加检查逻辑：

def login_user(request): # Try to login
    if request.method == 'POST':
        username = request.POST.get('username')
        password = request.POST.get('password')

        user = authenticate(request, username=username, password=password)
        
        if user is not None:
            # 检查是否存在关联的Customer，不存在则创建
            Customer.objects.get_or_create(user=user)
            login(request, user)
            return redirect('index')
        else:
            messages.info(request, "用户名或密码错误。")
            return redirect('login')

get_or_create会自动判断：用户已有Customer则返回现有实例，无则创建新实例。

方法3：使用Django信号自动关联 #

通过post_save信号，覆盖所有创建User的场景（包括后台创建）：
在models.py中添加信号处理：

from django.contrib.auth.models import User
from django.db.models.signals import post_save
from django.dispatch import receiver

class Customer(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, null=True, blank=True)

# 监听User的保存事件
@receiver(post_save, sender=User)
def create_user_customer(sender, instance, created, **kwargs):
    if created:
        Customer.objects.create(user=instance)

@receiver(post_save, sender=User)
def save_user_customer(sender, instance, **kwargs):
    instance.customer.save()

在apps.py中注册信号确保生效：

from django.apps import AppConfig

class YourAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'your_app_name'

    def ready(self):
        import your_app_name.models  # 导入信号模块

解决cartData中的异常处理 #

若暂时无法全局处理，可在cartData中直接创建Customer：

from django.db.models import ObjectDoesNotExist

def cartData(request): 
    if request.user.is_authenticated:
        try:
            customer = request.user.customer
        except ObjectDoesNotExist:
            # 不存在则创建关联的Customer
            customer = Customer.objects.create(user=request.user)
            
        order, created = Order.objects.get_or_create(customer=customer, complete=False)
        items = order.orderitem_set.all()
        cartItems = order.get_cart_items
    else:
        cookieData = cookieCart(request)
        cartItems = cookieData['cartItems']
        order = cookieData['order']
        items = cookieData['items']
    return {'cartItems': cartItems, 'order': order, 'items': items}

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