Handy decorators for timing and counting django db queries

django_perf_decorators.py

import functools
import time

from django.db import connection


def timer(func):
    @functools.wraps(func)
    def wrapper_timer(*args, **kwargs):
        tic = time.perf_counter()
        value = func(*args, **kwargs)
        toc = time.perf_counter()
        elapsed_time = toc - tic
        print(f"[{func.__name__}] elapsed time: {elapsed_time * 1000:0.2f} ms")
        return value

    return wrapper_timer


def query_count(func):
    @functools.wraps(func)
    def wrapper_timer(*args, **kwargs):
        initial_query_count = len(connection.queries)
        value = func(*args, **kwargs)
        print(
            f"[{func.__name__}] queries executed: "
            f"{len(connection.queries) - initial_query_count}"
        )
        return value

    return wrapper_timer