1 files changed, 6 insertions, 13 deletions
diff --git a/Lib/ast.py b/Lib/ast.py
index 76e0cac..62f6e07 100644
--- a/Lib/ast.py
+++ b/Lib/ast.py
@@ -597,22 +597,15 @@ class _Unparser(NodeVisitor):
         self._buffer.clear()
         return value
 
-    class _Block:
+    @contextmanager
+    def block(self):
         """A context manager for preparing the source for blocks. It adds
         the character':', increases the indentation on enter and decreases
         the indentation on exit."""
-        def __init__(self, unparser):
-            self.unparser = unparser
-
-        def __enter__(self):
-            self.unparser.write(":")
-            self.unparser._indent += 1
-
-        def __exit__(self, exc_type, exc_value, traceback):
-            self.unparser._indent -= 1
-
-    def block(self):
-        return self._Block(self)
+        self.write(":")
+        self._indent += 1
+        yield
+        self._indent -= 1
 
     @contextmanager
     def delimit(self, start, end):