How to count objects in django

Question

Thank you all for always sharing your knowledge here.

I have an issue with counting an object in Django. I am currently learning and working on a basic HR system and already have my views, models, et al set up. I plan to have an interface whereby I can print out employees count based on gender. The one I currently have set up is increasing the counts for both male and female any time I create a new employee. Please how do I correct this anomaly?

views.py

from django.contrib.auth.decorators import login_required
from django.contrib.auth.models import User
from django.shortcuts import render
from django_tables2 import RequestConfig
from django_tables2.export import TableExport

from .models import Employee
from .models import EmployeeFilter
from .tables import EmployeeTable

@login_required()
def employees(request):
    filter = EmployeeFilter(request.GET, queryset=Employee.objects.all())
    table = EmployeeTable(filter.qs)
    RequestConfig(request, paginate={"per_page": 15}).configure(table)
    count = Employee.objects.all().count()
    male_count = Employee.objects.filter(gender__contains='Male').count()
    female_count = Employee.objects.filter(gender__contains='Female').count()
    user_count = User.objects.all().count()

    export_format = request.GET.get("_export", None)

    if TableExport.is_valid_format(export_format):
        exporter = TableExport(export_format, table)
        return exporter.response("table.{}".format("csv", "xlsx"))

    return render(request, "employees/employees.html", {
        "table": table,
        "filter": filter,
        "count": count,
        "male_count": male_count,
        "female_count": female_count,
        "user_count": user_count,
    })

template.html

{% extends "employees/base.html" %}
{% load render_table from django_tables2 %}
{% load django_tables2 %}
{% load querystring from django_tables2 %}
{% block content %}

    
        
            
                
                    
                        
                            Total Employees
                            {{ count }}
                        

                        
                            
                        
                    
                
            

            
                
                    
                        
                            Male Employees
                            {{ male_count }}
                        

                        
                            
                        
                    
                
            

            
                
                    
                        
                            Female Employees
                            {{ female_count }}
                        

                        
                            
                        
                    
                
            

            
                
                    
                        
                            Active Users
                            {{ user_count }}
                        

                        
                            
                        
                    
                
            
        
    

    
    
        
            
                
                
                    Filter employee records
                    {{ form.non_field_errors }}
                    
                        {{ filter.form.first_name.errors }}
                        {{ filter.form.first_name }}
                    
                    
                        {{ filter.form.last_name.errors }}
                        {{ filter.form.last_name }}
                    
                    
                
            

            
                
                
                    
                    
                        csv
                        xlsx
                    
                
            
        
    
    
    {% render_table table 'django_tables2/bootstrap.html' %}
    
    {% endblock content %}

models.py

...
GENDER_CHOICES = (
    ('FEMALE', 'Female'),
    ('MALE', 'Male'),
    ("DWTS" "Don't want to say"),
)
...
@python_2_unicode_compatible
class Employee(models.Model):
    # basic information of employee
    first_name = models.CharField(_('first name'), max_length=40)
    last_name = models.CharField(_('last name'), max_length=40)
    emp_photo = models.ImageField(_('passport'))
    date_of_birth = models.DateField(_('birthday'))
    gender = models.CharField(_('gender'), max_length=15, choices=GENDER_CHOICES, default=['MALE', 'Male'])
    house_address = models.CharField(_('house address'), max_length=100)
    city_of_residence = models.CharField(_('city'), max_length=100)
    state_of_residence = models.CharField(_('state'), max_length=100, choices=NIGERIAN_STATE_CHOICES)
    country_of_residence = models.CharField(_('country'), max_length=100, choices=COUNTRY_CHOICES,
                                            default=[156, 'Nigeria'])
    state_of_origin = models.CharField(_('state of origin'), max_length=100, choices=NIGERIAN_STATE_CHOICES)

    class Meta:
        verbose_name = _('Employee')
        verbose_name_plural = _('Employees')

    def __str__(self):
        return "{} {}".format(self.first_name, self.last_name)

@python_2_unicode_compatible
class EmployeeFilter(django_filters.FilterSet):
    first_name = django_filters.CharFilter(lookup_expr='iexact', widget=TextInput(
        attrs={'placeholder': 'First name', 'class': 'input_search'}))
    last_name = django_filters.CharFilter(lookup_expr='iexact',
                                          widget=TextInput(attrs={'placeholder': 'Last name', 'class': 'input_search'}))

    class Meta:
        model = Employee
        fields = ["first_name", "last_name", ]

Sergey Pugach · Accepted Answer

I think you need to remove contains from your filter:

male_count = Employee.objects.filter(gender='Male').count()
female_count = Employee.objects.filter(gender='Female').count()

Or make a single request:

male_femail_count = Employee.objects.values('gender').annotate(count=Count('id'))
# result
# [{'gender': 'Male', 'count': 2}, {'gender': 'Female', 'count': 3}]

How to count objects in django

views.py

models.py

Answers (1)

Related Questions