Come profilare un comando di gestione personalizzata Django esclusivamente

Question

Vorrei creare un profilo per un comando di gestione personalizzato che richiede un utilizzo relativamente elevato della CPU (esegue il rendering di un'immagine utilizzando PIL). Quando uso il seguente comando ricevo tutti i tipi di moduli di Django (admin, ORM, ecc) nei miei risultati profilazione:

python -m cProfile manage.py testrender 

Ho rimosso tutte le importazioni che possono potenzialmente importare Django, ma sto indovinando il seguente è il colpevole :

from django.core.management.base import BaseCommand, CommandError 

c'è un modo per filtrare cProfile risultati? (vengono mostrati solo i nomi dei file, nessun percorso) Oppure esiste un altro modo per escludere/includere i rispettivi moduli/pacchetti dalla profilazione?

Answer 1

Ho risolto questo problema nel modo seguente:

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


class Command(BaseCommand): 
    ... 

    def _handle(self, *args, **options): 
     # Actual code I want to profile 
     pass 

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

In questo modo le statistiche di profiling sono raccolti nell'ambito di _handle. Così, invece di:

python -m cProfile manage.py testrender 

Dovrò correre:

python manage.py testrender --profile 

che è ancora meglio.

Answer 2

Se non riesco a trovare alcuna risposta. Gprof2Dot come explained here può essere un trucco accettabile.

Non filtra i moduli Non sono interessato, ma si spera che semplificherà l'ispezione dei risultati separando visivamente il mio codice e i moduli Django.

Answer 3

Separare la funzionalità PIL nella propria funzione/classe nel proprio modulo e importarla dal comando di gestione. Quindi puoi testare/profilare la funzionalità PIL indipendentemente da Django.