Django - требуется логин для указания c типа пользователя - PullRequest
Новая
       6

Django - требуется логин для указания c типа пользователя

1 голос
/ 05 апреля 2020

Я продлил логин пользователя с Django с помощью AbstractUser следующим образом: 

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 = {})

Есть идеи, как мне это сделать? Спасибо

1 Ответ

1 голос
/ 05 апреля 2020

Вы можете использовать декоратор @user_passes_test [Django -doc] : 

from django.contrib.auth.decorators import user_passes_test

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

<b>@user_passes_test(is_seller)</b>
def account(request):
    # &hellip;

Это позволит только тем пользователям, которые прошли аутентификацию и для которого существует связанный Seller объект.

...