From 590e083b144eeebc08604a1fb76a774e0c7c4713 Mon Sep 17 00:00:00 2001 From: vabene1111 Date: Mon, 9 Dec 2019 11:26:21 +0100 Subject: [PATCH] commment permission --- cookbook/views/edit.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/cookbook/views/edit.py b/cookbook/views/edit.py index 01cb16c3c..125740149 100644 --- a/cookbook/views/edit.py +++ b/cookbook/views/edit.py @@ -164,11 +164,12 @@ class CommentUpdate(LoginRequiredMixin, UpdateView): # TODO add msg box - def get_object(self, *args, **kwargs): - obj = super(CommentUpdate, self).get_object(*args, **kwargs) - if not obj.created_by == self.request.user: - return obj # TODO move to dispatch or build custom exception - return obj + def dispatch(self, request, *args, **kwargs): + obj = self.get_object() + if not obj.created_by == request.user: + messages.add_message(request, messages.ERROR, _('You cannot edit this comment!')) + return HttpResponseRedirect(reverse('view_recipe', args=[obj.recipe.pk])) + return super(CommentUpdate, self).dispatch(request, *args, **kwargs) def get_success_url(self): return reverse('edit_comment', kwargs={'pk': self.object.pk})