Changeset 601 for branches/decorate/timer.py

Timestamp:

12/03/12 11:57:08 (3 years ago)

Author:

mmckerns

Message:

memoized0 functions renamed to memoized, memoized class to memoize;
improved simple timer decorator to include verbosity

File:

• branches/decorate/timer.py (modified) (1 diff)

branches/decorate/timer.py

@@ -1 +1 @@
 
-def timed(f):
-    """decorator for timing a function"""
-    def func(*args, **kwds):
-        from time import time
-        t = time()
-        res = f(*args, **kwds)
-        print "Timed: %s" % (time() - t) #XXX: write to a monitor instead ?
-        return res
-    return func
+def timed(verbose=True):
+    def dec(f):
+        """decorator for timing a function"""
+        from time import time as now
+        __time = [True, None]
+        if not verbose: __time[0] = False
+        def time(stamp=False):
+            if not stamp: return __time[-1]
+            import datetime
+            return datetime.timedelta(seconds=__time[-1])
+        def __verbose(on=True):
+            __time[0] = bool(on)
+        def func(*args, **kwds):
+            t = now()
+            res = f(*args, **kwds)
+            __time[-1] = now() - t
+            if __time[0]:
+                print "Timed: %s" % __time[-1]
+            return res
+        func.time = time
+        func.verbose = __verbose
+        return func
+    return dec