登录后如何查看用户类型？ DJANGO

Question

我的应用程序中有两种用户类型。 登录后如何查看用户类型？ 我的登录视图扩展了 django 的 LoginView。 在我的登录视图中，我尝试使用 form_valid，但我认为它会在身份验证之前进行验证。

模型.py

class CustomUser(AbstractUser):
    is_client = models.BooleanField(default=False)
    is_supermarket = models.BooleanField(default=False)

    def __str__(self):
        return self.username

视图.py

class LoginView(LoginView):
    
    
    def get(self, request, *args, **kwargs):
    #check if the user is logged in and redirect to home page
        if request.user.is_authenticated:
            return redirect('produtos:home')
        return super(LoginView, self).get(request, *args, **kwargs)

Answer 1

这是你需要的吗？

class LoginView(LoginView):
    
    
    def get(self, request, *args, **kwargs):
    #check if the user is logged in and redirect to home page
        if request.user.is_authenticated:
            return redirect('produtos:home')
        if request.user.is_supermarket:
            return redirect('supermarket-login') #your unique login page for supermarket user
        return super(LoginView, self).get(request, *args, **kwargs)

Answer 2

我不知道这是否是最好的方法，但我得到了预期的行为

class LoginView(LoginView):
        
    def get(self, request, *args, **kwargs):
        if request.user.is_authenticated:
            return redirect('produtos:home')
        return super(LoginView, self).get(request, *args, **kwargs)
    
    def form_valid(self, form):
        if form.get_user().is_supermarket:
            self.request.session.flush()
            logout(self.request)
            return redirect('login')
        return super().form_valid(form)