Updated version of debug.py, which adds timing and profiling support functions.

- Legacy-Id: 4935

debug.py

@@ -1,23 +1,29 @@
-
 import sys
+import time as timeutils
 import inspect
 import syslog
 from pprint import pformat
+import cProfile
+try:
+    from django.conf import settings
+    debug = settings.DEBUG
+except ImportError:
+    debug = True
+from decorator import decorator
 
 # A debug decorator, written by Paul Butler, taken from
 # http://paulbutler.org/archives/python-debugging-with-decorators/
-# Additional functions and decorator functionality added by
-# Henrik Levkowetz
+
+
+increment = 2
 
 # Number of times to indent output
 # A list is used to force access by reference
-__report_indent = [4]
-increment = 2
-debug = True
-#syslog.openlog("debug", syslog.LOG_PID, syslog.LOG_USER)
+_report_indent = [4]
+_mark = timeutils.clock()
 
 def set_indent(i):
-    __report_indent[0] = i
+    _report_indent[0] = i
 
 def trace(fn):                 # renamed from 'report' by henrik 16 Jun 2011
     """Decorator to print information about a function
@@ -27,30 +33,64 @@ def trace(fn):                 # renamed from 'report' by henrik 16 Jun 2011
     again along with the return value when the function
     returns.
     """
-    def fix(s,n=32):    
+    def fix(s,n=36):
+        import re
+        s = re.sub(r'\\t', ' ', s)
+        s = re.sub(r'\s+', ' ', s)
         if len(s) > n+3:
             s = s[:n]+"..."
-        s = s.replace('\n',' ')
         return s
-    def wrap(*params,**kwargs):
+    def wrap(fn, *params,**kwargs):
         call = wrap.callcount = wrap.callcount + 1
 
-        indent = ' ' * __report_indent[0]
-        fc = "%s(%s)" % (fn.__name__, ', '.join(
+        indent = ' ' * _report_indent[0]
+        fc = "%s.%s(%s)" % (fn.__module__, fn.__name__, ', '.join(
             [fix(repr(a)) for a in params] +
             ["%s = %s" % (a, fix(repr(b))) for a,b in kwargs.items()]
         ))
 
-        print "%s* %s [#%s]" % (indent, fc, call)
-        __report_indent[0] += increment
+        sys.stderr.write("%s* %s [#%s]\n" % (indent, fc, call))
+        _report_indent[0] += increment
         ret = fn(*params,**kwargs)
-        __report_indent[0] -= increment
-        sys.stderr.write( "%s  %s [#%s] ==> %s\n" % (indent, fc, call, repr(ret)))
+        _report_indent[0] -= increment
+        sys.stderr.write("%s  %s [#%s] ==> %s\n" % (indent, fc, call, repr(ret)))
 
         return ret
     wrap.callcount = 0
     if debug:
-        return wrap
+        return decorator(wrap, fn)
+    else:
+        return fn
+
+def mark():
+    say("! mark")
+    _mark = timeutils.clock()
+
+def lap(s):
+    tau = timeutils.clock() - _mark
+    say(">  %s: %.3fs since mark" % (s, tau))
+
+def clock(s):
+    lap(s)
+    _mark = timeutils.clock()
+
+def time(fn):
+    """Decorator to print timing information about a function call.
+    """
+    def wrap(fn, *params,**kwargs):
+        mark = timeutils.clock()
+
+        indent = ' ' * _report_indent[0]
+        fc = "%s.%s()" % (fn.__module__, fn.__name__,)
+
+        ret = fn(*params,**kwargs)
+        tau = timeutils.clock() - mark
+        sys.stderr.write("%s| %s | %.3fs\n" % (indent, fc, tau))
+
+        return ret
+    wrap.callcount = 0
+    if debug:
+        return decorator(wrap, fn)
     else:
         return fn
 
@@ -58,28 +98,37 @@ def show(name):
     if debug:
         frame = inspect.stack()[1][0]
         value = eval(name, frame.f_globals, frame.f_locals)
-        indent = ' ' * (__report_indent[0])
+        indent = ' ' * (_report_indent[0])
         sys.stderr.write("%s%s: %s\n" % (indent, name, value))
 
 def log(name):
     if debug:
         frame = inspect.stack()[1][0]
         value = eval(name, frame.f_globals, frame.f_locals)
-        indent = ' ' * (__report_indent[0])
+        indent = ' ' * (_report_indent[0])
         syslog.syslog("%s%s: %s" % (indent, name, value))
 
 def pprint(name):
     if debug:
         frame = inspect.stack()[1][0]
         value = eval(name, frame.f_globals, frame.f_locals)
-        indent = ' ' * (__report_indent[0])
-        sys.stdout.write("%s%s:\n" % (indent, name))
+        indent = ' ' * (_report_indent[0])
+        sys.stderr.write("%s%s:\n" % (indent, name))
         lines = pformat(value).split('\n')
         for line in lines:
-            sys.stdout.write("%s %s\n"%(indent, line))
+            sys.stderr.write("%s %s\n"%(indent, line))
 
 def say(s):
     if debug:
-        indent = ' ' * (__report_indent[0])
+        indent = ' ' * (_report_indent[0])
         sys.stderr.write("%s%s\n" % (indent, s))
 
+def profile(fn):
+    def wrapper(*args, **kwargs):
+        datafn = fn.__name__ + ".profile" # Name the data file sensibly
+        prof = cProfile.Profile()
+        retval = prof.runcall(fn, *args, **kwargs)
+        prof.dump_stats(datafn)
+        return retval
+    return wrapper
+