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)

:)