diff options
Diffstat (limited to 'events/eventskeygen.py')
-rw-r--r-- | events/eventskeygen.py | 49 |
1 files changed, 27 insertions, 22 deletions
diff --git a/events/eventskeygen.py b/events/eventskeygen.py index f9bdb9fed52..5bb03191135 100644 --- a/events/eventskeygen.py +++ b/events/eventskeygen.py @@ -11,11 +11,14 @@ # import os +import sys GLUSTER_SRC_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) eventtypes_h = os.path.join(GLUSTER_SRC_ROOT, "libglusterfs/src/eventtypes.h") eventtypes_py = os.path.join(GLUSTER_SRC_ROOT, "events/src/eventtypes.py") +gen_header_type = sys.argv[1] + # When adding new keys add it to the END keys = ( "EVENT_PEER_ATTACH", @@ -39,29 +42,31 @@ ERRORS = ( "EVENT_ERROR_SEND" ) -# Generate eventtypes.h -with open(eventtypes_h, "w") as f: - f.write("#ifndef __EVENTTYPES_H__\n") - f.write("#define __EVENTTYPES_H__\n\n") - f.write("typedef enum {\n") - for k in ERRORS: - f.write(" {0},\n".format(k)) - f.write("} event_errors_t;\n") +if gen_header_type == "C_HEADER": + # Generate eventtypes.h + with open(eventtypes_h, "w") as f: + f.write("#ifndef __EVENTTYPES_H__\n") + f.write("#define __EVENTTYPES_H__\n\n") + f.write("typedef enum {\n") + for k in ERRORS: + f.write(" {0},\n".format(k)) + f.write("} event_errors_t;\n") - f.write("\n") + f.write("\n") - f.write("typedef enum {\n") - for k in keys: - f.write(" {0},\n".format(k)) + f.write("typedef enum {\n") + for k in keys: + f.write(" {0},\n".format(k)) - f.write(" {0}\n".format(LAST_EVENT)) - f.write("} eventtypes_t;\n") - f.write("\n#endif /* __EVENTTYPES_H__ */\n") + f.write(" {0}\n".format(LAST_EVENT)) + f.write("} eventtypes_t;\n") + f.write("\n#endif /* __EVENTTYPES_H__ */\n") -# Generate eventtypes.py -with open(eventtypes_py, "w") as f: - f.write("# -*- coding: utf-8 -*-\n") - f.write("all_events = [\n") - for ev in keys: - f.write(' "{0}",\n'.format(ev)) - f.write("]\n") +if gen_header_type == "PY_HEADER": + # Generate eventtypes.py + with open(eventtypes_py, "w") as f: + f.write("# -*- coding: utf-8 -*-\n") + f.write("all_events = [\n") + for ev in keys: + f.write(' "{0}",\n'.format(ev)) + f.write("]\n") |