Edytuje tylko właściciel

Część sekcji Użytkownicy i konta ścieżki Django w Coddy. Lekcja 11 z 16.

Strona notatki jest już bezpieczna, ale strony jej edycji i usuwania nie są: /3/edit/ i /3/delete/ nadal otwierają dowolną notatkę. Co gorsza, nie trzeba ich wcale otwierać. Formularz jest żądaniem POST, a każdy może wysłać żądanie POST na adres /3/delete/, nawet nie widząc tej strony.

Reguła dotyczy więc każdego widoku, a nie tylko strony, do której prowadzi link: każdy widok, który pobiera wiersz na podstawie jego numeru, decyduje, kto może go pobrać. Zobacz, gdzie znajduje się wyszukiwanie w widoku obsługującym zarówno GET, jak i POST:

def ticket_close(request, pk):
    ticket = ...  # wyszukiwanie: jest wykonywane przy każdym żądaniu
    if request.method == "POST":
        ticket.closed = True
        ticket.save()
        return redirect("tickets:list")
    return render(request, "tickets/close.html", {"ticket": ticket})

Pierwszy wiersz wykonuje się przed gałęzią if request.method == "POST". Wyszukiwanie w tym miejscu, które odrzuca osoby postronne, zwraca im odpowiedź 404 zarówno dla GET, jak i POST, i nic się nie zmienia.

Wyzwanie

Łatwy

W pliku notes/views.py spraw, by funkcje note_edit i note_delete odrzucały notatki, które nie należą do użytkownika. Próba dostępu do notatki innego użytkownika zwraca tam błąd 404 zarówno dla GET, jak i POST, a notatka pozostaje bez zmian. Swoje notatki nadal można edytować i usuwać.

Naciśnij Run, zaloguj się jako bob (hasło coddy-pass-123) i otwórz /1/edit/.

Spróbuj swoich sił

from django.contrib.auth import login
from django.contrib.auth.decorators import login_not_required
from django.contrib.auth.forms import UserCreationForm
from django.shortcuts import get_object_or_404, redirect, render

from .forms import NoteForm
from .models import Note


def note_list(request):
    notes = Note.objects.filter(owner=request.user)
    return render(request, "notes/note_list.html", {"notes": notes})


def note_detail(request, pk):
    note = get_object_or_404(Note, pk=pk, owner=request.user)
    return render(request, "notes/note_detail.html", {"note": note})


def note_create(request):
    if request.method == "POST":
        form = NoteForm(request.POST)
        if form.is_valid():
            note = form.save(commit=False)
            note.owner = request.user
            note.save()
            return redirect("notes:detail", pk=note.pk)
    else:
        form = NoteForm()
    return render(request, "notes/note_form.html", {"form": form})


def note_edit(request, pk):
    note = get_object_or_404(Note, pk=pk)
    if request.method == "POST":
        form = NoteForm(request.POST, instance=note)
        if form.is_valid():
            form.save()
            return redirect("notes:detail", pk=note.pk)
    else:
        form = NoteForm(instance=note)
    return render(request, "notes/note_form.html", {"form": form, "note": note})


def note_delete(request, pk):
    note = get_object_or_404(Note, pk=pk)
    if request.method == "POST":
        note.delete()
        return redirect("notes:list")
    return render(request, "notes/note_confirm_delete.html", {"note": note})


@login_not_required
def signup(request):
    if request.method == "POST":
        form = UserCreationForm(request.POST)
        if form.is_valid():
            user = form.save()
            login(request, user)
            return redirect("notes:list")
    else:
        form = UserCreationForm()
    return render(request, "notes/signup.html", {"form": form})