diff options
Diffstat (limited to 'servo/views')
-rw-r--r-- | servo/views/note.py | 6 |
1 files changed, 3 insertions, 3 deletions
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) |