직원 전용

Coddy Django 여정의 사용자와 계정 섹션에 포함된 레슨. 16개 중 12번째.

새 페이지 /stats/는 사용자와 노트의 수를 집계합니다. 이 페이지는 사이트를 운영하는 사람들을 위한 것이며, 로그인한 사용자는 누구나 열 수 있습니다.

로그인 여부는 한 가지 문제이고, 사용자가 할 수 있는 일은 또 다른 문제입니다. User에는 사이트를 운영하는 사람들을 위한 플래그가 있습니다. is_staff는 관리자 기능을 사용할 수 있는 사람에게 참입니다.

로그인한 사용자가 볼 수 없는 것을 요청하면 응답은 403 Forbidden입니다. 사이트는 사용자가 누구인지 알고 거부합니다. 뷰에서 PermissionDenied를 발생시키면 Django가 403 응답을 보냅니다. 포럼에서는 잠긴 스레드에 대한 답글을 거부합니다.

from django.core.exceptions import PermissionDenied


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

템플릿에서 이 플래그는 user에 있으므로, 해당 링크를 이용할 수 있는 사람에게만 링크를 표시할 수 있습니다. 링크는 편의를 위한 것이며, 보호 기능은 뷰의 검사가 담당합니다.

챌린지

쉬움

직원용 /stats/를 유지하세요.

notes/views.py에서 직원이 아닌 사용자가 stats에 요청하면 403 Forbidden을 응답하도록 하세요.
base.html에서 New note 뒤에 /stats/로 연결되는 Stats 링크를 추가하고, 직원에게만 표시하세요.

ada(직원)로 /stats/를 여세요. 통계가 표시됩니다. bob으로 열면 403 Forbidden이 표시됩니다. 비밀번호는 coddy-pass-123입니다.

직접 해보기

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)