kallithea Files · rhodecode/lib/profiler.py

Files @ 3a7f5b1a19dd

Branch filter:

Location: kallithea/rhodecode/lib/profiler.py

3a7f5b1a19dd 1.5 KiB text/x-python Show Annotation Show as Raw Download as Raw

Marcin Kuzminski

made rhodecode work with celery 2.2, made some tasks optimizations(forget results)

added celeryconfig.py with just the definitions of hosts, it seams just this is needed to get celery working nice, all other config options are taken from .ini files. This is a temp workaround until i get the proper soltuion to this problem.

from __future__ import with_statement

import cProfile
import pstats
import cgi
import pprint
import threading

from StringIO import StringIO

class ProfilingMiddleware(object):
    def __init__(self, app):
        self.lock = threading.Lock()
        self.app = app
    
    
    def __call__(self, environ, start_response):
        with self.lock:
            profiler = cProfile.Profile()
            def run_app(*a, **kw):
                self.response = self.app(environ, start_response)

            profiler.runcall(run_app, environ, start_response)

            profiler.snapshot_stats()

            stats = pstats.Stats(profiler)
            stats.sort_stats('cumulative')

            # Redirect output
            out = StringIO()
            stats.stream = out

            stats.print_stats()

            resp = ''.join(self.response)

            # Lets at least only put this on html-like responses.
            if resp.strip().startswith('<'):
                ## The profiling info is just appended to the response.
                ##  Browsers don't mind this.
                resp += '<pre style="text-align:left; border-top: 4px dashed red; padding: 1em;">'
                resp += cgi.escape(out.getvalue(), True)
                
                output = StringIO()
                pprint.pprint(environ, output, depth=3)
                
                resp += cgi.escape(output.getvalue(), True)
                resp += '</pre>'
                
            return resp