34 lines
No EOL
1.1 KiB
Python
34 lines
No EOL
1.1 KiB
Python
from django.utils.decorators import method_decorator
|
|
from django.shortcuts import redirect
|
|
from django.contrib import messages
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.conf import settings
|
|
|
|
class SuperUserRequiredMixin:
|
|
"""
|
|
View mixin which requires that the authenticated user is a super user
|
|
(i.e. `is_superuser` is True).
|
|
"""
|
|
|
|
@method_decorator(login_required)
|
|
def dispatch(self, request, *args, **kwargs):
|
|
if not request.user.is_superuser:
|
|
messages.error(
|
|
request,
|
|
'You do not have the permission required to perform the '
|
|
'requested operation.')
|
|
return redirect(settings.LOGIN_URL)
|
|
return super().dispatch(request, *args, **kwargs)
|
|
|
|
|
|
class LoginRequiredMixin:
|
|
"""
|
|
View mixin which verifies that the user has authenticated.
|
|
|
|
NOTE:
|
|
This should be the left-most mixin of a view.
|
|
"""
|
|
|
|
@method_decorator(login_required)
|
|
def dispatch(self, *args, **kwargs):
|
|
return super().dispatch(*args, **kwargs) |