User View Mixins

CBV Access

class UserDetailAccessMixin(UserPassesTestMixin):
    def test_func(self, user):
        """ 
        Tests if user should be allowed access to DetailView if one of two conditions:
        1. Has staff permissions
        2. Is the customer associated with the object in a DetailView
        """
        if user.is_staff:
            return True
        else:
            detail_object = self.get_object()
            return user == detail_object.customer


class UserViewMixin(object):
    def get_object(self, queryset=None):
        return self.request.user