django - 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)

我现在想创建一个只能由“卖家”用户访问的视图和表单，就像在 views.py 中这样：

@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对象存在的用户。