Wagtail front-end permissions mixin

frontend_permissions.py

# Mixin 

class PermissionRequiredMixin:
    required_permissions = []

    def route(self, request, path_components):
        if not request.user.has_perms(self.required_permissions):
            return RouteResult(self, kwargs={'permission_check_failed': True})

        return super().route(request, path_components)

    def serve_permission_error(self, request):
        """
        Respond with a polite message informing the user they need to log in
        """
        raise NotImplementedError

    def serve(self, request, *args, permission_check_failed=False, **kwargs):
        if permission_check_failed:
            return self.serve_permission_error(request)

        return super().serve(request, *args, **kwargs)


# Implementation:

class ClientAreaHomePage(PermissionRequiredMixin, Page):
    required_permissions = ['client_area.can_access_client_area']

    def serve_permission_error(self, request):
        return TemplateResponse(request, 'login.html')