An environment-controllable way to turn on call debugging

1 files changed, 7 insertions, 4 deletions

diff --git a/coverage/control.py b/coverage/control.py
index 2130de39..a246bb7a 100644
--- a/coverage/control.py
+++ b/coverage/control.py
@@ -39,10 +39,6 @@ except ImportError:                                         # pragma: only jytho
 os = isolate_module(os)
 
 
-if 1:
-    from coverage.debug import decorate_methods, break_in_pudb, show_calls
-
-#@decorate_methods(show_calls(show_args=True), butnot=['get_data'])
 class Coverage(object):
     """Programmatic access to coverage.py.
 
@@ -866,6 +862,13 @@ class Coverage(object):
         return info
 
 
+# Mega debugging...
+if int(os.environ.get("COVERAGE_DEBUG_CALLS", 0)):
+    from coverage.debug import decorate_methods, show_calls
+
+    Coverage = decorate_methods(show_calls(show_args=True), butnot=['get_data'])(Coverage)
+
+
 def process_startup():
     """Call this at Python start-up to perhaps measure coverage.