Move code to __main__ block

1 files changed, 10 insertions, 9 deletions

diff --git a/examples/rewrite_ast.py b/examples/rewrite_ast.py
index 2c42f99..d38e015 100644
--- a/examples/rewrite_ast.py
+++ b/examples/rewrite_ast.py
@@ -18,14 +18,15 @@ void func(void)
 }
 """
 
-parser = c_parser.CParser()
-ast = parser.parse(text)
-print("Before:")
-ast.show(offset=2)
+if __name__ == '__main__':
+    parser = c_parser.CParser()
+    ast = parser.parse(text)
+    print("Before:")
+    ast.show(offset=2)
 
-assign = ast.ext[0].body.block_items[0]
-assign.lvalue.name = "y"
-assign.rvalue.value = 2
+    assign = ast.ext[0].body.block_items[0]
+    assign.lvalue.name = "y"
+    assign.rvalue.value = 2
 
-print("After:")
-ast.show(offset=2)
+    print("After:")
+    ast.show(offset=2)