Джанго декоратор в зависимости от типа пользователя

Question

я иметь следующую модель:

class Developer(models.Model): 
    COUNTRIES = (
     ('U', 'United States'), 
     ('O', 'Other'), 
    ) 
    user = models.OneToOneField(User, related_name='freelancer_user') 
    country = models.CharField(max_length=1, choices=COUNTRIES, default='U') 


class Customer(models.Model): 
    user = models.OneToOneField(User, related_name='customer_user') 
    location = models.CharField(max_length=50, blank=True) 

У меня есть следующий вид:

@login_required 
def postjob(request): 
    job_form = JobCreateForm() 
    return render(request, 'myapp/postjob.html', {'job_form': job_form}) 

Это работает, насколько аутентификации. Но я хочу, чтобы только пользователь Клиента получал доступ к этому представлению, а не к разработчику. Как мне это сделать?

Answer 1

Вам нужен user_passes_test декоратора и функции тестирования:

@user_passes_test(is_customer) 
def postjob(request): 
    job_form = JobCreateForm() 
    return render(request, 'myapp/postjob.html', {'job_form': job_form}) 

Где is_customer что-то вроде:

def is_customer(user): 
    # how do you tell the user is a customer? 
    if user.is_staff or user.is_superuser: 
     return False 
    return True