diff --git a/users/views.py b/users/views.py index 00789ac1522c2c28fe3bb9a9c2cb92d6a7d22f19..7be4819359d1165a5268da3fe6ab7a5d1a70dcc6 100644 --- a/users/views.py +++ b/users/views.py @@ -79,7 +79,7 @@ def get_notes(request, model, pk, type): @require_http_methods(["POST"]) @login_required -def add_note(request, model, pk, type): +def add_note(request, model, pk, type=None): model = apps.get_model(*model.split('.')) subject = get_object_or_404(model, pk=pk) note = Note(author=request.user, subject=subject) @@ -94,7 +94,7 @@ def add_note(request, model, pk, type): @require_http_methods(["DELETE"]) @login_required -def delete_note(request, pk, type): +def delete_note(request, pk, type=None): note = get_object_or_404(Note.objects.for_user(request.user), pk=pk, type=type) if type else \ get_object_or_404(Note.objects.for_user(request.user), pk=pk) note.delete()