Use decorator.decorator consistenly in debug wrappers.

- Legacy-Id: 4936
2012-10-21 14:32:02 +00:00 · 2012-10-21 14:32:02 +00:00 · 1f8cc48792
parent ba95520523
commit 1f8cc48792
1 changed files with 4 additions and 1 deletions
--- a/debug.py
+++ b/debug.py
@ -130,5 +130,8 @@ def profile(fn):
        retval = prof.runcall(fn, *args, **kwargs)
        prof.dump_stats(datafn)
        return retval
-    return wrapper
+    if debug:
+        return decorator(wrapper, fn)
+    else:
+        return fn