diff options
| author | ianb <devnull@localhost> | 2005-12-18 21:35:03 +0000 |
|---|---|---|
| committer | ianb <devnull@localhost> | 2005-12-18 21:35:03 +0000 |
| commit | 5e543609a56386342f95e56e5b0635f71792b3dc (patch) | |
| tree | 13199c7c5b6ed7640e8e5e114cff5e4fce76db4b /paste/profilemiddleware.py | |
| parent | b8ca6badae7723920446824bff540164f2cb1e3e (diff) | |
| download | paste-5e543609a56386342f95e56e5b0635f71792b3dc.tar.gz | |
Moved wdg_validate and profile to debug package
Diffstat (limited to 'paste/profilemiddleware.py')
| -rw-r--r-- | paste/profilemiddleware.py | 201 |
1 files changed, 0 insertions, 201 deletions
diff --git a/paste/profilemiddleware.py b/paste/profilemiddleware.py deleted file mode 100644 index 07294ed..0000000 --- a/paste/profilemiddleware.py +++ /dev/null @@ -1,201 +0,0 @@ -# (c) 2005 Ian Bicking and contributors; written for Paste (http://pythonpaste.org) -# Licensed under the MIT license: http://www.opensource.org/licenses/mit-license.php - -import sys -import os -import hotshot -import hotshot.stats -import threading -import cgi -import time -from cStringIO import StringIO -from paste import wsgilib -from paste import response - -__all__ = ['ProfileMiddleware', 'profile_decorator'] - -class ProfileMiddleware(object): - - """ - Middleware that profiles all requests. - - You can enable this middleware by adding this to your - configuration:: - - middleware.append('paste.profilemiddleware.ProfileMiddleware') - - All HTML pages will have profiling information appended to them. - The data is isolated to that single request, and does not include - data from previous requests. - - This uses the ``hotshot`` module, which effects performance of the - application. It also runs in a single-threaded mode, so it is - only usable in development environments. - """ - - style = ('background-color: #ff9; color: #000; ' - 'border: 2px solid #000; padding: 5px;') - - def __init__(self, app, global_conf, - log_filename='profile.log.tmp', - limit=40): - self.app = app - self.lock = threading.Lock() - self.log_filename = log_filename - self.limit = limit - - def __call__(self, environ, start_response): - catch_response = [] - body = [] - def replace_start_response(status, headers): - catch_response.extend([status, headers]) - start_response(status, headers) - return body.append - def run_app(): - body.extend(self.app(environ, replace_start_response)) - self.lock.acquire() - try: - prof = hotshot.Profile(self.log_filename) - prof.addinfo('URL', environ.get('PATH_INFO', '')) - try: - prof.runcall(run_app) - finally: - prof.close() - body = ''.join(body) - headers = catch_response[1] - content_type = response.header_value(headers, 'content-type') - if not content_type.startswith('text/html'): - # We can't add info to non-HTML output - return [body] - stats = hotshot.stats.load(self.log_filename) - stats.strip_dirs() - stats.sort_stats('time', 'calls') - output = capture_output(stats.print_stats, self.limit) - output_callers = capture_output( - stats.print_callers, self.limit) - body += '<pre style="%s">%s\n%s</pre>' % ( - self.style, cgi.escape(output), cgi.escape(output_callers)) - return [body] - finally: - self.lock.release() - -def capture_output(func, *args, **kw): - # Not threadsafe! (that's okay when ProfileMiddleware uses it, - # though, since it synchronizes itself.) - out = StringIO() - old_stdout = sys.stdout - sys.stdout = out - try: - func(*args, **kw) - finally: - sys.stdout = old_stdout - return out.getvalue() - -def profile_decorator(**options): - - """ - Profile a single function call. - - Used around a function, like:: - - @profile_decorator(options...) - def ... - - All calls to the function will be profiled. The options are - all keywords, and are: - - log_file: - The filename to log to (or ``'stdout'`` or ``'stderr'``). - Default: stderr. - display_limit: - Only show the top N items, default: 20. - sort_stats: - A list of string-attributes to sort on. Default - ``('time', 'calls')``. - strip_dirs: - Strip directories/module names from files? Default True. - add_info: - If given, this info will be added to the report (for your - own tracking). Default: none. - log_filename: - The temporary filename to log profiling data to. Default; - ``./profile_data.log.tmp`` - """ - - def decorator(func): - def replacement(*args, **kw): - return DecoratedProfile(func, **options)(*args, **kw) - return replacement - return decorator - -class DecoratedProfile(object): - - lock = threading.Lock() - - def __init__(self, func, **options): - self.func = func - self.options = options - - def __call__(self, *args, **kw): - self.lock.acquire() - try: - return self.profile(self.func, *args, **kw) - finally: - self.lock.release() - - def profile(self, func, *args, **kw): - ops = self.options - prof_filename = ops.get('log_filename', 'profile_data.log.tmp') - prof = hotshot.Profile(prof_filename) - prof.addinfo('Function Call', - self.format_function(func, *args, **kw)) - if ops.get('add_info'): - prof.addinfo('Extra info', ops['add_info']) - exc_info = None - try: - start_time = time.time() - try: - result = prof.runcall(func, *args, **kw) - except: - exc_info = sys.exc_info() - end_time = time.time() - finally: - prof.close() - stats = hotshot.stats.load(prof_filename) - os.unlink(prof_filename) - if ops.get('strip_dirs', True): - stats.strip_dirs() - stats.sort_stats(*ops.get('sort_stats', ('time', 'calls'))) - display_limit = ops.get('display_limit', 20) - output = capture_output(stats.print_stats, display_limit) - output_callers = capture_output( - stats.print_callers, display_limit) - output_file = ops.get('log_file') - if output_file in (None, 'stderr'): - f = sys.stderr - elif output_file in ('-', 'stdout'): - f = sys.stdout - else: - f = open(output_file, 'a') - f.write('\n%s\n' % ('-'*60)) - f.write('Date: %s\n' % time.strftime('%c')) - f.write('Function call: %s\n' - % self.format_function(func, *args, **kw)) - f.write('Wall time: %0.2f seconds\n' - % (end_time - start_time)) - f.write(output) - f.write(output_callers) - if output_file not in (None, '-', 'stdout', 'stderr'): - f.close() - if exc_info: - # We captured an exception earlier, now we re-raise it - raise exc_info[0], exc_info[1], exc_info[2] - return result - - def format_function(self, func, *args, **kw): - args = map(repr, args) - args.extend( - ['%s=%r' % (k, v) for k, v in kw.items()]) - return '%s(%s)' % (func.__name__, ', '.join(args)) - - |
