pycallgraph_middleware.py

import time
from pycallgraph import Config, GlobbingFilter, PyCallGraph
from pycallgraph.output import GraphvizOutput
from django.conf import settings


class CallgraphMiddleware(object):
    def process_view(self, request, callback, callback_args, callback_kwargs):
        if settings.DEBUG and 'graph' in request.GET:
            config = Config(
                max_depth=getattr(settings, 'CALLGRAPH_MAXDEPTH', 100))
            config.trace_filter = GlobbingFilter(
                include=getattr(settings, 'CALLGRAPH_INCLUDES', [])
            )
            graphviz = GraphvizOutput(
                output_file='callgraph-' + str(time.time()) + '.svg',
                output_type='svg'
            )

            pycallgraph = PyCallGraph(
                output=graphviz,
                config=config
            )
            pycallgraph.start()
            self.pycallgraph = pycallgraph

    def process_response(self, request, response):
        if settings.DEBUG and 'graph' in request.GET:
            self.pycallgraph.done()
        return response