implement(fixed)

mate-academy · Sep 13, 2024 · 7ed1d81 · 7ed1d81
1 parent f1595b2
commit 7ed1d81
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/taxi/views.py b/taxi/views.py
@@ -48,14 +48,6 @@ class DriverListView(LoginRequiredMixin, generic.ListView):
     model = Driver
     paginate_by = 5
 
-    def get_queryset(self):
-        queryset = super().get_queryset()
-        for driver in queryset:
-            if driver == self.request.user:
-                driver.is_current_user = True
-            else:
-                driver.is_current_user = False
-        return queryset
 
 
 class DriverDetailView(LoginRequiredMixin, generic.DetailView):

diff --git a/templates/taxi/driver_list.html b/templates/taxi/driver_list.html
@@ -8,7 +8,9 @@ <h1>Driver list</h1>
         <li>
           <a href="{% url "taxi:driver-detail" pk=driver.id %}">{{ driver.username }}</a>
           ({{ driver.first_name }} {{ driver.last_name }})
-          {% if driver.is_current_user %}(Me){% endif %}
+          {% if driver == request.user %}
+            (Me)
+          {% endif %}
         </li>
       {% endfor %}
     </ul>