Simple Django profiling middleware

middleware.py

# -*- coding: utf-8 -*-
import cProfile
import pstats

from django.conf import settings


class ProfileStatsMiddleware(object):  # pragma: no cover

    """Super light-weight cProfile/pstats profile middleware."""

    def process_request(self, request):
        """Just enable the profiler before continuing if set."""
        if settings.DEBUG and 'prof' in request.GET:
            self.pr = cProfile.Profile()
            self.pr.enable()

    def process_response(self, request, response):
        """Disable the profiler and rewrite the content if set."""
        if settings.DEBUG and 'prof' in request.GET:
            response.content = ""
            self.pr.disable()
            sortby = request.GET.get('prof') or 'cumulative'
            ps = pstats.Stats(self.pr, stream=response).sort_stats(sortby)
            ps.print_stats()

        return response