Django - 特定用户类型需要登录

Question

我使用 AbstractUser 从 Django 扩展了用户登录，如下所示：

class User(AbstractUser):
    pass
class Seller(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    verified = models.BooleanField(default=False)
    money = models.IntegerField(default=0)

def __str__(self):
    return str(self.user)

我现在想创建一个只能由“卖家”用户访问的视图和表单，如下所示：

@login_required
def account(request):
    return render(request= request,
               template_name='main/account.html',
               context = {})

知道我该怎么做吗？谢谢

Answer 1

您可以使用装饰@user_passes_test器 [Django-doc]：

from django.contrib.auth.decorators import user_passes_test

def is_seller(user):
    try:
        return user.is_authenticated and user.seller is not None
    except Seller.DoesNotExist:
        return False

@user_passes_test(is_seller)
def account(request):
    # …

这将仅允许经过身份验证且Seller存在相关对象的用户。