Add a context manager for increasing indents

jedi/debug.py

@@ -1,6 +1,8 @@
-from jedi._compatibility import encoding, is_py3, u
 import os
 import time
+from contextlib import contextmanager
+
+from jedi._compatibility import encoding, is_py3, u
 
 _inited = False
 
@@ -76,13 +78,19 @@ def reset_time():
 def increase_indent(func):
     """Decorator for makin """
     def wrapper(*args, **kwargs):
+        with increase_indent_cm():
+            return func(*args, **kwargs)
+    return wrapper
+
+
+@contextmanager
+def increase_indent_cm():
     global _debug_indent
     _debug_indent += 1
     try:
-            return func(*args, **kwargs)
+        yield
     finally:
         _debug_indent -= 1
-    return wrapper
 
 
 def dbg(message, *args, **kwargs):