singular and plural on entry_list

core/mixins.py

@@ -18,6 +18,10 @@ def get_context_data(self, **kwargs):
         context['count'] = self.get_queryset().count()
         return context
 
+    def name_(self, singular='', plural=''):
+        name = self.singular if self.count == 1 else self.plural
+        return name
+
 
 class FirstnameSearchMixin(object):
 

core/templates/core/entry/entry_list.html

@@ -76,15 +76,7 @@ <h2><i class="fa fa-inbox"></i> Lista de Entradas </h2>
 
     <hr>
 	<div>
-		{% if count > 0 %}
-			<h3><b>Total:</b> {{ count }}
-				{% if count == 1 %}
-					entrada
-				{% elif count > 1 %}
-					entradas
-				{% endif %}
-			</h3>
-		{% endif %}
+		<h3><b>Total:</b> {{ count }} {{ name }}</h3>
 	</div>
 
 <!-- pagination -->

core/views.py

@@ -76,14 +76,8 @@ class EntryList(CounterMixin, ListView):
     model = Entry
     context_object_name = 'entrys'
     paginate_by = 10
-
-    # def get_context_data(self, **kwargs):
-    #     context = super(EntryList, self).get_context_data(**kwargs)
-    #     context['urgente'] = Entry.objects.filter(priority='u')
-    #     context['alta'] = Entry.objects.filter(priority='a')
-    #     context['normal'] = Entry.objects.filter(priority='n')
-    #     context['baixa'] = Entry.objects.filter(priority='b')
-    #     return context
+    singular = 'entrada'
+    plural = 'entradas'
 
     def get_context_data(self, **kwargs):
         priority_classes = {'u': 'fa-flash urgente',
@@ -94,6 +88,8 @@ def get_context_data(self, **kwargs):
         context.update({'priority_search_form': PrioritySearchForm(), })
         context['priority'] = [(item, item_display, priority_classes[item])
                                for item, item_display in priority_list]
+        self.name = self.name_(EntryList.singular, EntryList.plural)
+        context['name'] = self.name
         return context
 
     def get_queryset(self):