Profiling Django Management Command

django_management_profile.py

from cProfile import Profile
from django.core.management.base import BaseCommand

class ProfileEnabledBaseCommand(BaseCommand):
    """Enable profiling a command with --profile.

    Requires child class to define _handle instead of handle.

    via https://gist.github.com/dfrankow
    """
    def add_arguments(self, parser):
        parser.add_argument('--profile', action='store_true', default=False)

    def handle(self, *args, **options):
        if options.get('profile', False):
            profiler = Profile()
            profiler.runcall(self._handle, *args, **options)
            profiler.print_stats()
        else:
            self._handle(*args, **options)

Note that profiler.sort_stats doesn't work: http://stackoverflow.com/a/13690983

I think if you add action='store_true' you could drop the =True part on the command line.

Here's my code for Django 3.2:

from cProfile import Profile
from django.core.management.base import BaseCommand

class ProfileEnabledBaseCommand(BaseCommand):
    """Enable profiling a command with --profile.

    Requires child class to define _handle instead of handle.

    See also https://gist.github.com/tclancy/4236077
    """
    def add_arguments(self, parser):
        parser.add_argument('--profile', action='store_true', default=False)

    def handle(self, *args, **options):
        if options.get('profile', False):
            profiler = Profile()
            profiler.runcall(self._handle, *args, **options)
            profiler.print_stats()
        else:
            self._handle(*args, **options)

Awesome! Updated to your since hopefully no one is still running Django 1.2 or whatever this was for.

:)