12 lines
334 B
Python
12 lines
334 B
Python
from django.contrib.auth.mixins import UserPassesTestMixin
|
|
from django.contrib import messages
|
|
|
|
|
|
class SuperuserRequiredMixin(UserPassesTestMixin):
|
|
def test_func(self):
|
|
return self.request.user.is_superuser
|
|
|
|
|
|
class StaffRequiredMixin(UserPassesTestMixin):
|
|
def test_func(self):
|
|
return self.request.user.is_staff |