Configure profiler via Apache's SetEvn directive

wsgi_profiler.py

'''
Configure profiler via Apache's SetEvn directive
'''


class Profiler(object):
    '''
    Simple wrapper around werkzeug's ProfilerMiddleware. This allow enabling
    the profile via apache's SetEnv directive:

    SetEnv WSGI_PROFILER Yes
    '''
    def __init__(self, app, enabled=False):
        self._app = app
        self._wsgi_app = self._app.wsgi_app
        self.enabled = enabled

    def __call__(self, environ, start_response):
        if environ.get('WSGI_PROFILER', 'No') != 'No':
            self.enabled = True
        if self.enabled:
            from werkzeug.contrib.profiler import ProfilerMiddleware
            self._wsgi_app = ProfilerMiddleware(self._wsgi_app)
            self._app.debug = True
        return self._wsgi_app(environ, start_response)