management_command_profiler.py

import time

from django.db import connections


def management_command_profiler(func):
    """
    This decorator shows the execution time and query count of the function passed.
    Output is designed to use with methods of management commands.
    """
    def wrap_func(*args, **kwargs):

        t1 = time.perf_counter()
        q1 = sum(len(c.queries) for c in connections.all())

        result = func(*args, **kwargs)

        t2 = time.perf_counter()
        q2 = sum(len(c.queries) for c in connections.all())

        # FIXME: Add type check
        self = args[0]
        self.stdout.write(self.style.SUCCESS(f"Took {t2 - t1:0.2f} seconds."))
        self.stdout.write(self.style.SUCCESS(f"Took {q2 - q1} queries."))
        self.stdout.write("")

        return result

    return wrap_func

Took 524.19 seconds.
Took 2781 queries.