Tylko dla personelu

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

Nowa strona /stats/ zlicza użytkowników i notatki. Jest przeznaczona dla osób zarządzających witryną i może ją otworzyć każdy zalogowany użytkownik.

Zalogowanie się to jedno; to, co użytkownik może robić, to drugie. User ma flagę dla osób zarządzających witryną: is_staff, która ma wartość true dla osób, które mogą korzystać z panelu administracyjnego.

Gdy zalogowany użytkownik prosi o coś, czego nie może zobaczyć, odpowiedzią jest 403 Forbidden: witryna wie, kim jesteś, i odmawia. W widoku zgłoś wyjątek PermissionDenied, a Django wyśle odpowiedź 403. Forum odmawia odpowiedzi w zablokowanym wątku:

from django.core.exceptions import PermissionDenied


def thread_reply(request, pk):
    thread = get_object_or_404(Thread, pk=pk)
    if thread.locked:
        raise PermissionDenied
    ...

W szablonie flaga jest dostępna w user, więc link może być widoczny tylko dla osób uprawnionych do przejścia pod ten adres. Link to ułatwienie; zabezpieczeniem jest sprawdzenie w widoku.

Wyzwanie

Łatwy

Zachowaj /stats/ dla pracowników.

W pliku notes/views.py spraw, aby widok stats zwracał odpowiedź 403 Forbidden każdemu użytkownikowi, który nie jest pracownikiem.
W pliku base.html, po New note, dodaj link Stats do /stats/, widoczny tylko dla pracowników.

Otwórz /stats/ jako ada (pracownik): zobaczysz liczby. Jako bob: odpowiedź 403 Forbidden. Hasło: coddy-pass-123.

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.contrib.auth.models import User
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, owner=request.user)
    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, owner=request.user)
    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})


def stats(request):
    context = {"user_count": User.objects.count(), "note_count": Note.objects.count()}
    return render(request, "notes/stats.html", context)