Django地址模块权限控制及AttributeError问题修复求助

问题根源 #

未创建地址的用户直接访问user.address会触发AttributeError，因为一对一关联的地址对象不存在，而非返回None。

修复方案 #

1. 基础视图检查逻辑 #

在账单地址页和更新页的视图中，先判断用户是否绑定了地址，再决定是否允许访问：

# 账单地址页视图
def billing_address(request):
    # 检查用户是否已创建地址
    if not hasattr(request.user, 'address'):
        # 重定向到地址创建页，可替换为你项目中的创建页URL名称
        return redirect('address_create')
    # 正常渲染页面逻辑
    address = request.user.address
    return render(request, 'billing_address.html', {'address': address})

# 地址更新页视图
def update_address(request):
    if not hasattr(request.user, 'address'):
        return redirect('address_create')
    
    address = request.user.address
    if request.method == 'POST':
        form = AddressForm(request.POST, instance=address)
        if form.is_valid():
            form.save()
            return redirect('billing_address')
    else:
        form = AddressForm(instance=address)
    return render(request, 'update_address.html', {'form': form})

2. 用get_object_or_404简化判断 #

如果地址模型和User是一对一关联（OneToOneField(User, on_delete=models.CASCADE)），可以直接用反向查询捕获不存在的情况：

from django.shortcuts import get_object_or_404

def billing_address(request):
    # 用户无地址时直接返回404，或替换为重定向逻辑
    address = get_object_or_404(Address, user=request.user)
    return render(request, 'billing_address.html', {'address': address})

3. 封装装饰器复用权限逻辑 #

如果多个视图需要相同控制，封装成装饰器减少重复代码：

from django.shortcuts import redirect

def require_address(view_func):
    def wrapper(request, *args, **kwargs):
        if not hasattr(request.user, 'address'):
            return redirect('address_create')
        return view_func(request, *args, **kwargs)
    return wrapper

# 用装饰器修饰视图
@require_address
def billing_address(request):
    address = request.user.address
    return render(request, 'billing_address.html', {'address': address})

@require_address
def update_address(request):
    # 原有更新逻辑...

关键注意事项 #

• 确保地址模型与User是一对一关联，否则反向查询需用user.address_set.exists()判断
• 重定向的address_create需对应你项目中地址创建页的URL名称
• 若要给用户友好提示，可结合django.contrib.messages添加提示信息

内容的提问来源于stack exchange，提问作者Ibikunle The Creator