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

{% extends 'layout.html' %}
{% block content %}
    <h1>くじ引き内容</h1>

{% if user.is_authenticated %}　#追加
    <table class='table table-striped table-hover'>
        <tr>
            <th>内容</th><th>編集</th><th>削除</th>
        </tr>
        {% for tdl in object_list%}
        <tr>
            <td> {{ tdl.item }}</td>
            <td> <a href='{% url "tdl:update" tdl.id %}'>編集</a> </td>
            <td> <a href='{% url "tdl:delete" tdl.id %}'>削除</a> </td>
        </tr>
        {% endfor %}
    </table>

        <div>
            <a href='{% url "tdl:create" %}' class="btn btn-primary">新規作成</a>
            <a href='{% url "tdl:execution" %}' class="btn btn-secondary">実行</a>
        </div>

#以下を追加-------
{% else %}　
    <table class='table table-striped table-hover'>
        <tr>
            <th>内容</th>
        </tr>
        <tr>
            <td>サンプル1</td>
        </tr>
        <tr>
            <td>サンプル2</td>
        </tr>
    </table>
{% endif %}
#ここまで-------
{% endblock %}

ここでは、ログイン時にだけ新規作成ボタン、実行ボタン、編集リンク、削除リンクを表示させるため、if文を利用します。

以下のコードは、ログイン時にだけ次のコードを実行します。

{% if user.is_authenticated %}

{% else %}　以降のコードは、未ログイン時に表示されます。
未ログインのユーザには、サンプルの内容を表示するようにコードを追加しています。

次にログインユーザの内容だけを表示させるため、views.pyを修正します。

２．views.pyを修正

次はログイン時にだけ、各操作ができるようにコードを修正していきます。

ファイル：myapptdl>tdl>views.py

コード：

from django.urls import reverse_lazy
from django.views import generic
from .models import Tdl
import random
from django.contrib.auth.mixins import LoginRequiredMixin

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 = '__all__'

class UpdateView(LoginRequiredMixin, generic.edit.UpdateView):
  model = Tdl
  fields = '__all__'

class DeleteView(LoginRequiredMixin, generic.edit.DeleteView):
  model = Tdl
  success_url = reverse_lazy('tdl:index')

views.pyではログインユーザが作成した内容だけ表示させるためのコードを追記します。

id = self.request.user.id:ログインしているユーザIDをid変数へ代入する

return Tdl.objects.filter(author_id=id)：ログインユーザIDと内容投稿者のIDが一致しているものだけを抽出する

【参考URL】