Merge pull request #25 from elpres/master

Allow context manager invocation to suppress output
bos · Oct 15, 2015 · 1a33eba · 1a33eba
2 parents 52a6a7a + 8086b52
commit 1a33eba
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/statprof.py b/statprof.py
@@ -286,13 +286,14 @@ def reset(frequency=None):
 
 
 @contextmanager
-def profile():
+def profile(verbose=True):
     start()
     try:
         yield
     finally:
         stop()
-        display()
+        if verbose:
+            display()
 
 
 ###########################################################################