仅限员工

本课属于 Coddy Django 学习旅程的“用户和账户”单元。第 12 课，共 16 课。

新页面 /stats/ 会统计用户和笔记。这个页面供管理网站的人使用，任何已登录的用户都可以打开。

是否已登录是一个问题，用户可以做什么则是另一个问题。User 有一个供网站管理者使用的标志：is_staff，符合条件、可以使用管理后台的用户，其值为 true。

已登录的用户请求查看自己无权查看的内容时，响应是 403 禁止访问：网站知道你是谁，并拒绝了请求。在视图中，引发 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 禁止访问。
在 base.html 中，在 New note 后添加一个指向 /stats/ 的 Stats 链接，仅对员工显示。

以员工身份 ada 打开 /stats/：你会看到计数。以 bob 的身份打开：返回 403 禁止访问。密码为 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)