Djangoでくじ引きアプリを作ってみる_part12 | 30代未経験ネットワークエンジニアのblog

from django.urls import reverse_lazy
from django.views import generic
from .models import Tdl
import random
from django.contrib.auth.mixins import LoginRequiredMixin
from django.core.exceptions import PermissionDenied　#追加

class IndexView(generic.ListView):
   model = Tdl

   def get_queryset(self):
       id = self.request.user.id
       return Tdl.objects.filter(author_id=id)

class ExecutionView(LoginRequiredMixin, generic.ListView):
  model = Tdl
  template_name = 'tdl/tdl_execution.html'

  def get_queryset(self):
#    return Tdl.objects.order_by('?')[:1]
    pk = Tdl.objects.values_list('pk', flat=True)
    pk_list = list(pk)
    pk_random = random.choice(pk_list)
    queryset = Tdl.objects.filter(pk=pk_random)

    return queryset

class CreateView(LoginRequiredMixin, generic.edit.CreateView):
  model = Tdl
  fields = ['item'] #__all__'　#修正

#以下を追加----
  def form_valid(self, form):
      form.instance.author = self.request.user
      return super(CreateView, self).form_valid(form)

class UpdateView(LoginRequiredMixin, generic.edit.UpdateView):
  model = Tdl
 fields = ['item'] #__all__'　#修正

  def dispatch(self, request, *args, **kwargs):
      obj = self.get_object()
      if obj.author != self.request.user:
          raise PermissionError('編集権限がありません')
      return super(UpdateView, self).dispatch(request, *args, **kwargs)

#ここまで----
class DeleteView(LoginRequiredMixin, generic.edit.DeleteView):
  model = Tdl
  success_url = reverse_lazy('tdl:index')

●CreateViewクラス
fields変数を[‘item’]に変更しています。

そして「def form_valid(self, form):」メソッドをオーバライドします。
「form.instance.author」変数に「self.request.user」を挿入しています。
「self.request.user」は、ログイン中のユーザのことです。
そのユーザを投稿者ユーザにしています。

●UpdateViewクラス
fields変数を[‘item’]に変更しています。

次にdispatchメソッドで、データを作成したユーザとログインユーザが同じでない場合はエラーを出力するようにしています。
※今回のプログラムでは自分の登録した内容しか表示されませんが、念のために入れておきます。
if obj.author != self.request.user:
raise PermissionError(‘編集権限がありません’)

以上で設定は終わりなので、サーバを起動して確認します。

cd myapp 
python manage.py runserver

次の手順で確認していきます。

以下URLにブラウザでアクセスする

http://localhost:8000/tdl
localhost
新規作成をクリックする
「Item」しか表示されていないことを確認する
「20200814新規作成」を登録する
「20200814新規作成」の編集をクリックする
「Item」しか表示されていないことを確認する
ログインしたユーザで登録されているかを確認するため、以下URLにブラウザでアクセスする

http://localhost:8000/admin
localhost
「Tdls」をクリックして、「20200814新規作成」をクリックする
「Author」がadminになっていることを確認する