summaryrefslogtreecommitdiff
path: root/examples/serialize_ast.py
blob: 8669a0a2ce14ab5d2c788e1402055cbc0046efa5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#-----------------------------------------------------------------
# pycparser: serialize_ast.py
#
# Simple example of serializing AST
#
# Hart Chu [https://github.com/CtheSky]
# Eli Bendersky [https://eli.thegreenplace.net/]
# License: BSD
#-----------------------------------------------------------------
from __future__ import print_function
import pickle
import sys

sys.path.extend(['.', '..'])
from pycparser import c_parser

text = r"""
void func(void)
{
  x = 1;
}
"""

if __name__ == '__main__':
    parser = c_parser.CParser()
    ast = parser.parse(text)

    # Since AST nodes use __slots__ for faster attribute access and
    # space saving, it needs Pickle's protocol version >= 2.
    # The default version is 3 for python 3.x and 1 for python 2.7.
    # You can always select the highest available protocol with the -1 argument.

    with open('ast', 'wb') as f:
        pickle.dump(ast, f, protocol=-1)

    # Deserialize.
    with open('ast', 'rb') as f:
        ast = pickle.load(f)
        ast.show()