From 0046e20809d6b1350aea0c47a020b0a6112122c6 Mon Sep 17 00:00:00 2001
From: Taras Tkachuk <tarastkachuk123@gmail.com>
Date: Mon, 5 Aug 2024 22:30:05 +0300
Subject: [PATCH] Solution

---
 taxi/views.py                          | 15 +++++++++++----
 taxi_service/settings.py               |  2 ++
 taxi_service/urls.py                   |  1 +
 templates/includes/sidebar.html        |  9 +++++++++
 templates/registration/logged_out.html |  6 ++++++
 templates/registration/login.html      | 14 ++++++++++++++
 templates/taxi/driver_list.html        |  5 ++++-
 templates/taxi/index.html              |  1 +
 8 files changed, 48 insertions(+), 5 deletions(-)
 create mode 100644 templates/registration/logged_out.html
 create mode 100644 templates/registration/login.html

diff --git a/taxi/views.py b/taxi/views.py
index 82ad312f..b60e361a 100644
--- a/taxi/views.py
+++ b/taxi/views.py
@@ -1,33 +1,40 @@
+from django.contrib.auth.decorators import login_required
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.http import HttpRequest, HttpResponse
 from django.shortcuts import render
 from django.views import generic
 
 from .models import Driver, Car, Manufacturer
 
 
-def index(request):
+@login_required
+def index(request: HttpRequest) -> HttpResponse:
     """View function for the home page of the site."""
 
     num_drivers = Driver.objects.count()
     num_cars = Car.objects.count()
     num_manufacturers = Manufacturer.objects.count()
+    num_visits = request.session.get("num_visits", 0) + 1
+    request.session["num_visits"] = num_visits
 
     context = {
         "num_drivers": num_drivers,
         "num_cars": num_cars,
         "num_manufacturers": num_manufacturers,
+        "num_visits": num_visits,
     }
 
     return render(request, "taxi/index.html", context=context)
 
 
-class ManufacturerListView(generic.ListView):
+class ManufacturerListView(LoginRequiredMixin, generic.ListView):
     model = Manufacturer
     context_object_name = "manufacturer_list"
     template_name = "taxi/manufacturer_list.html"
     paginate_by = 5
 
 
-class CarListView(generic.ListView):
+class CarListView(LoginRequiredMixin, generic.ListView):
     model = Car
     paginate_by = 5
     queryset = Car.objects.select_related("manufacturer")
@@ -37,7 +44,7 @@ class CarDetailView(generic.DetailView):
     model = Car
 
 
-class DriverListView(generic.ListView):
+class DriverListView(LoginRequiredMixin, generic.ListView):
     model = Driver
     paginate_by = 5
 
diff --git a/taxi_service/settings.py b/taxi_service/settings.py
index b6c0cf19..cc71a76f 100644
--- a/taxi_service/settings.py
+++ b/taxi_service/settings.py
@@ -108,6 +108,8 @@
 
 AUTH_USER_MODEL = "taxi.Driver"
 
+LOGIN_REDIRECT_URL = "/"
+
 # Internationalization
 # https://docs.djangoproject.com/en/4.0/topics/i18n/
 
diff --git a/taxi_service/urls.py b/taxi_service/urls.py
index 8b94449f..baf83d1a 100644
--- a/taxi_service/urls.py
+++ b/taxi_service/urls.py
@@ -22,4 +22,5 @@
 urlpatterns = [
     path("admin/", admin.site.urls),
     path("", include("taxi.urls", namespace="taxi")),
+    path("accounts/", include("django.contrib.auth.urls")),
 ] + static(settings.STATIC_URL, document_root=settings.STATIC_ROOT)
diff --git a/templates/includes/sidebar.html b/templates/includes/sidebar.html
index b7cd72dc..b17bc43f 100644
--- a/templates/includes/sidebar.html
+++ b/templates/includes/sidebar.html
@@ -1,4 +1,13 @@
 <ul class="sidebar-nav">
+  {% if user.is_authenticated %}
+    <li>
+      <a href="{% url "taxi:driver-detail" pk=user.id %}">{{ user.username }}</a>
+    </li>
+    <li><a href="{% url "logout" %}">Logout</a></li>
+  {% else %}
+    <li><a href="{% url "login" %}">Login</a></li>
+  {% endif %}
+  <br>
   <li><a href="{% url "taxi:index" %}">Home page</a></li>
   <li><a href="{% url "taxi:manufacturer-list" %}">Manufacturers</a></li>
   <li><a href="{% url "taxi:car-list" %}">Cars</a></li>
diff --git a/templates/registration/logged_out.html b/templates/registration/logged_out.html
new file mode 100644
index 00000000..d36e1682
--- /dev/null
+++ b/templates/registration/logged_out.html
@@ -0,0 +1,6 @@
+{% extends "base.html" %}
+
+{% block content %}
+  <h1>Logged out</h1>
+  <a href="{% url "login" %}">Click here to log in again.</a>
+{% endblock %}
diff --git a/templates/registration/login.html b/templates/registration/login.html
new file mode 100644
index 00000000..b2b9b79b
--- /dev/null
+++ b/templates/registration/login.html
@@ -0,0 +1,14 @@
+{% extends "base.html" %}
+
+{% block content %}
+  <h1>Login</h1>
+  {% if form.errors %}
+    <p>Invalid credentials</p>
+  {% endif %}
+  <form action="{% url "login" %}" method="POST">
+    {% csrf_token %}
+  {{ form.as_p }}
+    <input type="submit" value="Submit">
+    <input type="hidden" name="next" value="{{ next }}">
+  </form>
+{% endblock %}
diff --git a/templates/taxi/driver_list.html b/templates/taxi/driver_list.html
index c6c3e823..fa0b1329 100644
--- a/templates/taxi/driver_list.html
+++ b/templates/taxi/driver_list.html
@@ -7,7 +7,10 @@ <h1>Driver list</h1>
       {% for driver in driver_list%}
         <li>
           <a href="{% url "taxi:driver-detail" pk=driver.id %}">{{ driver.username }}</a>
-          ({{ driver.first_name }} {{ driver.last_name }})
+          ({{ driver.first_name }} {{ driver.last_name }}) 
+        {% if user.id == driver.id %}
+          (Me)
+        {% endif %}
         </li>
       {% endfor %}
     </ul>
diff --git a/templates/taxi/index.html b/templates/taxi/index.html
index 13c59aa8..549801ae 100644
--- a/templates/taxi/index.html
+++ b/templates/taxi/index.html
@@ -10,4 +10,5 @@ <h2>Dynamic content</h2>
     <li><strong>Drivers:</strong> {{ num_drivers }}</li>
     <li><strong>Manufacturers:</strong> {{ num_manufacturers }}</li>
   </ul>
+  <p>You have visited this page {{ num_visits }} time{{ num_visits|pluralize }}</p>
 {% endblock %}