Somente o proprietário edita

Parte da seção Usuários e contas do Journey de Django da Coddy. Lição 11 de 16.

A página de uma nota está segura agora, mas as páginas de edição e exclusão não estão: /3/edit/ e /3/delete/ ainda abrem qualquer nota. Pior ainda, nem é necessário abri-las. Um formulário é uma solicitação POST, e qualquer pessoa pode enviar uma solicitação POST para /3/delete/ sem sequer ver a página.

Portanto, a regra se aplica por visualização, não por página para a qual você cria um link: cada visualização que carrega uma linha pelo número decide ali quem pode carregá-la. Veja onde a busca fica em uma visualização que lida tanto com GET quanto com POST:

def ticket_close(request, pk):
    ticket = ...  # a busca: ela é executada para cada requisição
    if request.method == "POST":
        ticket.closed = True
        ticket.save()
        return redirect("tickets:list")
    return render(request, "tickets/close.html", {"ticket": ticket})

A primeira linha é executada antes do ramo if request.method == "POST". Se essa busca também verificar quem é o proprietário, uma pessoa desconhecida recebe 404 tanto para o GET quanto para o POST, e nada muda.

Desafio

Fácil

Em notes/views.py, faça com que note_edit e note_delete recusem notas que não pertençam ao usuário. A nota de outro usuário retorna 404 nessa página, tanto para GET quanto para POST, e permanece inalterada. Você ainda pode editar e excluir suas próprias notas.

Faça login como bob (senha coddy-pass-123) e abra /1/edit/: página 404 não encontrada.

Experimente você mesmo

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})