A very rough and simple memory profiling middleware

memorymiddleware.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

#Based on a StackOverflow post by Graham Dumpleton

import os
import psutil


import logging
from logging import StreamHandler
shandler = StreamHandler()
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
logger.addHandler(shandler)


class MemoryPerViewMiddleware(object):

    def process_request(self, request):
        request._mem = psutil.Process(os.getpid()).get_memory_info()

    def process_response(self, request, response):
        mem = psutil.Process(os.getpid()).get_memory_info()
        if hasattr(request, '_mem'):
            diff = mem.rss - request._mem.rss
            logger.debug('MEMORY USAGE %r: %s' % (int(diff), request.path))
        return response