mybuddy/babybuddy/mixins.py

# -*- coding: utf-8 -*-
from django.contrib.auth.mixins import (
    AccessMixin,
    LoginRequiredMixin as LoginRequiredMixInBase,
    PermissionRequiredMixin as PermissionRequiredMixinBase,
)
from django.utils.decorators import method_decorator
from django.views.decorators.cache import never_cache


@method_decorator(never_cache, name="dispatch")
class LoginRequiredMixin(LoginRequiredMixInBase):
    pass


@method_decorator(never_cache, name="dispatch")
class PermissionRequiredMixin(PermissionRequiredMixinBase):
    login_url = "/login"


@method_decorator(never_cache, name="dispatch")
class StaffOnlyMixin(AccessMixin):
    """
    Verify the current user is staff.
    """

    def dispatch(self, request, *args, **kwargs):
        if not request.user.is_staff:
            return self.handle_no_permission()
        return super().dispatch(request, *args, **kwargs)