| Akira Hayakawa | a5c4925 | 2014-08-22 13:34:53 +0900 | [diff] [blame] | 1 | #----------------------------------------------------------------- |
| 2 | # pycparser: func_write.py |
| 3 | # |
| 4 | # Tiny example of rewriting a AST node |
| 5 | # |
| 6 | # Copyright (C) 2014, Akira Hayakawa |
| 7 | # License: BSD |
| 8 | #----------------------------------------------------------------- |
| 9 | from __future__ import print_function |
| 10 | import sys |
| 11 | |
| 12 | from pycparser import c_parser |
| 13 | |
| 14 | text = r""" |
| 15 | void func(void) |
| 16 | { |
| 17 | x = 1; |
| 18 | } |
| 19 | """ |
| 20 | |
| 21 | parser = c_parser.CParser() |
| 22 | ast = parser.parse(text) |
| 23 | print("Before:") |
| 24 | ast.show(offset=2) |
| 25 | |
| 26 | assign = ast.ext[0].body.block_items[0] |
| 27 | assign.lvalue.name = "y" |
| 28 | assign.rvalue.value = 2 |
| 29 | |
| 30 | print("After:") |
| 31 | ast.show(offset=2) |