From 38f69b156cdf1f10ea237984f5c88b362fbb51b7 Mon Sep 17 00:00:00 2001 From: Filipp Lepalaan Date: Sun, 8 Nov 2015 20:15:11 +0200 Subject: Cleanup --- servo/views/note.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'servo/views') diff --git a/servo/views/note.py b/servo/views/note.py index eb19c61..4793e17 100644 --- a/servo/views/note.py +++ b/servo/views/note.py @@ -291,7 +291,7 @@ def templates(request, template_id=None): def toggle_flag(request, pk, flag): field = 'is_%s' % flag - note = Note.objects.get(pk=pk) + note = get_object_or_404(Note, pk=pk) attr = getattr(note, field) setattr(note, field, not attr) note.save() @@ -300,8 +300,8 @@ def toggle_flag(request, pk, flag): def toggle_tag(request, pk, tag_id): - note = Note.objects.get(pk=pk) - tag = Tag.objects.get(pk=tag_id) + note = get_object_or_404(Note, pk=pk) + tag = get_object_or_404(Tag, pk=tag_id) if tag in note.labels.all(): note.labels.remove(tag) -- cgit v1.2.3