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 da 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 é preciso abri-las. Um formulário é uma solicitação POST, e qualquer pessoa pode enviar uma solicitação POST para /3/delete/ sem nunca ver a página.

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

def ticket_close(request, pk):
    ticket = ...  # a busca: é 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". Uma busca nesse ponto que rejeite pessoas não autorizadas responde a elas com 404 tanto para o GET quanto para o POST, e nada é alterado.

Desafio

Fácil

Em notes/views.py, faça com que note_edit e note_delete recusem notas que não sejam do próprio usuário. A nota de outro usuário retorna 404 nessa situação, tanto para GET quanto para POST, e permanece como estava. Suas próprias notas continuam podendo ser editadas e excluídas.

Clique em Run, entre como bob (senha coddy-pass-123) e abra /1/edit/.

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