foxmask / django-th

class UserServiceMixin(object):

    """

        Mixin for UpdateView and DeleteView

    """

    queryset = UserService.objects.all()

    @method_decorator(login_required)

    def dispatch(self, *args, **kwargs):

        return super(UserServiceMixin, self).dispatch(*args, **kwargs)

    def get_queryset(self):

        # get the Service of the connected user

        if self.request.user.is_authenticated():

            return self.queryset.filter(user=self.request.user,

                                        id=self.kwargs.get('pk'))

        # otherwise return nothing

        return UserService.objects.none()

class UserServiceUpdateView(UserServiceMixin, UpdateView):

    """