p7

toh/hero/admin.py

@@ -1,3 +1,4 @@
 from django.contrib import admin
+from .models import Hero
 
-# Register your models here.
+admin.site.register(Hero)

toh/hero/tests.py

@@ -1,3 +1,19 @@
-from django.test import TestCase
+from pydoc import cli
+from django.test import TestCase, Client
+from .models import Hero
 
-# Create your tests here.
+class HeroTestCase(TestCase):
+    def setUp(self) -> None:
+        Hero.objects.create(name='Superman')
+        Hero.objects.create(name='Batman')
+        Hero.objects.create(name='Ironman')
+
+    def test_hero_count(self):
+        self.assertEqual(Hero.objects.all().count(), 3)
+
+    def test_hero_id(self) -> None:
+        client = Client()
+        response = client.get('/hero/1/')
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIn('1', response.content.decode())

toh/hero/urls.py

@@ -2,8 +2,10 @@
 from . import views
 
 urlpatterns = [
-    path('', views.hero_list),
+    path('', views.index),
+    path('list/', views.hero_list),
     path('<int:id>/', views.id, name='hero_id'),
+    path('token/', views.token, name='token'),
     path('<str:name>/', views.name, name='hero_name'),
     path('info/<int:id>/', views.hero_info, name='hero_info'),
 ]

toh/hero/views.py

@@ -1,19 +1,29 @@
 from django.http import HttpResponse, HttpResponseBadRequest, HttpResponseNotAllowed, JsonResponse
-from django.views.decorators.csrf import csrf_exempt
+from django.views.decorators.csrf import csrf_exempt, ensure_csrf_cookie
 import json
 from json.decoder import JSONDecodeError
 from .models import Hero
 
-# def index(request):
-#     return HttpResponse('Hello, world!')
+@ensure_csrf_cookie
+def token(request):
+    if request.method == 'GET':
+        return HttpResponse(status=204)
+    else:
+        return HttpResponseNotAllowed(['GET'])
+
+def index(request):
+    if 'visit_count' not in request.session:
+        request.session['visit_count'] = 1
+    else:
+        request.session['visit_count'] += 1
+    return HttpResponse(f'Hello, world! You visited {request.session["visit_count"]}')
 
 def id(request, id):
     return HttpResponse(f'Your id is {id}!')
 
 def name(request, name):
     return HttpResponse(f'Your name is {name}!')
 
-@csrf_exempt
 def hero_list(request):
     if request.method == 'GET':
         hero_all_list = [hero for hero in Hero.objects.all().values()]