callpraphmiddleware.py

from pycallgraph import PyCallGraph
from pycallgraph import Config
from pycallgraph import GlobbingFilter
from pycallgraph.output import GraphvizOutput


class ProfilerMiddleware(object):
    includes = []
    excludes = []

    def can(self, request):
        return settings.DEBUG and 'prof' in request.GET

    def process_request(self, request, *args, **kwargs):
        if self.can(request):
            config = Config()
            config.trace_filter = GlobbingFilter(include=self.includes)
            graphviz = GraphvizOutput(output_file='callgraph.png')
            self.profiler = PyCallGraph(output=graphviz, config=config)
            self.profiler.start()

    def process_response(self, request, response):
        if self.can(request):
            self.profiler.done()
        return response