python - 如何根据 django admin 中指定的用户类型重定向不同的登录页面？

Question

我创建了用户-

1. 用户 1 分配的组 A
2. 用户 2 分配了组 B 现在我创建了登录页面，在以 user1 身份登录后，我想重定向到 page1.html，如果以 user2 身份登录，则重定向到 page2.html，为此我写了

def login(request):
    if request.method == 'POST': 
        user = auth.authenticate(username=request.POST['username'], password=request.POST['password'])
        if user is not None:
            auth.login(request,user)
            if user.usergroup=='A':
                return redirect('pages/page1.html')
            else:
                return redirect('pages/page2.html')
        else:
            messages.error(request,'Invalid credentials')
            return redirect('login')
            # User is authenticate  
    else:
        return render(request,'pages/login.html')

有了这个，我以 user1 身份登录， 在 / 'User' object has no attribute 'usergroup' 得到错误 AttributeError ， 请帮助是否还有其他代码？

Answer 1

尝试这个

if user.groups.filter(name='A').exists():
    return redirect('pages/page1.html')
else:
    return redirect('pages/page2.html')